Chilkat Online Tools

BatchGetQueryExecution Ruby Example

Athena

require 'chilkat'

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

rest = Chilkat::CkRest.new()

authAws = Chilkat::CkAuthAws.new()
authAws.put_AccessKey("AWS_ACCESS_KEY")
authAws.put_SecretKey("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.)
authAws.put_Region("us-west-2")
authAws.put_ServiceName("athena")
# SetAuthAws causes Chilkat to automatically add the following headers: Authorization, X-Amz-Date
rest.SetAuthAws(authAws)

# URL: https://athena.us-west-2.amazonaws.com/
# Use the same region as specified above.
success = rest.Connect("athena.us-west-2.amazonaws.com",443,true,true)
if (success != true)
    print "ConnectFailReason: " + rest.get_ConnectFailReason().to_s() + "\n";
    print rest.lastErrorText() + "\n";
    exit
end

# 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 = Chilkat::CkJsonObject.new()
json.UpdateString("QueryExecutionIds[0]","string")

# The JSON request body created by the above code:

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

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

sbRequestBody = Chilkat::CkStringBuilder.new()
json.EmitSb(sbRequestBody)
sbResponseBody = Chilkat::CkStringBuilder.new()
success = rest.FullRequestSb("POST","/",sbRequestBody,sbResponseBody)
if (success != true)
    print rest.lastErrorText() + "\n";
    exit
end

respStatusCode = rest.get_ResponseStatusCode()
print "response status code = " + respStatusCode.to_s() + "\n";
if (respStatusCode != 200)
    print "Response Header:" + "\n";
    print rest.responseHeader() + "\n";
    print "Response Body:" + "\n";
    print sbResponseBody.getAsString() + "\n";
    exit
end

jResp = Chilkat::CkJsonObject.new()
jResp.LoadSb(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

i = 0
count_i = jResp.SizeOfArray("QueryExecutions")
while i < count_i
    jResp.put_I(i)
    EffectiveEngineVersion = jResp.stringOf("QueryExecutions[i].EngineVersion.EffectiveEngineVersion")
    SelectedEngineVersion = jResp.stringOf("QueryExecutions[i].EngineVersion.SelectedEngineVersion")
    Query = jResp.stringOf("QueryExecutions[i].Query")
    Catalog = jResp.stringOf("QueryExecutions[i].QueryExecutionContext.Catalog")
    Database = jResp.stringOf("QueryExecutions[i].QueryExecutionContext.Database")
    QueryExecutionId = jResp.stringOf("QueryExecutions[i].QueryExecutionId")
    S3AclOption = jResp.stringOf("QueryExecutions[i].ResultConfiguration.AclConfiguration.S3AclOption")
    EncryptionOption = jResp.stringOf("QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.EncryptionOption")
    KmsKey = jResp.stringOf("QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.KmsKey")
    ExpectedBucketOwner = jResp.stringOf("QueryExecutions[i].ResultConfiguration.ExpectedBucketOwner")
    OutputLocation = jResp.stringOf("QueryExecutions[i].ResultConfiguration.OutputLocation")
    StatementType = jResp.stringOf("QueryExecutions[i].StatementType")
    DataManifestLocation = jResp.stringOf("QueryExecutions[i].Statistics.DataManifestLocation")
    DataScannedInBytes = jResp.IntOf("QueryExecutions[i].Statistics.DataScannedInBytes")
    EngineExecutionTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.EngineExecutionTimeInMillis")
    QueryPlanningTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.QueryPlanningTimeInMillis")
    QueryQueueTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.QueryQueueTimeInMillis")
    ServiceProcessingTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.ServiceProcessingTimeInMillis")
    TotalExecutionTimeInMillis = jResp.IntOf("QueryExecutions[i].Statistics.TotalExecutionTimeInMillis")
    ErrorCategory = jResp.IntOf("QueryExecutions[i].Status.AthenaError.ErrorCategory")
    ErrorMessage = jResp.stringOf("QueryExecutions[i].Status.AthenaError.ErrorMessage")
    ErrorType = jResp.IntOf("QueryExecutions[i].Status.AthenaError.ErrorType")
    Retryable = jResp.IntOf("QueryExecutions[i].Status.AthenaError.Retryable")
    CompletionDateTime = jResp.IntOf("QueryExecutions[i].Status.CompletionDateTime")
    State = jResp.stringOf("QueryExecutions[i].Status.State")
    StateChangeReason = jResp.stringOf("QueryExecutions[i].Status.StateChangeReason")
    SubmissionDateTime = jResp.IntOf("QueryExecutions[i].Status.SubmissionDateTime")
    WorkGroup = jResp.stringOf("QueryExecutions[i].WorkGroup")
    i = i + 1
end
i = 0
count_i = jResp.SizeOfArray("UnprocessedQueryExecutionIds")
while i < count_i
    jResp.put_I(i)
    ErrorCode = jResp.stringOf("UnprocessedQueryExecutionIds[i].ErrorCode")
    ErrorMessage = jResp.stringOf("UnprocessedQueryExecutionIds[i].ErrorMessage")
    QueryExecutionId = jResp.stringOf("UnprocessedQueryExecutionIds[i].QueryExecutionId")
    i = i + 1
end

# 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"
#     }
#   ]
# }