feat: add env for disabling worker

Author: stanley2058

src/api.js

@@ -19,6 +19,7 @@ let ydocUpdateCallback = env.getConf('ydoc-update-callback')
 if (ydocUpdateCallback != null && ydocUpdateCallback.slice(-1) !== '/') {
   ydocUpdateCallback += '/'
 }
+const WORKER_DISABLED = env.getConf('y-worker-disabled') === 'true'
 
 /**
  * @param {string} a
@@ -117,20 +118,25 @@ export class Api {
     this.redisWorkerGroupName = this.prefix + ':worker'
     this.workerSetName = `${this.prefix}:worker:${this.consumername}:idset`
     this._destroyed = false
+
+    const addScript = WORKER_DISABLED
+      ? 'redis.call("XADD", KEYS[1], "*", "m", ARGV[1])'
+      : `
+          if redis.call("EXISTS", KEYS[1]) == 0 then
+            redis.call("XADD", "${this.redisWorkerStreamName}", "*", "compact", KEYS[1])
+          elseif redis.call("XLEN", KEYS[1]) > 100 then
+            redis.call("SADD", "${this.prefix}:worker:checklist", KEYS[1])
+          end
+          redis.call("XADD", KEYS[1], "*", "m", ARGV[1])
+        `
+
     this.redis = redis.createClient({
       url,
       // scripting: https://github.com/redis/node-redis/#lua-scripts
       scripts: {
         addMessage: redis.defineScript({
           NUMBER_OF_KEYS: 1,
-          SCRIPT: `
-            -- if redis.call("EXISTS", KEYS[1]) == 0 then
-            --   redis.call("XADD", "${this.redisWorkerStreamName}", "*", "compact", KEYS[1])
-            -- elseif redis.call("XLEN", KEYS[1]) > 100 then
-            --   redis.call("SADD", "${this.prefix}:worker:checklist", KEYS[1])
-            -- end
-            redis.call("XADD", KEYS[1], "*", "m", ARGV[1])
-          `,
+          SCRIPT: addScript,
           /**
            * @param {string} key
            * @param {Buffer} message

src/y-socket-io/y-socket-io.js

@@ -17,6 +17,7 @@ import toobusy from 'toobusy-js'
 const logSocketIO = createModuleLogger('@y/socket-io/server')
 const PERSIST_INTERVAL = number.parseInt(env.getConf('y-socket-io-server-persist-interval') || '3000')
 const REVALIDATE_TIMEOUT = number.parseInt(env.getConf('y-socket-io-server-revalidate-timeout') || '60000')
+const WORKER_DISABLED = env.getConf('y-worker-disabled') === 'true'
 
 process.on('SIGINT', function () {
   // calling .shutdown allows your process to exit normally
@@ -200,10 +201,8 @@ export class YSocketIO {
       ;(async () => {
         assert(this.client)
         assert(socket.user)
-        const doc =
-          this.namespaceDocMap.get(namespace) ||
-          (await this.client.getDoc(namespace, 'index'))
-        this.namespaceDocMap.set(namespace, doc)
+        const doc = WORKER_DISABLED && this.namespaceDocMap.get(namespace) || (await this.client.getDoc(namespace, 'index'))
+        if (WORKER_DISABLED) this.namespaceDocMap.set(namespace, doc)
 
         if (
           api.isSmallerRedisId(doc.redisLastId, socket.user.initialRedisSubId)
@@ -248,10 +247,8 @@ export class YSocketIO {
       ) => {
         assert(this.client)
         const namespace = this.getNamespaceString(socket.nsp)
-        const doc =
-          this.namespaceDocMap.get(namespace) ||
-          (await this.client.getDoc(namespace, 'index'))
-        this.namespaceDocMap.set(namespace, doc)
+        const doc = WORKER_DISABLED && this.namespaceDocMap.get(namespace) || (await this.client.getDoc(namespace, 'index'))
+        if (WORKER_DISABLED) this.namespaceDocMap.set(namespace, doc)
         assert(doc)
         syncStep2(Y.encodeStateAsUpdate(doc.ydoc, stateVector))
       }
@@ -404,6 +401,8 @@ export class YSocketIO {
       nsp.emit('awareness-update', msg)
     }
 
+    if (!WORKER_DISABLED) return
+
     let changed = false
     const existDoc = this.namespaceDocMap.get(namespace)
     if (existDoc) {