Auto switch to the selected database.

Author: fenying

.vscode/launch.json

@@ -24,6 +24,16 @@
                 "${workspaceFolder}/dist/**/*.js"
             ]
         },
+        {
+            "type": "node",
+            "request": "launch",
+            "name": "Commands",
+            "program": "${workspaceFolder}/sources/samples/commands.ts",
+            "sourceMaps": true,
+            "outFiles": [
+                "${workspaceFolder}/dist/**/*.js"
+            ]
+        },
         {
             "type": "node",
             "request": "launch",

.vscode/tasks.json

@@ -0,0 +1,29 @@
+{
+    // See https://go.microsoft.com/fwlink/?LinkId=733558
+    // for the documentation about the tasks.json format
+    "version": "2.0.0",
+    "tasks": [
+        {
+            "label": "[Redis] Build (Watching)",
+            "command": "tsc",
+            "type": "shell",
+            "args": [
+                "-w",
+                "-p",
+                "."
+            ],
+            "presentation": {
+                "echo": true,
+                "reveal": "never",
+                "focus": false,
+                "panel": "shared"
+            },
+            "isBackground": true,
+            "group": {
+                "kind": "build",
+                "isDefault": true
+            },
+            "problemMatcher": "$tsc-watch"
+        }
+    ]
+}

CHANGES.md

@@ -1,5 +1,9 @@
 # Changes Logs
 
+## v0.1.6
+
+- Auto switch to the selected database.
+
 ## v0.1.5
 
 - Updated the dependency @litert/core to v0.6.0.

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@litert/redis",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "A redis protocol implement for Node.js.",
   "main": "./dist/index.js",
   "scripts": {

package.json

@@ -269,7 +269,7 @@ implements Abstract.ProtocolClient {
 
             this.emit("reconnected");
 
-            this._onReconnected();
+            this._onReconnected().catch((e) => this.emit("error", e));
 
         }).catch((e) => {
 
@@ -280,14 +280,11 @@ implements Abstract.ProtocolClient {
         });
     }
 
-    protected _onReconnected(): void {
+    protected async _onReconnected(): Promise<void> {
 
         // do something after reconnected event.
 
-        if (this._buffers.length > 0) {
-
-            this._stopPipeline();
-        }
+        this._stopPipeline();
     }
 
     protected _onError(err: NodeJS.ErrnoException): void {

sources/lib/ProtocolClient.ts

@@ -29,6 +29,8 @@ implements Abstract.RedisClient {
 
     private _password!: string;
 
+    private _database!: number;
+
     private _subscriber!: SubscriberClient;
 
     private _listeners!: Dict<Array<Abstract.SubscriptionCallback | Readable>>;
@@ -42,6 +44,8 @@ implements Abstract.RedisClient {
     ) {
 
         super(connection, host, port, createDecoder, createEncoder);
+
+        this._database = 0;
     }
 
     protected async _initializeSubscriber(): Promise<void> {
@@ -145,32 +149,33 @@ implements Abstract.RedisClient {
         return this;
     }
 
-    protected _onReconnected(): void {
+    protected async _onReconnected(): Promise<void> {
 
         if (this._password) {
 
             /**
              * Cache all imcoming commands before authentication completed.
              */
-            this._forcePipeline().executeNow(
-                "AUTH", this._password
-            ).catch((e) => {
-
-                delete this._password;
+            this._forcePipeline();
 
-                this.emit("error", e);
+            await this.executeNow(
+                "AUTH", this._password
+            );
+        }
 
-                super._onReconnected();
+        if (this._database) {
 
-            }).then(() => {
+            /**
+             * Cache all imcoming commands before authentication completed.
+             */
+            this._forcePipeline();
 
-                super._onReconnected();
-            });
+            await this.executeNow(
+                "SELECT", this._database.toString()
+            );
         }
-        else {
 
-            super._onReconnected();
-        }
+        return super._onReconnected();
     }
 
     public async exists(key: ItemKey): Promise<boolean> {
@@ -555,6 +560,8 @@ implements Abstract.RedisClient {
     public async select(index: number): Promise<void> {
 
         await this.execute<Buffer>("SELECT", index.toString());
+
+        this._database = index;
     }
 
     public async swapDB(db1: number, db2: number): Promise<void> {

sources/lib/RedisClient.ts

@@ -49,48 +49,43 @@ implements Abstract.ProtocolClient {
         this._password = password;
     }
 
-    protected _onReconnected(): void {
+    protected async _onReconnected(): Promise<void> {
 
         if (this._password) {
 
-            this._forcePipeline().executeNow(
-                "AUTH", this._password
-            ).then(() => {
+            this._forcePipeline();
 
-                this._resubscribe().catch((e) => {
-
-                    this.emit("error", new Exception(
-                        Constants.SUBSCRIBE_FAILURE,
-                        "Failed to resubscribe subjects.",
-                        e
-                    ));
-
-                    super._onReconnected();
-                });
+            try {
 
-            }).catch((e) => {
+                await this.executeNow(
+                    "AUTH", this._password
+                );
+            }
+            catch (e) {
 
                 this.emit("error", new Exception(
                     Constants.SUBSCRIBE_FAILURE,
                     "Failed to resubscribe subjects.",
                     e
                 ));
 
-                super._onReconnected();
-            });
+                return super._onReconnected();
+            }
         }
-        else {
 
-            this._resubscribe().catch((e) => {
+        try {
 
-                this.emit("error", new Exception(
-                    Constants.SUBSCRIBE_FAILURE,
-                    "Failed to resubscribe subjects.",
-                    e
-                ));
+            await this._resubscribe();
+        }
+        catch (e) {
+
+            this.emit("error", new Exception(
+                Constants.SUBSCRIBE_FAILURE,
+                "Failed to resubscribe subjects.",
+                e
+            ));
 
-                super._onReconnected();
-            });
+            return super._onReconnected();
         }
     }
 

sources/lib/SubscriberClient.ts

@@ -33,7 +33,7 @@ async function sleep(ms: number): Promise<void> {
 
         let result: any;
 
-        await client.auth("redis-password");
+        await client.auth("redis-passwd");
 
         result = await client.ping();
 
@@ -43,6 +43,8 @@ async function sleep(ms: number): Promise<void> {
 
         console.log(result.toString());
 
+        result = await client.select(1);
+
         await sleep(5000);
 
         while (1) {