feat: RedisServerAffinity RFC: Lock one document to one server

packages/extension-redis-affinity/CHANGELOG.md

@@ -0,0 +1,4 @@
+# Change Log
+
+All notable changes to this project will be documented in this file.
+See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.

packages/extension-redis-affinity/package.json

@@ -0,0 +1,40 @@
+{
+	"name": "@hocuspocus/extension-redis-affinity",
+	"version": "3.2.5",
+	"description": "Guarantee server affinity with Redis",
+	"homepage": "https://hocuspocus.dev",
+	"keywords": [
+		"hocuspocus",
+		"redis",
+		"yjs"
+	],
+	"license": "MIT",
+	"type": "module",
+	"main": "dist/hocuspocus-redis-affinity.cjs",
+	"module": "dist/hocuspocus-redis-affinity.esm.js",
+	"types": "dist/packages/extension-redis-affinity/src/index.d.ts",
+	"exports": {
+		"source": {
+			"import": "./src/index.ts"
+		},
+		"default": {
+			"import": "./dist/hocuspocus-redis-affinity.esm.js",
+			"require": "./dist/hocuspocus-redis-affinity.cjs",
+			"types": "./dist/packages/extension-redis-affinity/src/index.d.ts"
+		}
+	},
+	"files": [
+		"src",
+		"dist"
+	],
+	"dependencies": {
+		"@hocuspocus/server": "^3.2.5",
+		"ioredis": "^5.6.1",
+		"lib0": "^0.2.114",
+		"tseep": "^1.3.1"
+	},
+	"peerDependencies": {
+		"ws": "^8.5.0"
+	},
+	"gitHead": "b3454a4ca289a84ddfb7fa5607a2d4b8d5c37e9d"
+}

packages/extension-redis-affinity/src/HocusPocusProxySocket.ts

@@ -0,0 +1,41 @@
+import type RedisClient from 'ioredis'
+import {EventEmitter} from 'tseep'
+import type {
+  Pack,
+  RSAMessageClose,
+  RSAMessagePing,
+  RSAMessageSend
+} from './RedisServerAffinity'
+
+export class HocusPocusProxySocket extends EventEmitter {
+  private replyTo: string
+  private socketId: string
+  private pub: RedisClient
+  private pack: Pack
+  readyState = 1
+  constructor(pub: RedisClient, pack: Pack, replyTo: string, socketId: string) {
+    super()
+    this.replyTo = replyTo
+    this.socketId = socketId
+    this.pub = pub
+    this.pack = pack
+    this.on('close', () => {
+      this.readyState = 3
+    })
+  }
+  private publish(msg: RSAMessageClose | RSAMessagePing | RSAMessageSend) {
+    this.pub.publish(this.replyTo, this.pack(msg))
+  }
+  close(code?: number, reason?: string) {
+    const msg: RSAMessageClose = {type: 'close', code, reason, socketId: this.socketId}
+    this.publish(msg)
+  }
+  ping() {
+    const msg: RSAMessagePing = {type: 'ping', socketId: this.socketId}
+    this.publish(msg)
+  }
+  send(message: Uint8Array) {
+    const msg: RSAMessageSend = {type: 'send', socketId: this.socketId, message}
+    this.publish(msg)
+  }
+}