@nocobase/server 1.9.0-beta.5 → 1.9.0-beta.6

package/lib/application.js

@@ -279,6 +279,9 @@ const _Application = class _Application extends import_koa.default {
     if (!WORKER_MODE) {
       return true;
     }
+    if (WORKER_MODE === "-") {
+      return false;
+    }
     const topics = WORKER_MODE.trim().split(",");
     if (key) {
       if (WORKER_MODE === "*") {

package/lib/audit-manager/index.js

@@ -252,6 +252,9 @@ const _AuditManager = class _AuditManager {
   async output(ctx, reqId, metadata) {
     var _a;
     try {
+      if (!ctx.action) {
+        return;
+      }
       const { resourceName, actionName } = ctx.action;
       const action = this.getAction(actionName, resourceName);
       if (!action) {

package/lib/event-queue.js

@@ -195,6 +195,7 @@ const _MemoryEventQueueAdapter = class _MemoryEventQueueAdapter {
   publish(channel, content, options = { timestamp: Date.now() }) {
     const event = this.events.get(channel);
     if (!event) {
+      console.debug(`memory queue (${channel}) not subscribed, skip`);
       return;
     }
     if (!this.queues.get(channel)) {
@@ -265,14 +266,16 @@ const _EventQueue = class _EventQueue {
   constructor(app, options = {}) {
     this.app = app;
     this.options = options;
-    this.setAdapter(new MemoryEventQueueAdapter({ appName: this.app.name }));
-    app.on("afterStart", async () => {
-      await this.connect();
-    });
-    app.on("beforeStop", async () => {
-      app.logger.info("[queue] gracefully shutting down...");
-      await this.close();
-    });
+    if (app.serving()) {
+      this.setAdapter(new MemoryEventQueueAdapter({ appName: this.app.name }));
+      app.on("afterStart", async () => {
+        await this.connect();
+      });
+      app.on("beforeStop", async () => {
+        app.logger.info("[queue] gracefully shutting down...");
+        await this.close();
+      });
+    }
   }
   adapter;
   events = /* @__PURE__ */ new Map();
@@ -296,7 +299,12 @@ const _EventQueue = class _EventQueue {
     if (!this.adapter) {
       throw new Error("no adapter set, cannot connect");
     }
+    if (!this.app.serving()) {
+      this.app.logger.warn("app is not serving, will not connect to event queue");
+      return;
+    }
     await this.adapter.connect();
+    this.app.logger.debug(`connected to adapter, using memory? ${this.adapter instanceof MemoryEventQueueAdapter}`);
     for (const [channel, event] of this.events.entries()) {
       this.adapter.subscribe(this.getFullChannel(channel), event);
     }
@@ -337,7 +345,7 @@ const _EventQueue = class _EventQueue {
       throw new Error("event queue not connected, cannot publish");
     }
     const c = this.getFullChannel(channel);
-    this.app.logger.debug("event queue publishing:", { channel: c, message });
+    this.app.logger.debug(`event queue publishing to channel(${c})`, { message });
     await this.adapter.publish(c, message, {
       timeout: QUEUE_DEFAULT_ACK_TIMEOUT,
       ...options,

package/lib/gateway/ws-server.js

@@ -172,6 +172,20 @@ const _WSServer = class _WSServer extends import_events.default {
         message
       );
     });
+    app.on("ws:sendToUser", ({ userId, message }) => {
+      this.sendToAppUser(app.name, userId, message);
+      app.logger.trace(`[broadcasting message] ws:sendToUser for user ${userId}`, { message });
+      app.pubSubManager.publish(
+        "ws:sendToUser",
+        {
+          userId,
+          message
+        },
+        {
+          skipSelf: true
+        }
+      );
+    });
     app.on("ws:sendToClient", ({ clientId, message }) => {
       this.sendToClient(clientId, message);
     });
@@ -184,6 +198,12 @@ const _WSServer = class _WSServer extends import_events.default {
     app.on("ws:authorized", ({ clientId, userId }) => {
       this.sendToClient(clientId, { type: "authorized" });
     });
+    app.on("afterLoad", () => {
+      app.pubSubManager.subscribe("ws:sendToUser", ({ userId, message }) => {
+        app.logger.debug(`[receive broadcasting message] ws:sendToUser for user ${userId}`, { message });
+        this.sendToAppUser(app.name, userId, message);
+      });
+    });
   }
   addNewConnection(ws, request) {
     const id = (0, import_nanoid.nanoid)();

package/lib/pub-sub-manager/pub-sub-manager.d.ts

@@ -11,11 +11,12 @@ import { HandlerManager } from './handler-manager';
 import { PubSubCallback, type IPubSubAdapter, type PubSubManagerOptions, type PubSubManagerPublishOptions, type PubSubManagerSubscribeOptions } from './types';
 export declare const createPubSubManager: (app: Application, options: PubSubManagerOptions) => PubSubManager;
 export declare class PubSubManager {
+    protected app: Application;
     protected options: PubSubManagerOptions;
     protected publisherId: string;
     protected adapter: IPubSubAdapter;
     protected handlerManager: HandlerManager;
-    constructor(options?: PubSubManagerOptions);
+    constructor(app: Application, options?: PubSubManagerOptions);
     get channelPrefix(): string;
     setAdapter(adapter: IPubSubAdapter): void;
     isConnected(): Promise<boolean>;
@@ -23,5 +24,5 @@ export declare class PubSubManager {
     close(): Promise<any>;
     subscribe(channel: string, callback: PubSubCallback, options?: PubSubManagerSubscribeOptions): Promise<void>;
     unsubscribe(channel: string, callback: PubSubCallback): Promise<any>;
-    publish(channel: string, message: any, options?: PubSubManagerPublishOptions): Promise<any>;
+    publish(channel: string, message: any, options?: PubSubManagerPublishOptions): Promise<void>;
 }

package/lib/pub-sub-manager/pub-sub-manager.js

@@ -34,17 +34,20 @@ module.exports = __toCommonJS(pub_sub_manager_exports);
 var import_utils = require("@nocobase/utils");
 var import_handler_manager = require("./handler-manager");
 const createPubSubManager = /* @__PURE__ */ __name((app, options) => {
-  const pubSubManager = new PubSubManager(options);
-  app.on("afterStart", async () => {
-    await pubSubManager.connect();
-  });
-  app.on("afterStop", async () => {
-    await pubSubManager.close();
-  });
+  const pubSubManager = new PubSubManager(app, options);
+  if (app.serving()) {
+    app.on("afterStart", async () => {
+      await pubSubManager.connect();
+    });
+    app.on("afterStop", async () => {
+      await pubSubManager.close();
+    });
+  }
   return pubSubManager;
 }, "createPubSubManager");
 const _PubSubManager = class _PubSubManager {
-  constructor(options = {}) {
+  constructor(app, options = {}) {
+    this.app = app;
     this.options = options;
     this.publisherId = (0, import_utils.uid)();
     this.handlerManager = new import_handler_manager.HandlerManager(this.publisherId);
@@ -69,8 +72,13 @@ const _PubSubManager = class _PubSubManager {
     if (!this.adapter) {
       return;
     }
+    if (!this.app.serving()) {
+      this.app.logger.warn("app is not serving, will not connect to event queue");
+      return;
+    }
     await this.adapter.connect();
     await this.handlerManager.each(async (channel, headler) => {
+      this.app.logger.debug(`[PubSubManager] subscribe ${channel} added before connected`);
       await this.adapter.subscribe(`${this.channelPrefix}${channel}`, headler);
     });
   }
@@ -84,6 +92,7 @@ const _PubSubManager = class _PubSubManager {
     await this.unsubscribe(channel, callback);
     const handler = this.handlerManager.set(channel, callback, options);
     if (await this.isConnected()) {
+      this.app.logger.debug(`[PubSubManager] subscribe ${channel} added after connected`);
       await this.adapter.subscribe(`${this.channelPrefix}${channel}`, handler);
     }
   }
@@ -97,6 +106,9 @@ const _PubSubManager = class _PubSubManager {
   async publish(channel, message, options) {
     var _a;
     if (!((_a = this.adapter) == null ? void 0 : _a.isConnected())) {
+      this.app.logger.warn(
+        `[PubSubManager] adapter is not exist or not connected, cannot publish message to channel ${channel}`
+      );
       return;
     }
     const wrappedMessage = JSON.stringify({
@@ -104,7 +116,8 @@ const _PubSubManager = class _PubSubManager {
       ...options,
       message
     });
-    return this.adapter.publish(`${this.channelPrefix}${channel}`, wrappedMessage);
+    await this.adapter.publish(`${this.channelPrefix}${channel}`, wrappedMessage);
+    this.app.logger.trace(`[PubSubManager] published message to channel ${channel}`);
   }
 };
 __name(_PubSubManager, "PubSubManager");

package/lib/sync-message-manager.d.ts

@@ -16,7 +16,7 @@ export declare class SyncMessageManager {
     protected pubSubManager: PubSubManager;
     constructor(app: Application, options?: any);
     get debounce(): any;
-    publish(channel: string, message: any, options?: PubSubManagerPublishOptions & Transactionable): Promise<any>;
+    publish(channel: string, message: any, options?: PubSubManagerPublishOptions & Transactionable): Promise<unknown>;
     subscribe(channel: string, callback: PubSubCallback): Promise<void>;
     unsubscribe(channel: string, callback: PubSubCallback): Promise<any>;
     sync(): Promise<void>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nocobase/server",
-  "version": "1.9.0-beta.5",
+  "version": "1.9.0-beta.6",
   "main": "lib/index.js",
   "types": "./lib/index.d.ts",
   "license": "AGPL-3.0",
@@ -10,19 +10,19 @@
     "@koa/cors": "^5.0.0",
     "@koa/multer": "^3.1.0",
     "@koa/router": "^13.1.0",
-    "@nocobase/acl": "1.9.0-beta.5",
-    "@nocobase/actions": "1.9.0-beta.5",
-    "@nocobase/auth": "1.9.0-beta.5",
-    "@nocobase/cache": "1.9.0-beta.5",
-    "@nocobase/data-source-manager": "1.9.0-beta.5",
-    "@nocobase/database": "1.9.0-beta.5",
-    "@nocobase/evaluators": "1.9.0-beta.5",
-    "@nocobase/lock-manager": "1.9.0-beta.5",
-    "@nocobase/logger": "1.9.0-beta.5",
-    "@nocobase/resourcer": "1.9.0-beta.5",
-    "@nocobase/sdk": "1.9.0-beta.5",
-    "@nocobase/telemetry": "1.9.0-beta.5",
-    "@nocobase/utils": "1.9.0-beta.5",
+    "@nocobase/acl": "1.9.0-beta.6",
+    "@nocobase/actions": "1.9.0-beta.6",
+    "@nocobase/auth": "1.9.0-beta.6",
+    "@nocobase/cache": "1.9.0-beta.6",
+    "@nocobase/data-source-manager": "1.9.0-beta.6",
+    "@nocobase/database": "1.9.0-beta.6",
+    "@nocobase/evaluators": "1.9.0-beta.6",
+    "@nocobase/lock-manager": "1.9.0-beta.6",
+    "@nocobase/logger": "1.9.0-beta.6",
+    "@nocobase/resourcer": "1.9.0-beta.6",
+    "@nocobase/sdk": "1.9.0-beta.6",
+    "@nocobase/telemetry": "1.9.0-beta.6",
+    "@nocobase/utils": "1.9.0-beta.6",
     "@types/decompress": "4.2.7",
     "@types/ini": "^1.3.31",
     "@types/koa-send": "^4.1.3",
@@ -57,5 +57,5 @@
     "@types/serve-handler": "^6.1.1",
     "@types/ws": "^8.5.5"
   },
-  "gitHead": "9a61c60dd3db5af64244e82447309b0cb17aabb6"
+  "gitHead": "ada5e359a91135cb9baf605c04c5c380f067a046"
 }