增加线程名称

This commit is contained in:
lubeilin
2023-09-23 23:09:37 +08:00
parent 17b206bace
commit ba69ba78af
+41 -30
View File
@@ -678,15 +678,20 @@ impl Channel {
let main_channel = context.inner.main_channel.clone(); let main_channel = context.inner.main_channel.clone();
let buf_sender = if parallel > 1 { let buf_sender = if parallel > 1 {
let (buf_sender, buf_receiver) = buf_channel_group(parallel); let (buf_sender, buf_receiver) = buf_channel_group(parallel);
let mut num = 0;
for buf_receiver in buf_receiver.0 { for buf_receiver in buf_receiver.0 {
let context = context.clone(); let context = context.clone();
let handler = handler.clone(); let handler = handler.clone();
std::thread::spawn(move || { std::thread::Builder::new()
while let Ok((mut buf, start, end, route_key)) = buf_receiver.recv() { .name(format!("recv-handler-{}", num))
handler.handle(&mut buf, start, end, route_key, &context); .spawn(move || {
} while let Ok((mut buf, start, end, route_key)) = buf_receiver.recv() {
log::warn!("异步处理停止"); handler.handle(&mut buf, start, end, route_key, &context);
}); }
log::warn!("异步处理停止");
})
.unwrap();
num += 1;
} }
Some(buf_sender) Some(buf_sender)
} else { } else {
@@ -709,18 +714,21 @@ impl Channel {
let main_channel_ipv6 = main_channel_ipv6.clone(); let main_channel_ipv6 = main_channel_ipv6.clone();
let handler = handler.clone(); let handler = handler.clone();
let buf_sender = buf_sender.clone(); let buf_sender = buf_sender.clone();
std::thread::spawn(move || { std::thread::Builder::new()
log::info!("启动udp v6"); .name("ipv6-recv".into())
Self::main_start_( .spawn(move || {
worker, log::info!("启动udp v6");
context, Self::main_start_(
UDP_V6_ID, worker,
main_channel_ipv6, context,
handler, UDP_V6_ID,
buf_sender, main_channel_ipv6,
head_reserve, handler,
) buf_sender,
}); head_reserve,
)
})
.unwrap();
} }
{ {
let worker = worker.worker("main_channel_1"); let worker = worker.worker("main_channel_1");
@@ -728,18 +736,21 @@ impl Channel {
let main_channel = main_channel.clone(); let main_channel = main_channel.clone();
let handler = handler.clone(); let handler = handler.clone();
let buf_sender = buf_sender.clone(); let buf_sender = buf_sender.clone();
std::thread::spawn(move || { std::thread::Builder::new()
log::info!("启动udp v4"); .name("ipv4-recv".into())
Self::main_start_( .spawn(move || {
worker, log::info!("启动udp v4");
context, Self::main_start_(
UDP_ID, worker,
main_channel, context,
handler, UDP_ID,
buf_sender, main_channel,
head_reserve, handler,
) buf_sender,
}); head_reserve,
)
})
.unwrap();
} }
if relay { if relay {
worker.stop_wait().await; worker.stop_wait().await;