sune-org/ORP
1
0
Fork 0
mirror of https://github.com/sune-org/ORP.git synced 2026-01-13 16:17:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Add extensive DO lifecycle notifications

Browse Source
This commit is contained in:
multipleof4
2025-11-06 18:29:12 -08:00
parent 46e233eabb
commit 61f04c5bf6
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
index.js
View File

@@ -56,6 +56,7 @@ export class MyDurableObject {
this.env = env; this.env = env;
this.sockets = new Set(); this.sockets = new Set();
this.reset(); this.reset();
this.notify(`DO instance created`, 2, ['construction_worker']);
} }
reset() { reset() {
@@ -84,10 +85,11 @@ export class MyDurableObject {
notify(msg, pri = 3, tags = []) { notify(msg, pri = 3, tags = []) {
if (!this.env.NTFY_TOPIC) return; if (!this.env.NTFY_TOPIC) return;
const title = `Sune ORP [${this.state.id.toString().slice(-8)}]`;
this.state.waitUntil(fetch(`https://ntfy.sh/${this.env.NTFY_TOPIC}`, { this.state.waitUntil(fetch(`https://ntfy.sh/${this.env.NTFY_TOPIC}`, {
method: 'POST', method: 'POST',
body: msg, body: msg,
headers: { 'Title': 'Sune ORP', 'Priority': `${pri}`, 'Tags': tags.join(',') }, headers: { 'Title': title, 'Priority': `${pri}`, 'Tags': tags.join(',') },
}).catch(e => console.error('ntfy failed:', e))); }).catch(e => console.error('ntfy failed:', e)));
} }
@@ -154,7 +156,11 @@ export class MyDurableObject {
const [client, server] = Object.values(new WebSocketPair()); const [client, server] = Object.values(new WebSocketPair());
server.accept(); server.accept();
this.sockets.add(server); this.sockets.add(server);
server.addEventListener('close', () => this.sockets.delete(server)); this.notify(`WS connected. Sockets: ${this.sockets.size}`, 2, ['wave']);
server.addEventListener('close', () => {
this.sockets.delete(server);
this.notify(`WS disconnected. Sockets: ${this.sockets.size}`, 2, ['end']);
});
server.addEventListener('message', e => this.state.waitUntil(this.onMessage(server, e))); server.addEventListener('message', e => this.state.waitUntil(this.onMessage(server, e)));
return new Response(null, { status: 101, webSocket: client }); return new Response(null, { status: 101, webSocket: client });
} }
@@ -336,6 +342,7 @@ export class MyDurableObject {
try { this.oaStream?.controller?.abort(); } catch {} try { this.oaStream?.controller?.abort(); } catch {}
this.saveSnapshot(); this.saveSnapshot();
this.bcast({ type: 'done' }); this.bcast({ type: 'done' });
this.notify(`Run ${this.rid} finished successfully`, 3, ['tada']);
this.state.waitUntil(this.stopHeartbeat()); this.state.waitUntil(this.stopHeartbeat());
} }
@@ -359,7 +366,9 @@ export class MyDurableObject {
} }
async stopHeartbeat() { async stopHeartbeat() {
if (!this.hbActive) return;
this.hbActive = false; this.hbActive = false;
this.notify(`Heartbeat stopped`, 2, ['stop_sign']);
await this.state.storage.setAlarm(null).catch(() => {}); await this.state.storage.setAlarm(null).catch(() => {});
} }