Api: only return session messages

1 files changed, 6 insertions, 8 deletions

diff --git a/packages/function/src/api.ts b/packages/function/src/api.ts
index a312b1280..e2a773453 100644
--- a/packages/function/src/api.ts
+++ b/packages/function/src/api.ts
@@ -19,9 +19,9 @@ export class SyncServer extends DurableObject<Env> {
     this.ctx.acceptWebSocket(server)
 
     const data = await this.ctx.storage.list()
-    for (const [key, content] of data.entries()) {
-      server.send(JSON.stringify({ key, content }))
-    }
+    Array.from(data.entries())
+      .filter(([key, _]) => key.startsWith("session/"))
+      .map(([key, content]) => server.send(JSON.stringify({ key, content })))
 
     return new Response(null, {
       status: 101,
@@ -71,11 +71,9 @@ export class SyncServer extends DurableObject<Env> {
 
   public async getData() {
     const data = await this.ctx.storage.list()
-    const messages = []
-    for (const [key, content] of data.entries()) {
-      messages.push({ key, content })
-    }
-    return messages
+    return Array.from(data.entries())
+      .filter(([key, _]) => key.startsWith("session/"))
+      .map(([key, content]) => ({ key, content }))
   }
 
   private async getSecret() {