update logic

Author: howardchung

svc/cycler.ts

@@ -5,6 +5,7 @@ import { shuffle } from "../util/utility";
 const projectId = config.GOOGLE_CLOUD_PROJECT_ID;
 const lifetime = 600;
 const template = 'retriever-20250324';
+const count = 3;
 
 async function getToken() {
   const tokenResponse = await axios.get("http://metadata.google.internal/computeMetadata/v1/instance/service-accounts/default/token", {headers: {"Metadata-Flavor": "Google"}});
@@ -13,30 +14,28 @@ async function getToken() {
 }
 
 async function cycle() {
+  const zonesResponse = await axios.get(`https://compute.googleapis.com/compute/v1/projects/${projectId}/zones`, { headers: { "Authorization": "Bearer " + await getToken() }});
+  const zones = zonesResponse.data.items.map((zone: any) => zone.name);
+  console.log(zones, zones.length);
   while (true) {
-    const zonesResponse = await axios.get(`https://compute.googleapis.com/compute/v1/projects/${projectId}/zones`, { headers: { "Authorization": "Bearer " + await getToken() }});
-    const zones = zonesResponse.data.items.map((zone: any) => zone.name);
-    console.log(zones, zones.length);
     shuffle(zones);
-    for (let i = 0; i < 3; i++) {
-        const zone = zones[i % zones.length];
-        const config = {
-            "name": "retriever-" + process.hrtime.bigint(),
-            "scheduling": {
-              "automaticRestart": false,
-              "instanceTerminationAction": "DELETE",
-              "maxRunDuration": {
-                "seconds": lifetime.toString(),
-              },
-              "onHostMaintenance": "TERMINATE",
-              "provisioningModel": "SPOT"
-            },
-            "zone": `projects/${projectId}/zones/` + zone,
-          };
-          const resp = await axios.post(`https://compute.googleapis.com/compute/v1/projects/${projectId}/zones/${zone}/instances?sourceInstanceTemplate=global/instanceTemplates/${template}`, config, { headers: { "Authorization": "Bearer " + await getToken() } });
-          console.log(resp.data);
-    }
-    await new Promise(resolve => setTimeout(resolve, (lifetime - 45) * 1000));
+    const zone = zones[0];
+    const config = {
+      "name": "retriever-" + process.hrtime.bigint(),
+      "scheduling": {
+        "automaticRestart": false,
+        "instanceTerminationAction": "DELETE",
+        "maxRunDuration": {
+          "seconds": lifetime.toString(),
+        },
+        "onHostMaintenance": "TERMINATE",
+        "provisioningModel": "SPOT"
+      },
+      "zone": `projects/${projectId}/zones/` + zone,
+    };
+    const resp = await axios.post(`https://compute.googleapis.com/compute/v1/projects/${projectId}/zones/${zone}/instances?sourceInstanceTemplate=global/instanceTemplates/${template}`, config, { headers: { "Authorization": "Bearer " + await getToken() } });
+    console.log(resp.data);
+    await new Promise(resolve => setTimeout(resolve, lifetime * 1000 / count));
   }
 }