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(); let liveness_tx = liveness_tx.clone();
join_set.spawn(async move { join_set.spawn(async move {
let _guard = liveness_tx; let _guard = liveness_tx;
let mut inner_set = JoinSet::new();
loop { loop {
if cancel_token.is_cancelled() {
return;
}
let client = sender.recv(); let client = sender.recv();
if client.is_none() { if client.is_none() {
continue; continue;
@ -152,18 +156,27 @@ async fn main() -> anyhow::Result<()> {
if cancel_token.is_cancelled() { if cancel_token.is_cancelled() {
return; return;
} }
let semaphore = semaphore.clone();
let cancel_token = cancel_token.clone();
let profile = profile.clone();
inner_set.spawn(async move {
loop { loop {
if client.get_client().await.is_err() { if client.get_client().await.is_err() {
break; return;
} }
if cancel_token.is_cancelled() { if cancel_token.is_cancelled() {
return; return;
} }
{
let permit = semaphore.clone().acquire_owned().await.unwrap(); let permit = semaphore.clone().acquire_owned().await.unwrap();
if cancel_token.is_cancelled() {
return;
}
let result = run(client.clone(), profile.clone()).await; let result = run(client.clone(), profile.clone()).await;
match result { match result {
Ok(true) => { Ok(true) => {
post_task(profile.clone(), permit, cancel_token.clone()).await; post_task(profile.clone(), permit, cancel_token.clone())
.await;
} }
Err(error) => { Err(error) => {
error!("{}", 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() { if cancel_token.is_cancelled() {
return; return;
} }
{
let permit = semaphore.clone().acquire_owned().await.unwrap(); let permit = semaphore.clone().acquire_owned().await.unwrap();
if cancel_token.is_cancelled() {
return;
}
let result = run(client.clone(), profile.clone()).await; let result = run(client.clone(), profile.clone()).await;
match result { match result {
Ok(true) => { Ok(true) => {
post_task(profile.clone(), permit, cancel_token.clone()).await; post_task(profile.clone(), permit, cancel_token.clone())
.await;
} }
Err(error) => { Err(error) => {
error!("{}", 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() { if cancel_token.is_cancelled() {
return; return;