logos-messaging/js-waku
1
0
Fork 0
mirror of https://github.com/logos-messaging/js-waku.git synced 2026-01-02 13:53:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add active query tracking to QueryOnConnect and await its stop()

Browse Source
This commit is contained in:
Levente Kiss 2025-10-31 18:46:10 +13:00 committed by fryorcraken
parent 9793b40275
commit a97e7eceb9
No known key found for this signature in database
GPG Key ID: A82ED75A8DFC50A4
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
packages/sdk/src/query_on_connect/query_on_connect.ts
View File

@ -54,6 +54,7 @@ export class QueryOnConnect<
private isRunning: boolean = false;
private abortController?: AbortController;
private activeQueryPromise?: Promise<void>;
private boundStoreConnectHandler?: (event: CustomEvent<PeerId>) => void;
private boundHealthHandler?: (event: CustomEvent<HealthStatus>) => void;
@ -87,7 +88,7 @@ export class QueryOnConnect<
this.setupEventListeners();
}
public stop(): void {
public async stop(): Promise<void> {
if (!this.isRunning) {
return;
}
@ -99,6 +100,15 @@ export class QueryOnConnect<
this.abortController = undefined;
}
if (this.activeQueryPromise) {
log.info("Waiting for active query to complete...");
try {
await this.activeQueryPromise;
} catch (error) {
log.warn("Active query failed during stop:", error);
}
}
this.unsetEventListeners();
}
@ -130,7 +140,10 @@ export class QueryOnConnect<
this.lastTimeOffline > this.lastSuccessfulQuery ||
timeSinceLastQuery > this.forceQueryThresholdMs
) {
await this.query(peerId);
this.activeQueryPromise = this.query(peerId).finally(() => {
this.activeQueryPromise = undefined;
});
await this.activeQueryPromise;
} else {
log.info(`no querying`);
}