| 498 | } |
| 499 | |
| 500 | func (s *Session) ExecuteStatementWithContext(ctx context.Context, sql string) (*SessionDataSet, error) { |
| 501 | request := rpc.TSExecuteStatementReq{ |
| 502 | SessionId: s.sessionId, |
| 503 | Statement: sql, |
| 504 | StatementId: s.requestStatementId, |
| 505 | FetchSize: &s.config.FetchSize, |
| 506 | } |
| 507 | resp, err := s.client.ExecuteStatementV2(ctx, &request) |
| 508 | |
| 509 | if err != nil && resp == nil { |
| 510 | if s.reconnect() { |
| 511 | request.SessionId = s.sessionId |
| 512 | request.StatementId = s.requestStatementId |
| 513 | resp, err = s.client.ExecuteStatementV2(ctx, &request) |
| 514 | } |
| 515 | } |
| 516 | if statusErr := VerifySuccess(resp.Status); statusErr != nil { |
| 517 | return nil, statusErr |
| 518 | } |
| 519 | |
| 520 | return s.genDataSet(sql, resp) |
| 521 | } |
| 522 | |
| 523 | func (s *Session) ExecuteStatement(sql string) (*SessionDataSet, error) { |
| 524 | return s.ExecuteStatementWithContext(context.Background(), sql) |