fix(aliyun):DescribeInvocationResults need to wait for the results

404tk · 404tk · commit f5d189b1a7d8 · 2023-09-26T14:45:52.000+08:00
diff --git a/pkg/providers/alibaba/ecs/exec.go b/pkg/providers/alibaba/ecs/exec.go
@@ -1,6 +1,8 @@
 package ecs
 
 import (
+	"time"
+
 	"github.com/404tk/cloudtoolkit/pkg/schema"
 	"github.com/404tk/cloudtoolkit/utils/logger"
 	"github.com/aliyun/alibaba-cloud-sdk-go/services/ecs"
@@ -33,13 +35,33 @@ func RunCommand(client *ecs.Client, instanceId, region, ostype, cmd string) stri
 }
 
 func describeInvocationResults(client *ecs.Client, region, cid string) string {
-	request := ecs.CreateDescribeInvocationResultsRequest()
-	request.Scheme = "https"
-	request.RegionId = region
-	request.CommandId = cid
-	response, err := client.DescribeInvocationResults(request)
-	if err != nil {
-		return err.Error()
+	t := 0
+	for {
+		time.Sleep(1 * time.Second)
+		t += 1
+		request := ecs.CreateDescribeInvocationResultsRequest()
+		request.Scheme = "https"
+		request.RegionId = region
+		request.CommandId = cid
+		request.ContentEncoding = "PlainText"
+		response, err := client.DescribeInvocationResults(request)
+		if err != nil {
+			logger.Error(err)
+			return ""
+		}
+		status := response.Invocation.InvocationResults.InvocationResult[0].InvokeRecordStatus
+		switch status {
+		case "Running":
+			if t < 5 {
+				continue
+			}
+			logger.Error("Timeout: Wait 5s by default.")
+			return ""
+		case "Finished":
+			return response.Invocation.InvocationResults.InvocationResult[0].Output
+		default:
+			logger.Error("Exception status: " + status)
+			return ""
+		}
 	}
-	return response.Invocation.InvocationResults.InvocationResult[0].Output
 }
