1.添加多线程支持

2.优化代码结构
3.升级依赖
This commit is contained in:
Lkhsss
2024-08-17 22:05:34 +08:00
parent 7245260604
commit 4af4ce7405
9 changed files with 996 additions and 164 deletions

90
src/threadpool.rs Normal file
View File

@ -0,0 +1,90 @@
use std::sync::{mpsc, Arc, Mutex};
use std::thread::{self, JoinHandle};
use log::{info,debug};
use serde::de;
type Job = Box<dyn FnOnce() + 'static + Send>;
enum Message {
ByeBye,
NewJob(Job),
}
struct Worker {
_id: usize,
t: Option<JoinHandle<()>>,
}
impl Worker {
fn new(id: usize, receiver: Arc<Mutex<mpsc::Receiver<Message>>>) -> Worker {
let t = thread::spawn(move || loop {
let message = receiver.lock().unwrap().recv().unwrap();
match message {
Message::NewJob(job) => {
debug!("线程[{}]获得任务", id);
job();
}
Message::ByeBye => {
debug!("线程[{}]结束任务", id);
break;
}
}
});
Worker {
_id: id,
t: Some(t),
}
}
}
pub struct Pool {
workers: Vec<Worker>,
max_workers: usize,
sender: mpsc::Sender<Message>,
}
impl Pool {
pub fn new(max_workers: usize) -> Pool {
if max_workers == 0 {
panic!("最大线程数不能小于零!")
}else {
debug!("将开启{}线程",max_workers);
}
let (tx, rx) = mpsc::channel();
let mut workers = Vec::with_capacity(max_workers);
let receiver = Arc::new(Mutex::new(rx));
for i in 0..max_workers {
workers.push(Worker::new(i, Arc::clone(&receiver)));
}
Pool {
workers: workers,
max_workers: max_workers,
sender: tx,
}
}
pub fn execute<F>(&self, f: F)
where
F: FnOnce() + 'static + Send,
{
let job = Message::NewJob(Box::new(f));
self.sender.send(job).unwrap();
}
}
impl Drop for Pool {
fn drop(&mut self) {
for _ in 0..self.max_workers {
self.sender.send(Message::ByeBye).unwrap();
}
for w in self.workers.iter_mut() {
if let Some(t) = w.t.take() {
t.join().unwrap();
}
}
}
}