Automatically publish backend address to the pool

src/io/backend/iobackend.js

@@ -1,12 +1,20 @@
 import Server from 'cytube-common/lib/proxy/server';
 import FrontendManager from './frontendmanager';
+import uuid from 'uuid';
+import PoolEntryUpdater from 'cytube-common/lib/redis/poolentryupdater';
+import JSONProtocol from 'cytube-common/lib/proxy/protocol';
+
+const BACKEND_POOL = 'backend-hosts';
 
 export default class IOBackend {
-    constructor(proxyListenerConfig, socketEmitter) {
+    constructor(proxyListenerConfig, socketEmitter, poolRedisClient) {
         this.proxyListenerConfig = proxyListenerConfig;
         this.socketEmitter = socketEmitter;
+        this.poolRedisClient = poolRedisClient;
+        this.protocol = new JSONProtocol();
         this.initFrontendManager();
         this.initProxyListener();
+        this.initBackendPoolUpdater();
     }
 
     initFrontendManager() {
@@ -14,8 +22,21 @@ export default class IOBackend {
     }
 
     initProxyListener() {
-        this.proxyListener = new Server(this.proxyListenerConfig);
+        this.proxyListener = new Server(this.proxyListenerConfig, this.protocol);
         this.proxyListener.on('connection',
                 this.frontendManager.onConnection.bind(this.frontendManager));
     }
+
+    initBackendPoolUpdater() {
+        const entry = {
+            address: this.proxyListenerConfig.getHost() + '/' + this.proxyListenerConfig.getPort()
+        }
+        this.poolEntryUpdater = new PoolEntryUpdater(
+                this.poolRedisClient,
+                BACKEND_POOL,
+                uuid.v4(),
+                entry
+        );
+        this.poolEntryUpdater.start();
+    }
 }

src/server.js

@@ -140,7 +140,10 @@ var Server = function () {
             return '127.0.0.1';
         }
     };
-    const backend = new IOBackend(listenerConfig, sioEmitter);
+    const redis = require('redis');
+    Promise.promisifyAll(redis.RedisClient.prototype);
+    Promise.promisifyAll(redis.Multi.prototype);
+    const backend = new IOBackend(listenerConfig, sioEmitter, redis.createClient());
 
     // background tasks init ----------------------------------------------
     require("./bgtask")(self);