fix: remove singleton instance after consumer close (#35)

lib/client/client.js

@@ -79,11 +79,14 @@ class RpcClient extends Base {
       debug('create consumer for %s', key);
       consumerClass = consumerClass || this.consumerClass;
       consumer = new consumerClass(options);
+      this._consumerCache.set(key, consumer);
       // delegate consumer's error to client
       consumer.on('error', err => { this.emit('error', err); });
       consumer.on('request', req => { this.emit('request', req); });
       consumer.on('response', info => { this.emit('response', info); });
-      this._consumerCache.set(key, consumer);
+      consumer.once('close', () => {
+        this._consumerCache.delete(key);
+      });
     }
     return consumer;
   }

lib/client/consumer.js

@@ -168,6 +168,7 @@ class RpcConsumer extends Base {
     }
     this.removeAllListeners('request');
     this.removeAllListeners('response');
+    this.emit('close');
   }
 }
 

test/client/client.test.js

@@ -41,7 +41,7 @@ describe('test/client/client.test.js', () => {
     });
     client.consumerClass = RpcConsumer;
 
-    const consumer = client.createConsumer({
+    let consumer = client.createConsumer({
       interfaceName: 'com.alipay.sofa.rpc.test.ProtoService',
       targetAppName: 'pb',
       loadbalancerClass: 'consistentHash',
@@ -58,12 +58,21 @@ describe('test/client/client.test.js', () => {
       group: 'A',
     }];
     const ctx = { foo: 'bar' };
-    const res = await consumer.invoke('echoObj', args, { ctx });
+    let res = await consumer.invoke('echoObj', args, { ctx });
     assert.deepEqual(res, { code: 200, message: 'hello Peter, you are in A' });
 
     assert(req && req.targetAppName === 'pb');
     assert(req.ctx === ctx);
 
+    consumer.close();
+
+    consumer = client.createConsumer({
+      interfaceName: 'com.alipay.sofa.rpc.test.ProtoService',
+      targetAppName: 'pb',
+    });
+    res = await consumer.invoke('echoObj', args, { ctx });
+    assert.deepEqual(res, { code: 200, message: 'hello Peter, you are in A' });
+
     await client.close();
   });