我有一個運行 Athena 查詢的工作代碼，并通過使用以下代碼輪詢error返回來等待查詢完成：GetQueryResultsfunc GetQueryResults(client *athena.Client, QueryID *string) []types.Row { params := &athena.GetQueryResultsInput{ QueryExecutionId: QueryID, } data, err := client.GetQueryResults(context.TODO(), params) for err != nil { println(err.Error()) time.Sleep(time.Second) data, err = client.GetQueryResults(context.TODO(), params) } return data.ResultSet.Rows}問題是萬一查詢失敗，我絕對沒有辦法打破循環。例如，在 Python 中我可以這樣做： while athena.get_query_execution(QueryExecutionId=execution_id)["QueryExecution"][ "Status" ]["State"] in ["RUNNING", "QUEUED"]: sleep(2)strings.Contains(err.Error(),"FAILED")我可以像在循環內部一樣進行檢查for，但我正在尋找一種更簡潔的方法。我嘗試尋找 Go 的等價物，但沒有成功。Go SDK 有返回執行狀態的函數嗎？有沒有更好的方法來進一步檢查 Go 中的錯誤而不是err != nil？
1 回答
九州編程
TA貢獻1785條經驗 獲得超4個贊
SDK已經提供了重試。
這是一個使用 aws-sdk-go-v2 的示例。
package main
import (
"context"
"fmt"
"time"
"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/service/athena"
"github.com/aws/aws-sdk-go-v2/service/athena/types"
)
func main() {
cfg := aws.NewConfig()
ath := athena.NewFromConfig(*cfg)
ctx, cancelFunc := context.WithTimeout(context.Background(), time.Minute*5)
defer cancelFunc()
rows, err := GetQueryResults(ctx, ath, aws.String("query-id"), 10)
if err != nil {
panic(err) // TODO: handle error
}
fmt.Println(rows)
}
func GetQueryResults(ctx context.Context, client *athena.Client, QueryID *string, attempts int) ([]types.Row, error) {
t := time.NewTicker(time.Second * 5)
defer t.Stop()
attemptsFunc := func(o *athena.Options) { o.RetryMaxAttempts = attempts }
WAIT:
for {
select {
case <-t.C:
out, err := client.GetQueryExecution(ctx, &athena.GetQueryExecutionInput{
QueryExecutionId: QueryID,
}, attemptsFunc)
if err != nil {
return nil, err
}
switch out.QueryExecution.Status.State {
case types.QueryExecutionStateCancelled,
types.QueryExecutionStateFailed,
types.QueryExecutionStateSucceeded:
break WAIT
}
case <-ctx.Done():
break WAIT
}
}
data, err := client.GetQueryResults(ctx, &athena.GetQueryResultsInput{
QueryExecutionId: QueryID,
})
if err != nil {
return nil, err
}
return data.ResultSet.Rows, nil
}
- 1 回答
- 0 關注
- 204 瀏覽
添加回答
舉報
0/150
提交
取消