feat: works
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,3 +1,4 @@
|
||||
.idea/
|
||||
# ---> macOS
|
||||
# General
|
||||
.DS_Store
|
||||
|
||||
8
Cargo.toml
Normal file
8
Cargo.toml
Normal file
@@ -0,0 +1,8 @@
|
||||
[package]
|
||||
name = "rotate-puts"
|
||||
version = "0.1.2"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
7
justfile
Normal file
7
justfile
Normal file
@@ -0,0 +1,7 @@
|
||||
_:
|
||||
@just --list
|
||||
|
||||
build-x64:
|
||||
cross build --target x86_64-unknown-linux-musl --release
|
||||
|
||||
|
||||
90
src/main.rs
Normal file
90
src/main.rs
Normal file
@@ -0,0 +1,90 @@
|
||||
use std::fs::File;
|
||||
use std::io::{BufWriter, Read, Write};
|
||||
use std::process::exit;
|
||||
use std::time::Duration;
|
||||
|
||||
fn main() {
|
||||
let (sender, receiver) = std::sync::mpsc::channel::<Vec<u8>>();
|
||||
std::thread::spawn(move || {
|
||||
let mut file_index = 0;
|
||||
let mut written_len = 0;
|
||||
|
||||
let file_name = make_new_file_name("temp", &mut file_index);
|
||||
let mut out_file = BufWriter::new(File::create(&file_name)
|
||||
.expect(&format!("Create file failed: {}", file_name)));
|
||||
|
||||
let mut write_buffer = Vec::with_capacity(1024 * 8);
|
||||
loop {
|
||||
match receiver.recv_timeout(Duration::from_secs(1)) {
|
||||
Err(_e) => {
|
||||
if write_buffer.len() > 0 {
|
||||
written_len += write_buffer.len();
|
||||
out_file.write(&write_buffer).ok();
|
||||
write_buffer.clear();
|
||||
}
|
||||
}
|
||||
Ok(buff) => {
|
||||
if buff.len() == 0 {
|
||||
out_file.write(&write_buffer).ok();
|
||||
out_file.flush().ok();
|
||||
exit(0);
|
||||
}
|
||||
write_buffer.extend_from_slice(&buff);
|
||||
if write_buffer.len() > 1024 {
|
||||
let mut pos_of_n = -1_isize;
|
||||
write_buffer.iter().enumerate().for_each(|(i, c)| {
|
||||
if *c == b'\n' { pos_of_n = i as isize; }
|
||||
});
|
||||
if pos_of_n > -1 {
|
||||
let left_buffer = write_buffer.split_off(pos_of_n as usize + 1);
|
||||
written_len += write_buffer.len();
|
||||
out_file.write(&write_buffer).ok();
|
||||
write_buffer = left_buffer;
|
||||
out_file.flush().ok();
|
||||
|
||||
if written_len >= 10 * 1024 * 1024 {
|
||||
written_len = 0;
|
||||
let file_name = make_new_file_name("temp", &mut file_index);
|
||||
println!("new file: {}", file_name);
|
||||
out_file = BufWriter::new(File::create(&file_name)
|
||||
.expect(&format!("Create file failed: {}", file_name)));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
let mut std_in = std::io::stdin();
|
||||
let mut buff = [0_u8; 128];
|
||||
loop {
|
||||
match std_in.read(&mut buff) {
|
||||
Ok(len) if len == 0 => {
|
||||
sender.send(Vec::new()).ok();
|
||||
}
|
||||
Ok(len) => {
|
||||
let mut vec = Vec::with_capacity(len);
|
||||
for b in &buff[0..len] {
|
||||
vec.push(*b);
|
||||
}
|
||||
if let Err(e) = sender.send(vec) {
|
||||
eprintln!("[ERROR] Send error: {}", e);
|
||||
}
|
||||
}
|
||||
Err(e) => eprintln!("[ERROR] Send error: {}", e),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn make_new_file_name(prefix: &str, index: &mut i32) -> String {
|
||||
let i = *index;
|
||||
*index = i + 1;
|
||||
|
||||
let pending_rm = format!("{}_{:03}.log", prefix, i - 10);
|
||||
if let Ok(_) = std::fs::metadata(&pending_rm) {
|
||||
std::fs::remove_file(&pending_rm).ok();
|
||||
}
|
||||
|
||||
format!("{}_{:03}.log", prefix, i)
|
||||
}
|
||||
Reference in New Issue
Block a user