Class

apps/client/src/code/realtime/client.ts:26–490

export const RealtimeClient = once(
  () =>
    class extends ClientSocket() {
      private _nextCommandId = 0;

      private readonly _logger = mainLogger.sub('RealtimeClient');

      private readonly _hsetBuffer = new Map<string, any>();

      readonly commandBuffer: RealtimeCommand[] = [];
      private readonly _responsePromises = new Map<number, Resolvable<any>>();

      readonly subscriptions: Record<string, Set<RealtimeContext>> = {};
      readonly pending = reactive(new Map<string, Resolvable<any>>());
      readonly changed = new Set<string>();

      get loading() {
        return this.pending.size > 0;
      }

      readonly values: Record<string, any> = reactive({});

      readonly globalCtx = new (RealtimeContext())();

      private _isFirstConnection = true;

      constructor() {
        super(process.env.REALTIME_SERVER_URL);
      }

      connect() {
        super.connect();

        this.socket?.addEventListener('open', () => {
          if (this._isFirstConnection) {
            this._logger.info('First connection');
            this._isFirstConnection = false;
            return;
          }

          this._logger.info('Reconnecting');

          for (const fullKey of Object.keys(this.subscriptions)) {
            const [key, field] = splitStr(fullKey, '>', 2);
            const [prefix, suffix] = splitStr(key, ':', 2);

            this.commandBuffer.push({
              type: RealtimeCommandType.SUBSCRIBE,
              args: [prefix, suffix, field],
            });
          }

          if (this.commandBuffer.length > 0) {
            setTimeout(this.flushCommandBuffer);
          }
        });

        this.socket?.addEventListener('message', (event) => {
          this._handleMessage(new Uint8Array(event.data));
        });