Apply and configure rustfmt in the project
This commit is contained in:
parent
e19805f872
commit
ab587c98e4
3 changed files with 44 additions and 31 deletions
|
|
@ -1 +1,10 @@
|
||||||
language: rust
|
language: rust
|
||||||
|
|
||||||
|
before_script:
|
||||||
|
- rustup component add rustfmt-preview
|
||||||
|
|
||||||
|
script:
|
||||||
|
- cargo fmt --all -- --check
|
||||||
|
- cargo build
|
||||||
|
- cargo test
|
||||||
|
|
||||||
|
|
|
||||||
40
src/lib.rs
40
src/lib.rs
|
|
@ -1,45 +1,49 @@
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
use std::io::SeekFrom;
|
|
||||||
use std::io::BufReader;
|
|
||||||
use std::io::prelude::*;
|
|
||||||
use std::io;
|
use std::io;
|
||||||
|
use std::io::prelude::*;
|
||||||
|
use std::io::BufReader;
|
||||||
|
use std::io::ErrorKind;
|
||||||
|
use std::io::SeekFrom;
|
||||||
|
use std::os::unix::fs::MetadataExt;
|
||||||
use std::thread::sleep;
|
use std::thread::sleep;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
use std::os::unix::fs::MetadataExt;
|
|
||||||
use std::io::ErrorKind;
|
|
||||||
|
|
||||||
pub struct LogWatcher {
|
pub struct LogWatcher {
|
||||||
filename: String,
|
filename: String,
|
||||||
inode: u64,
|
inode: u64,
|
||||||
pos: u64,
|
pos: u64,
|
||||||
reader: BufReader<File>,
|
reader: BufReader<File>,
|
||||||
finish: bool
|
finish: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl LogWatcher {
|
impl LogWatcher {
|
||||||
pub fn register(filename: String) -> Result<LogWatcher, io::Error> {
|
pub fn register(filename: String) -> Result<LogWatcher, io::Error> {
|
||||||
let f = match File::open(filename.clone()) {
|
let f = match File::open(filename.clone()) {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(err) => return Err(err)
|
Err(err) => return Err(err),
|
||||||
};
|
};
|
||||||
|
|
||||||
let metadata = match f.metadata() {
|
let metadata = match f.metadata() {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(err) => return Err(err)
|
Err(err) => return Err(err),
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut reader = BufReader::new(f);
|
let mut reader = BufReader::new(f);
|
||||||
let pos = metadata.len();
|
let pos = metadata.len();
|
||||||
reader.seek(SeekFrom::Start(pos)).unwrap();
|
reader.seek(SeekFrom::Start(pos)).unwrap();
|
||||||
Ok(LogWatcher{filename: filename,
|
Ok(LogWatcher {
|
||||||
|
filename: filename,
|
||||||
inode: metadata.ino(),
|
inode: metadata.ino(),
|
||||||
pos: pos,
|
pos: pos,
|
||||||
reader: reader,
|
reader: reader,
|
||||||
finish: false})
|
finish: false,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn reopen_if_log_rotated<F: ?Sized>(&mut self, callback: &F)
|
fn reopen_if_log_rotated<F: ?Sized>(&mut self, callback: &F)
|
||||||
where F: Fn(String) {
|
where
|
||||||
|
F: Fn(String),
|
||||||
|
{
|
||||||
loop {
|
loop {
|
||||||
match File::open(self.filename.clone()) {
|
match File::open(self.filename.clone()) {
|
||||||
Ok(x) => {
|
Ok(x) => {
|
||||||
|
|
@ -59,12 +63,11 @@ impl LogWatcher {
|
||||||
self.reader = BufReader::new(f);
|
self.reader = BufReader::new(f);
|
||||||
self.pos = 0;
|
self.pos = 0;
|
||||||
self.inode = metadata.ino();
|
self.inode = metadata.ino();
|
||||||
}
|
} else {
|
||||||
else{
|
|
||||||
sleep(Duration::new(1, 0));
|
sleep(Duration::new(1, 0));
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
},
|
}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
if err.kind() == ErrorKind::NotFound {
|
if err.kind() == ErrorKind::NotFound {
|
||||||
sleep(Duration::new(1, 0));
|
sleep(Duration::new(1, 0));
|
||||||
|
|
@ -76,7 +79,9 @@ impl LogWatcher {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn watch<F: ?Sized>(&mut self, callback: &F)
|
pub fn watch<F: ?Sized>(&mut self, callback: &F)
|
||||||
where F: Fn(String) {
|
where
|
||||||
|
F: Fn(String),
|
||||||
|
{
|
||||||
loop {
|
loop {
|
||||||
let mut line = String::new();
|
let mut line = String::new();
|
||||||
let resp = self.reader.read_line(&mut line);
|
let resp = self.reader.read_line(&mut line);
|
||||||
|
|
@ -90,13 +95,12 @@ impl LogWatcher {
|
||||||
} else {
|
} else {
|
||||||
if self.finish {
|
if self.finish {
|
||||||
break;
|
break;
|
||||||
}
|
} else {
|
||||||
else{
|
|
||||||
self.reopen_if_log_rotated(callback);
|
self.reopen_if_log_rotated(callback);
|
||||||
self.reader.seek(SeekFrom::Start(self.pos)).unwrap();
|
self.reader.seek(SeekFrom::Start(self.pos)).unwrap();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
println!("{}", err);
|
println!("{}", err);
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue