use std::io;
use std::path::Path;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use thiserror::Error;
use crate::logs::content::LogEvent;
use crate::modules::logs::blocking::LogFileReaderError;
use crate::modules::shared::blocking::sync_blocker::SyncBlocker;
use super::RawLiveLogFileReader;
#[derive(Debug)]
pub struct LiveLogFileReader {
inner: RawLiveLogFileReader,
}
#[derive(Debug, Error)]
pub enum LiveLogFileReaderError {
#[error(transparent)]
IO(#[from] io::Error),
#[error(transparent)]
NotifyError(#[from] notify::Error),
#[error(transparent)]
LogFileReaderError(#[from] LogFileReaderError),
}
impl LiveLogFileReader {
pub fn open<P: AsRef<Path>>(path: P) -> Result<Self, LiveLogFileReaderError> {
Ok(LiveLogFileReader {
inner: RawLiveLogFileReader::open(path)?,
})
}
pub fn handle(&self) -> LiveLogFileHandle {
self.inner.handle()
}
}
#[derive(Debug, Clone)]
pub struct LiveLogFileHandle {
active: Arc<AtomicBool>,
blocker: SyncBlocker,
}
impl LiveLogFileHandle {
pub fn new(active: Arc<AtomicBool>, blocker: SyncBlocker) -> Self {
LiveLogFileHandle { active, blocker }
}
pub fn stop(&self) {
self.active.swap(false, Ordering::Relaxed);
self.blocker.unblock();
}
}
impl Iterator for LiveLogFileReader {
type Item = Result<LogEvent, LogFileReaderError>;
fn next(&mut self) -> Option<Self::Item> {
let result = match self.inner.next()? {
Ok(x) => x,
Err(e) => return Some(Err(e)),
};
Some(serde_json::from_value(result).map_err(|e| e.into()))
}
}