[ASI-836] [CN] Add reconfigNodewhitelist envvar & consume in reconfig (…

…#2447)

creator-node/src/config.js

@@ -617,6 +617,12 @@ const config = convict({
     env: 'openRestyCacheCIDEnabled',
     default: false
   },
+  reconfigNodeWhitelist: {
+    doc: 'Comma separated string - list of Content Nodes to select from for reconfig. Empty string = whitelist all.',
+    format: String,
+    env: 'reconfigNodeWhitelist',
+    default: ''
+  },
   minimumTranscodingSlotsAvailable: {
     doc: 'The minimum number of slots needed to be available for TranscodingQueue to accept more jobs',
     format: 'nat',

creator-node/src/snapbackSM/snapbackSM.js

@@ -118,6 +118,11 @@ class SnapbackSM {
       'maxSyncMonitoringDurationInMs'
     )
 
+    const reconfigNodeWhitelist = this.nodeConfig.get('reconfigNodeWhitelist')
+    this.reconfigNodeWhitelist = reconfigNodeWhitelist
+      ? new Set(reconfigNodeWhitelist.split(','))
+      : null
+
     // Throw an error if no libs are provided
     if (
       !this.audiusLibs ||
@@ -1040,7 +1045,8 @@ class SnapbackSM {
         const { services: healthyServicesMap } =
           await this.audiusLibs.ServiceProvider.autoSelectCreatorNodes({
             performSyncCheck: false,
-            log: false
+            whitelist: this.reconfigNodeWhitelist,
+            log: true
           })
 
         const healthyNodes = Object.keys(healthyServicesMap)

libs/src/services/creatorNode/CreatorNodeSelection.test.ts

@@ -107,6 +107,43 @@ describe('test CreatorNodeSelection', () => {
     )
   })
 
+  it('Return nothing if whitelist is empty set', async function () {
+    const healthy = 'https://healthy.audius.co'
+    nock(healthy)
+      .get('/health_check/verbose')
+      .reply(200, { data: defaultHealthCheckData })
+
+    const healthyButSlow = 'https://healthybutslow.audius.co'
+    nock(healthyButSlow)
+      .get('/health_check/verbose')
+      .delay(100)
+      .reply(200, { data: defaultHealthCheckData })
+
+    const healthyButSlowest = 'https://healthybutslowest.audius.co'
+    nock(healthyButSlowest)
+      .get('/health_check/verbose')
+      .delay(200)
+      .reply(200, { data: defaultHealthCheckData })
+
+    const cns = new CreatorNodeSelection({
+      creatorNode: mockCreatorNode,
+      numberOfNodes: 3,
+      ethContracts: mockEthContracts(
+        [healthy, healthyButSlow, healthyButSlowest],
+        '1.2.3'
+      ),
+      whitelist: new Set([]),
+      blacklist: null
+    })
+
+    const { primary, secondaries, services } = await cns.select()
+    assert(primary === undefined)
+    assert(secondaries.length === 0)
+
+    const returnedHealthyServices = new Set(Object.keys(services))
+    assert(returnedHealthyServices.size === 0)
+  })
+
   it('select healthy nodes as the primary and secondary, and do not select unhealthy nodes', async () => {
     const upToDate = 'https://upToDate.audius.co'
     nock(upToDate)
@@ -183,7 +220,7 @@ describe('test CreatorNodeSelection', () => {
     const { primary, secondaries, services } = await cns.select(true, false)
 
     // All unhealthy are bad candidates so don't select anything
-    assert(!primary)
+    assert(primary === undefined)
     assert(secondaries.length === 0)
 
     const returnedHealthyServices = new Set(Object.keys(services))