rusty_tip/

logger.rs

use log::debug;
use serde::{de::DeserializeOwned, Serialize};
use std::{io::Write, path::PathBuf};

use crate::NanonisError;

// Removed LogEntry wrapper - ActionLogEntry already has timestamps

#[derive(Debug)]
pub struct Logger<T>
where
    T: Serialize + Clone + DeserializeOwned,
{
    buffer: Vec<T>,
    buffer_size: usize,
    file_path: PathBuf,
    final_format_json: bool, // If true, convert to JSON on final flush
    flush_failures: usize,
    max_flush_failures: usize,
}

impl<T> Logger<T>
where
    T: Serialize + Clone + DeserializeOwned,
{
    pub fn new<P: Into<PathBuf>>(
        file_path: P,
        buffer_size: usize,
        final_format_json: bool,
    ) -> Self {
        let mut path = file_path.into();

        // Automatically add appropriate file extension
        if final_format_json {
            // For JSON output, ensure .json extension
            if path.extension().is_none()
                || path.extension() != Some(std::ffi::OsStr::new("json"))
            {
                path.set_extension("json");
            }
        } else {
            // For JSONL output, ensure .jsonl extension
            if path.extension().is_none()
                || path.extension() != Some(std::ffi::OsStr::new("jsonl"))
            {
                path.set_extension("jsonl");
            }
        }

        Self {
            buffer: Vec::with_capacity(buffer_size),
            buffer_size,
            file_path: path,
            final_format_json,
            flush_failures: 0,
            max_flush_failures: 10,
        }
    }

    pub fn add(&mut self, data: T) -> Result<(), NanonisError> {
        self.buffer.push(data);

        if self.buffer.len() >= self.buffer_size {
            self.flush()?;
        }

        Ok(())
    }

    pub fn flush(&mut self) -> Result<(), NanonisError> {
        if self.buffer.is_empty() {
            return Ok(());
        }

        // Always write JSONL for intermediate flushes (efficient)
        let file_result = std::fs::OpenOptions::new()
            .create(true)
            .append(true)
            .open(&self.file_path);

        let file = match file_result {
            Ok(f) => f,
            Err(e) => {
                self.flush_failures += 1;
                log::error!(
                    "Flush failure {}/{}: Failed to open log file: {}",
                    self.flush_failures,
                    self.max_flush_failures,
                    e
                );

                // Periodic warning
                if self.flush_failures > 0 && self.flush_failures % 3 == 0 {
                    log::warn!(
                        "Experiencing intermittent flush failures ({}/{})",
                        self.flush_failures,
                        self.max_flush_failures
                    );
                }

                if self.flush_failures >= self.max_flush_failures {
                    return Err(NanonisError::Io {
                        source: e,
                        context: format!(
                            "Too many consecutive flush failures ({}) for {:?}",
                            self.max_flush_failures, self.file_path
                        ),
                    });
                }

                // Don't fail the experiment for transient flush errors
                return Ok(());
            }
        };

        let mut writer = std::io::BufWriter::new(file);

        // Write data
        let write_result = (|| {
            for data in &self.buffer {
                let json_line = serde_json::to_string(data)?;
                writeln!(writer, "{}", json_line)?;
            }
            writer.flush()?;
            Ok::<(), NanonisError>(())
        })();

        match write_result {
            Ok(_) => {
                self.flush_failures = 0; // Reset on success
                self.buffer.clear();
                debug!("Logger flushed successfully to file");
                Ok(())
            }
            Err(e) => {
                self.flush_failures += 1;
                log::error!(
                    "Flush failure {}/{}: Write error: {}",
                    self.flush_failures,
                    self.max_flush_failures,
                    e
                );

                // Periodic warning
                if self.flush_failures > 0 && self.flush_failures % 3 == 0 {
                    log::warn!(
                        "Experiencing intermittent flush failures ({}/{})",
                        self.flush_failures,
                        self.max_flush_failures
                    );
                }

                if self.flush_failures >= self.max_flush_failures {
                    return Err(NanonisError::Io {
                        source: std::io::Error::other(e.to_string()),
                        context: format!(
                            "Too many consecutive flush failures ({}) for {:?}",
                            self.max_flush_failures, self.file_path
                        ),
                    });
                }

                // Don't fail the experiment for transient flush errors
                Ok(())
            }
        }
    }

    /// Convert JSONL file to JSON array format (for final post-experiment analysis)
    pub fn finalize_as_json(&mut self) -> Result<(), NanonisError> {
        if !self.final_format_json {
            return Ok(()); // No conversion needed
        }

        // First flush any remaining buffer
        self.flush()?;

        // Read all JSONL entries
        let content =
            std::fs::read_to_string(&self.file_path).map_err(|source| {
                NanonisError::Io {
                    source,
                    context: format!(
                        "Could not read JSONL file at {:?}",
                        self.file_path
                    ),
                }
            })?;

        let mut entries = Vec::new();
        for line in content.lines() {
            if !line.trim().is_empty() {
                let data: T = serde_json::from_str(line)?;
                entries.push(data);
            }
        }

        // Write as JSON array with pretty formatting
        let json_output = serde_json::to_string_pretty(&entries)?;
        std::fs::write(&self.file_path, json_output).map_err(|source| {
            NanonisError::Io {
                source,
                context: format!(
                    "Could not write JSON file at {:?}",
                    self.file_path
                ),
            }
        })?;

        debug!(
            "Converted {} entries from JSONL to JSON format",
            entries.len()
        );
        Ok(())
    }

    pub fn len(&self) -> usize {
        self.buffer.len()
    }

    pub fn is_empty(&self) -> bool {
        self.buffer.len() == 0
    }
}

impl<T> Drop for Logger<T>
where
    T: Serialize + Clone + DeserializeOwned,
{
    fn drop(&mut self) {
        let _ = self.flush();
        let _ = self.finalize_as_json();
    }
}