Chilkat Online Tools

BatchGetQueryExecution PureBasic Example

Athena

IncludeFile "CkJsonObject.pb"
IncludeFile "CkStringBuilder.pb"
IncludeFile "CkRest.pb"
IncludeFile "CkAuthAws.pb"

Procedure ChilkatExample()

    ; This example requires the Chilkat API to have been previously unlocked.
    ; See Global Unlock Sample for sample code.

    rest.i = CkRest::ckCreate()
    If rest.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success.i

    authAws.i = CkAuthAws::ckCreate()
    If authAws.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkAuthAws::setCkAccessKey(authAws, "AWS_ACCESS_KEY")
    CkAuthAws::setCkSecretKey(authAws, "AWS_SECRET_KEY")

    ; Don't forget to change the region to your particular region. (Also make the same change in the call to Connect below.)
    CkAuthAws::setCkRegion(authAws, "us-west-2")
    CkAuthAws::setCkServiceName(authAws, "athena")
    ; SetAuthAws causes Chilkat to automatically add the following headers: Authorization, X-Amz-Date
    CkRest::ckSetAuthAws(rest,authAws)

    ; URL: https://athena.us-west-2.amazonaws.com/
    ; Use the same region as specified above.
    success = CkRest::ckConnect(rest,"athena.us-west-2.amazonaws.com",443,1,1)
    If success <> 1
        Debug "ConnectFailReason: " + Str(CkRest::ckConnectFailReason(rest))
        Debug CkRest::ckLastErrorText(rest)
        CkRest::ckDispose(rest)
        CkAuthAws::ckDispose(authAws)
        ProcedureReturn
    EndIf

    ; The following code creates the JSON request body.
    ; The JSON created by this code is shown below.

    ; Use this online tool to generate code from sample JSON:
    ; Generate Code to Create JSON

    json.i = CkJsonObject::ckCreate()
    If json.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkJsonObject::ckUpdateString(json,"QueryExecutionIds[0]","string")

    ; The JSON request body created by the above code:

    ; {
    ;   "QueryExecutionIds": [
    ;     "string"
    ;   ]
    ; }

    CkRest::ckAddHeader(rest,"Content-Type","application/x-amz-json-1.1")
    CkRest::ckAddHeader(rest,"X-Amz-Target","AmazonAthena.BatchGetQueryExecution")

    sbRequestBody.i = CkStringBuilder::ckCreate()
    If sbRequestBody.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkJsonObject::ckEmitSb(json,sbRequestBody)
    sbResponseBody.i = CkStringBuilder::ckCreate()
    If sbResponseBody.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    success = CkRest::ckFullRequestSb(rest,"POST","/",sbRequestBody,sbResponseBody)
    If success <> 1
        Debug CkRest::ckLastErrorText(rest)
        CkRest::ckDispose(rest)
        CkAuthAws::ckDispose(authAws)
        CkJsonObject::ckDispose(json)
        CkStringBuilder::ckDispose(sbRequestBody)
        CkStringBuilder::ckDispose(sbResponseBody)
        ProcedureReturn
    EndIf

    respStatusCode.i = CkRest::ckResponseStatusCode(rest)
    Debug "response status code = " + Str(respStatusCode)
    If respStatusCode <> 200
        Debug "Response Header:"
        Debug CkRest::ckResponseHeader(rest)
        Debug "Response Body:"
        Debug CkStringBuilder::ckGetAsString(sbResponseBody)
        CkRest::ckDispose(rest)
        CkAuthAws::ckDispose(authAws)
        CkJsonObject::ckDispose(json)
        CkStringBuilder::ckDispose(sbRequestBody)
        CkStringBuilder::ckDispose(sbResponseBody)
        ProcedureReturn
    EndIf

    jResp.i = CkJsonObject::ckCreate()
    If jResp.i = 0
        Debug "Failed to create object."
        ProcedureReturn
    EndIf

    CkJsonObject::ckLoadSb(jResp,sbResponseBody)

    ; The following code parses the JSON response.
    ; A sample JSON response is shown below the sample code.

    ; Use this online tool to generate parsing code from sample JSON:
    ; Generate Parsing Code from JSON

    EffectiveEngineVersion.s
    SelectedEngineVersion.s
    Query.s
    Catalog.s
    Database.s
    QueryExecutionId.s
    S3AclOption.s
    EncryptionOption.s
    KmsKey.s
    ExpectedBucketOwner.s
    OutputLocation.s
    StatementType.s
    DataManifestLocation.s
    DataScannedInBytes.i
    EngineExecutionTimeInMillis.i
    QueryPlanningTimeInMillis.i
    QueryQueueTimeInMillis.i
    ServiceProcessingTimeInMillis.i
    TotalExecutionTimeInMillis.i
    ErrorCategory.i
    ErrorMessage.s
    ErrorType.i
    Retryable.i
    CompletionDateTime.i
    State.s
    StateChangeReason.s
    SubmissionDateTime.i
    WorkGroup.s
    ErrorCode.s

    i.i = 0
    count_i.i = CkJsonObject::ckSizeOfArray(jResp,"QueryExecutions")
    While i < count_i
        CkJsonObject::setCkI(jResp, i)
        EffectiveEngineVersion = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].EngineVersion.EffectiveEngineVersion")
        SelectedEngineVersion = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].EngineVersion.SelectedEngineVersion")
        Query = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].Query")
        Catalog = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].QueryExecutionContext.Catalog")
        Database = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].QueryExecutionContext.Database")
        QueryExecutionId = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].QueryExecutionId")
        S3AclOption = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].ResultConfiguration.AclConfiguration.S3AclOption")
        EncryptionOption = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.EncryptionOption")
        KmsKey = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.KmsKey")
        ExpectedBucketOwner = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].ResultConfiguration.ExpectedBucketOwner")
        OutputLocation = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].ResultConfiguration.OutputLocation")
        StatementType = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].StatementType")
        DataManifestLocation = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].Statistics.DataManifestLocation")
        DataScannedInBytes = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.DataScannedInBytes")
        EngineExecutionTimeInMillis = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.EngineExecutionTimeInMillis")
        QueryPlanningTimeInMillis = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.QueryPlanningTimeInMillis")
        QueryQueueTimeInMillis = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.QueryQueueTimeInMillis")
        ServiceProcessingTimeInMillis = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.ServiceProcessingTimeInMillis")
        TotalExecutionTimeInMillis = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Statistics.TotalExecutionTimeInMillis")
        ErrorCategory = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Status.AthenaError.ErrorCategory")
        ErrorMessage = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].Status.AthenaError.ErrorMessage")
        ErrorType = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Status.AthenaError.ErrorType")
        Retryable = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Status.AthenaError.Retryable")
        CompletionDateTime = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Status.CompletionDateTime")
        State = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].Status.State")
        StateChangeReason = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].Status.StateChangeReason")
        SubmissionDateTime = CkJsonObject::ckIntOf(jResp,"QueryExecutions[i].Status.SubmissionDateTime")
        WorkGroup = CkJsonObject::ckStringOf(jResp,"QueryExecutions[i].WorkGroup")
        i = i + 1
    Wend
    i = 0
    count_i = CkJsonObject::ckSizeOfArray(jResp,"UnprocessedQueryExecutionIds")
    While i < count_i
        CkJsonObject::setCkI(jResp, i)
        ErrorCode = CkJsonObject::ckStringOf(jResp,"UnprocessedQueryExecutionIds[i].ErrorCode")
        ErrorMessage = CkJsonObject::ckStringOf(jResp,"UnprocessedQueryExecutionIds[i].ErrorMessage")
        QueryExecutionId = CkJsonObject::ckStringOf(jResp,"UnprocessedQueryExecutionIds[i].QueryExecutionId")
        i = i + 1
    Wend

    ; A sample JSON response body parsed by the above code:

    ; {
    ;   "QueryExecutions": [
    ;     {
    ;       "EngineVersion": {
    ;         "EffectiveEngineVersion": "string",
    ;         "SelectedEngineVersion": "string"
    ;       },
    ;       "Query": "string",
    ;       "QueryExecutionContext": {
    ;         "Catalog": "string",
    ;         "Database": "string"
    ;       },
    ;       "QueryExecutionId": "string",
    ;       "ResultConfiguration": {
    ;         "AclConfiguration": {
    ;           "S3AclOption": "string"
    ;         },
    ;         "EncryptionConfiguration": {
    ;           "EncryptionOption": "string",
    ;           "KmsKey": "string"
    ;         },
    ;         "ExpectedBucketOwner": "string",
    ;         "OutputLocation": "string"
    ;       },
    ;       "StatementType": "string",
    ;       "Statistics": {
    ;         "DataManifestLocation": "string",
    ;         "DataScannedInBytes": number,
    ;         "EngineExecutionTimeInMillis": number,
    ;         "QueryPlanningTimeInMillis": number,
    ;         "QueryQueueTimeInMillis": number,
    ;         "ServiceProcessingTimeInMillis": number,
    ;         "TotalExecutionTimeInMillis": number
    ;       },
    ;       "Status": {
    ;         "AthenaError": {
    ;           "ErrorCategory": number,
    ;           "ErrorMessage": "string",
    ;           "ErrorType": number,
    ;           "Retryable": boolean
    ;         },
    ;         "CompletionDateTime": number,
    ;         "State": "string",
    ;         "StateChangeReason": "string",
    ;         "SubmissionDateTime": number
    ;       },
    ;       "WorkGroup": "string"
    ;     }
    ;   ],
    ;   "UnprocessedQueryExecutionIds": [
    ;     {
    ;       "ErrorCode": "string",
    ;       "ErrorMessage": "string",
    ;       "QueryExecutionId": "string"
    ;     }
    ;   ]
    ; }


    CkRest::ckDispose(rest)
    CkAuthAws::ckDispose(authAws)
    CkJsonObject::ckDispose(json)
    CkStringBuilder::ckDispose(sbRequestBody)
    CkStringBuilder::ckDispose(sbResponseBody)
    CkJsonObject::ckDispose(jResp)


    ProcedureReturn
EndProcedure