feat(consul support): consul instance store (#163)

* feat(consul support): consul instance store

add context to getInstanceGroup call

* adds intial consul support and unit test stub

* mock consul k/v client

* further fixes

* updates for consul group store

* updates for ctx passthrough

package.json

@@ -15,6 +15,7 @@
     "@types/shortid": "0.0.31",
     "@types/sshpk": "^1.17.3",
     "bee-queue": "^1.6.0",
+    "consul": "^2.0.1",
     "dotenv": "^16.3.1",
     "dots-wrapper": "^3.11.3",
     "envalid": "^8.0.0",

src/app.ts

@@ -447,7 +447,7 @@ app.put(
     body('maxDesired').optional().isInt({ min: 0 }).withMessage('Value must be positive'),
     body('desiredCount').optional().isInt({ min: 0 }).withMessage('Value must be positive'),
     body().custom(async (value, { req }) => {
-        if (!(await validator.groupHasValidDesiredInput(req.params.name, value))) {
+        if (!(await validator.groupHasValidDesiredInput(req.context, req.params.name, value))) {
             throw new Error('Desired count must be between min and max; min cannot be grater than max');
         }
         return true;

src/autoscaler.ts

@@ -46,7 +46,7 @@ export default class AutoscaleProcessor {
         }
 
         try {
-            const group = await this.instanceGroupManager.getInstanceGroup(groupName);
+            const group = await this.instanceGroupManager.getInstanceGroup(ctx, groupName);
             if (!group) {
                 throw new Error(`Group ${groupName} not found, failed to process autoscaling`);
             }

src/cloud_manager.ts

@@ -64,7 +64,12 @@ export default class CloudManager {
                 };
                 await this.instanceTracker.track(ctx, state);
                 if (isScaleDownProtected) {
-                    await this.shutdownManager.setScaleDownProtected(ctx, instanceId, group.protectedTTLSec);
+                    await this.shutdownManager.setScaleDownProtected(
+                        ctx,
+                        group.name,
+                        instanceId,
+                        group.protectedTTLSec,
+                    );
                     ctx.logger.info(
                         `[CloudManager] Instance ${instanceId} from group ${group.name} is in protected mode`,
                     );