feat: works

This commit is contained in:
2022-07-30 23:42:53 +08:00
parent f0ef479988
commit 36582e624a
4 changed files with 106 additions and 0 deletions

1
.gitignore vendored
View File

@@ -1,3 +1,4 @@
.idea/
# ---> macOS
# General
.DS_Store

8
Cargo.toml Normal file
View 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
View File

@@ -0,0 +1,7 @@
_:
@just --list
build-x64:
cross build --target x86_64-unknown-linux-musl --release

90
src/main.rs Normal file
View 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)
}