Ver código fonte

wait on queued state

AjayTripathy 6 anos atrás
pai
commit
40b08cc9ca
1 arquivos alterados com 3 adições e 3 exclusões
  1. 3 3
      pkg/cloud/awsprovider.go

+ 3 - 3
pkg/cloud/awsprovider.go

@@ -1130,7 +1130,7 @@ func (a *AWS) QueryAthenaBillingData(query string) (*athena.GetQueryResultsOutpu
 		if err != nil {
 		if err != nil {
 			return nil, err
 			return nil, err
 		}
 		}
-		if *qrop.QueryExecution.Status.State != "RUNNING" {
+		if *qrop.QueryExecution.Status.State != "RUNNING" && *qrop.QueryExecution.Status.State != "QUEUED" {
 			break
 			break
 		}
 		}
 		time.Sleep(duration)
 		time.Sleep(duration)
@@ -1313,7 +1313,7 @@ func (a *AWS) ExternalAllocations(start string, end string, aggregators []string
 		if err != nil {
 		if err != nil {
 			return nil, err
 			return nil, err
 		}
 		}
-		if *qrop.QueryExecution.Status.State != "RUNNING" {
+		if *qrop.QueryExecution.Status.State != "RUNNING" && *qrop.QueryExecution.Status.State != "QUEUED" {
 			break
 			break
 		}
 		}
 		time.Sleep(duration)
 		time.Sleep(duration)
@@ -1439,7 +1439,7 @@ func (a *AWS) QuerySQL(query string) ([]byte, error) {
 		if err != nil {
 		if err != nil {
 			return nil, err
 			return nil, err
 		}
 		}
-		if *qrop.QueryExecution.Status.State != "RUNNING" {
+		if *qrop.QueryExecution.Status.State != "RUNNING" && *qrop.QueryExecution.Status.State != "QUEUED" {
 			break
 			break
 		}
 		}
 		time.Sleep(duration)
 		time.Sleep(duration)