Bluemangoo/sekai-unpacker
1
0
Fork 0
mirror of https://github.com/Bluemangoo/sekai-unpacker.git synced 2026-05-06 20:44:47 +08:00
Code

fix multiple server

Browse Source
This commit is contained in:
Bluemangoo 2026-04-11 12:21:51 +08:00
parent 3d5ca3b1f8
commit c07e5b835e
Signed by: Bluemangoo
GPG Key ID: F2F7E46880A1C4CF
1 changed files with 47 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
client/src/main.rs
View File

@ -143,7 +143,11 @@ async fn main() -> anyhow::Result<()> {
let liveness_tx = liveness_tx.clone();
join_set.spawn(async move {
let _guard = liveness_tx;
let mut inner_set = JoinSet::new();
loop {
if cancel_token.is_cancelled() {
return;
}
let client = sender.recv();
if client.is_none() {
continue;
@ -152,18 +156,27 @@ async fn main() -> anyhow::Result<()> {
if cancel_token.is_cancelled() {
return;
}
let semaphore = semaphore.clone();
let cancel_token = cancel_token.clone();
let profile = profile.clone();
inner_set.spawn(async move {
loop {
if client.get_client().await.is_err() {
break;
return;
}
if cancel_token.is_cancelled() {
return;
}
{
let permit = semaphore.clone().acquire_owned().await.unwrap();
if cancel_token.is_cancelled() {
return;
}
let result = run(client.clone(), profile.clone()).await;
match result {
Ok(true) => {
post_task(profile.clone(), permit, cancel_token.clone()).await;
post_task(profile.clone(), permit, cancel_token.clone())
.await;
}
Err(error) => {
error!("{}", error);
@ -171,6 +184,9 @@ async fn main() -> anyhow::Result<()> {
_ => {}
}
}
sleep(Duration::from_millis(100)).await; // let other client get semaphore
}
});
}
});
}
@ -202,11 +218,16 @@ async fn main() -> anyhow::Result<()> {
if cancel_token.is_cancelled() {
return;
}
{
let permit = semaphore.clone().acquire_owned().await.unwrap();
if cancel_token.is_cancelled() {
return;
}
let result = run(client.clone(), profile.clone()).await;
match result {
Ok(true) => {
post_task(profile.clone(), permit, cancel_token.clone()).await;
post_task(profile.clone(), permit, cancel_token.clone())
.await;
}
Err(error) => {
error!("{}", error);
@ -214,6 +235,8 @@ async fn main() -> anyhow::Result<()> {
_ => {}
}
}
sleep(Duration::from_millis(100)).await; // let other client get semaphore
}
}
if cancel_token.is_cancelled() {
return;