Support STS token and retry

Signed-off-by: denverdino <[email protected]>
aliyun · Nov 29, 2022 · d2631e5 · d2631e5
1 parent 5f33ef1
commit d2631e5
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 5 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/index.js b/src/index.js
@@ -21,7 +21,7 @@ async function run() {
             endpoint: APIEndpoint,
             apiVersion: '2015-12-15'
         });
-        let result = await client.request('GET', `/k8s/${clusterId}/user_config`)
+        let result = await requestWithRetry(client, 'GET', `/k8s/${clusterId}/user_config`)
         let kubeconfig = result.config
         const runnerTempDirectory = process.env['RUNNER_TEMP']; // Using process.env until the core libs are updated
         const kubeconfigPath = path.join(runnerTempDirectory, `kubeconfig_${Date.now()}`);
@@ -34,4 +34,19 @@ async function run() {
         core.setFailed(`Failed to get kubeconfig file for Kubernetes cluster: ${err}`);
     }
 }
+
+async function requestWithRetry(client, method, path, retries = 3, retryDelay = 1000) {
+	try {
+		return await client.request(method, path);
+	} catch (err) {
+		if (retries > 0) {
+			core.info(`Retrying after ${retryDelay}ms...`);
+			await new Promise(resolve => setTimeout(resolve, retryDelay));
+			return await requestWithRetry(client, method, path, retries - 1, retryDelay * 2);
+		} else {
+			throw err;
+		}
+	}
+}
+
 run();