add redis key for feature server and integration test

Author: xquanluu

lib/cli/feature-server-config.js

@@ -1,5 +1,4 @@
 const runtimeConfigModule = require('./runtime-config');
-const runtimeConfig = runtimeConfigModule.getInstance();
 
 function isValidIP(ip) {
   if (!ip || typeof ip !== 'string') return false;
@@ -23,90 +22,60 @@ function isValidIP(ip) {
   return false;
 }
 
-async function getDrainedServers() {
-  const drained = await runtimeConfig.get('drainedFeatureServers');
-
-  if (!drained) return [];
-
-  if (typeof drained === 'string') {
-    return drained.split(',').map((s) => s.trim()).filter((s) => s.length > 0);
-  }
-
-  if (Array.isArray(drained)) return drained;
-
-  return [];
-}
-
 async function isDrained(serverIP) {
-  const drained = await getDrainedServers();
-  return drained.includes(serverIP);
-}
-
-async function drainServer(serverIP) {
   if (!isValidIP(serverIP)) {
-    return {
-      added: false,
-      array: await getDrainedServers(),
-      error: `Invalid IP: ${serverIP}`
-    };
+    return false;
   }
 
-  return await runtimeConfig.addToArray('drainedFeatureServers', serverIP);
+  try {
+    const runtimeConfig = runtimeConfigModule.getInstance();
+    return await runtimeConfig.isServerDrained(serverIP);
+  } catch {
+    return false;
+  }
 }
 
-async function undrainServer(serverIP) {
-  if (!isValidIP(serverIP)) {
-    return {
-      removed: false,
-      array: await getDrainedServers(),
-      error: `Invalid IP: ${serverIP}`
-    };
+async function getDrainedServers() {
+  try {
+    const runtimeConfig = runtimeConfigModule.getInstance();
+    return await runtimeConfig.getDrainedFeatureServers();
+  } catch {
+    return [];
   }
-
-  return await runtimeConfig.removeFromArray('drainedFeatureServers', serverIP);
 }
 
-async function setDrainedServers(servers) {
-  let serverList = [];
-
-  if (typeof servers === 'string') {
-    serverList = servers.split(',').map((s) => s.trim()).filter((s) => s.length > 0);
-  } else if (Array.isArray(servers)) {
-    serverList = servers;
-  } else {
-    return {
-      key: 'drainedFeatureServers',
-      value: await getDrainedServers(),
-      error: 'Invalid format'
-    };
+async function getActiveServers() {
+  try {
+    const runtimeConfig = runtimeConfigModule.getInstance();
+    return await runtimeConfig.getActiveFeatureServers();
+  } catch {
+    return [];
   }
+}
 
-  // Check all IPs are valid
-  const badIPs = serverList.filter((ip) => !isValidIP(ip));
-  if (badIPs.length > 0) {
-    return {
-      key: 'drainedFeatureServers',
-      value: await getDrainedServers(),
-      error: `Invalid IPs: ${badIPs.join(', ')}`
-    };
+async function getAvailableServers() {
+  try {
+    const runtimeConfig = runtimeConfigModule.getInstance();
+    return await runtimeConfig.getAvailableFeatureServers();
+  } catch {
+    return [];
   }
+}
 
-  return await runtimeConfig.set('drainedFeatureServers', serverList);
+async function getAllServersWithStatus() {
+  try {
+    const runtimeConfig = runtimeConfigModule.getInstance();
+    return await runtimeConfig.getAllFeatureServersWithStatus();
+  } catch {
+    return { servers: [], drained: [] };
+  }
 }
 
 module.exports = {
-  getDrainedServers,
-  isDrained,
-  drainServer,
-  undrainServer,
-  setDrainedServers,
   isValidIP,
-  runtimeConfig,
-
-  // Legacy names for backward compatibility
-  getDrainedFeatureServers: getDrainedServers,
-  isFeatureServerDrained: isDrained,
-  drainFeatureServer: drainServer,
-  undrainFeatureServer: undrainServer,
-  setDrainedFeatureServers: setDrainedServers
+  isDrained,
+  getDrainedServers,
+  getActiveServers,
+  getAvailableServers,
+  getAllServersWithStatus
 };