feat: do loadBalancer close when address_group close

sofastack · Apr 12, 2022 · a88bf21 · a88bf21
1 parent 512634e
commit a88bf21
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/lib/client/address_group.js b/lib/client/address_group.js
@@ -360,6 +360,9 @@ class AddressGroup extends Base {
     if (this._healthCounter) {
       this._healthCounter.close();
     }
+    if (this._loadbalancer && this._loadbalancer.close) {
+      this._loadbalancer.close();
+    }
   }
 }
 

diff --git a/lib/client/loadbalancer/base.js b/lib/client/loadbalancer/base.js
@@ -149,6 +149,10 @@ class LoadBalancer extends Base {
   getWeight(address) {
     return this.addressGroup.getWeight(address);
   }
+
+  close() {
+    // do nothing here, for override
+  }
 }
 
 module.exports = LoadBalancer;