Chilkat Online Tools

BatchGetQueryExecution SQL Server Example

Athena

CREATE PROCEDURE ChilkatSample
AS
BEGIN
    DECLARE @hr int
    DECLARE @iTmp0 int
    DECLARE @sTmp0 nvarchar(max)
    -- This example requires the Chilkat API to have been previously unlocked.
    -- See Global Unlock Sample for sample code.

    DECLARE @rest int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.Rest', @rest OUT
    IF @hr <> 0
    BEGIN
        PRINT 'Failed to create ActiveX component'
        RETURN
    END

    DECLARE @success int

    DECLARE @authAws int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.AuthAws', @authAws OUT

    EXEC sp_OASetProperty @authAws, 'AccessKey', 'AWS_ACCESS_KEY'
    EXEC sp_OASetProperty @authAws, '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.)
    EXEC sp_OASetProperty @authAws, 'Region', 'us-west-2'
    EXEC sp_OASetProperty @authAws, 'ServiceName', 'athena'
    -- SetAuthAws causes Chilkat to automatically add the following headers: Authorization, X-Amz-Date
    EXEC sp_OAMethod @rest, 'SetAuthAws', @success OUT, @authAws

    -- URL: https://athena.us-west-2.amazonaws.com/
    -- Use the same region as specified above.
    EXEC sp_OAMethod @rest, 'Connect', @success OUT, 'athena.us-west-2.amazonaws.com', 443, 1, 1
    IF @success <> 1
      BEGIN

        EXEC sp_OAGetProperty @rest, 'ConnectFailReason', @iTmp0 OUT
        PRINT 'ConnectFailReason: ' + @iTmp0
        EXEC sp_OAGetProperty @rest, 'LastErrorText', @sTmp0 OUT
        PRINT @sTmp0
        EXEC @hr = sp_OADestroy @rest
        EXEC @hr = sp_OADestroy @authAws
        RETURN
      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

    DECLARE @json int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.JsonObject', @json OUT

    EXEC sp_OAMethod @json, 'UpdateString', @success OUT, 'QueryExecutionIds[0]', 'string'

    -- The JSON request body created by the above code:

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

    EXEC sp_OAMethod @rest, 'AddHeader', @success OUT, 'Content-Type', 'application/x-amz-json-1.1'
    EXEC sp_OAMethod @rest, 'AddHeader', @success OUT, 'X-Amz-Target', 'AmazonAthena.BatchGetQueryExecution'

    DECLARE @sbRequestBody int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.StringBuilder', @sbRequestBody OUT

    EXEC sp_OAMethod @json, 'EmitSb', @success OUT, @sbRequestBody
    DECLARE @sbResponseBody int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.StringBuilder', @sbResponseBody OUT

    EXEC sp_OAMethod @rest, 'FullRequestSb', @success OUT, 'POST', '/', @sbRequestBody, @sbResponseBody
    IF @success <> 1
      BEGIN
        EXEC sp_OAGetProperty @rest, 'LastErrorText', @sTmp0 OUT
        PRINT @sTmp0
        EXEC @hr = sp_OADestroy @rest
        EXEC @hr = sp_OADestroy @authAws
        EXEC @hr = sp_OADestroy @json
        EXEC @hr = sp_OADestroy @sbRequestBody
        EXEC @hr = sp_OADestroy @sbResponseBody
        RETURN
      END
    DECLARE @respStatusCode int
    EXEC sp_OAGetProperty @rest, 'ResponseStatusCode', @respStatusCode OUT

    PRINT 'response status code = ' + @respStatusCode
    IF @respStatusCode <> 200
      BEGIN

        PRINT 'Response Header:'
        EXEC sp_OAGetProperty @rest, 'ResponseHeader', @sTmp0 OUT
        PRINT @sTmp0

        PRINT 'Response Body:'
        EXEC sp_OAMethod @sbResponseBody, 'GetAsString', @sTmp0 OUT
        PRINT @sTmp0
        EXEC @hr = sp_OADestroy @rest
        EXEC @hr = sp_OADestroy @authAws
        EXEC @hr = sp_OADestroy @json
        EXEC @hr = sp_OADestroy @sbRequestBody
        EXEC @hr = sp_OADestroy @sbResponseBody
        RETURN
      END

    DECLARE @jResp int
    EXEC @hr = sp_OACreate 'Chilkat_9_5_0.JsonObject', @jResp OUT

    EXEC sp_OAMethod @jResp, 'LoadSb', @success OUT, @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

    DECLARE @EffectiveEngineVersion nvarchar(max)

    DECLARE @SelectedEngineVersion nvarchar(max)

    DECLARE @Query nvarchar(max)

    DECLARE @Catalog nvarchar(max)

    DECLARE @Database nvarchar(max)

    DECLARE @QueryExecutionId nvarchar(max)

    DECLARE @S3AclOption nvarchar(max)

    DECLARE @EncryptionOption nvarchar(max)

    DECLARE @KmsKey nvarchar(max)

    DECLARE @ExpectedBucketOwner nvarchar(max)

    DECLARE @OutputLocation nvarchar(max)

    DECLARE @StatementType nvarchar(max)

    DECLARE @DataManifestLocation nvarchar(max)

    DECLARE @DataScannedInBytes int

    DECLARE @EngineExecutionTimeInMillis int

    DECLARE @QueryPlanningTimeInMillis int

    DECLARE @QueryQueueTimeInMillis int

    DECLARE @ServiceProcessingTimeInMillis int

    DECLARE @TotalExecutionTimeInMillis int

    DECLARE @ErrorCategory int

    DECLARE @ErrorMessage nvarchar(max)

    DECLARE @ErrorType int

    DECLARE @Retryable int

    DECLARE @CompletionDateTime int

    DECLARE @State nvarchar(max)

    DECLARE @StateChangeReason nvarchar(max)

    DECLARE @SubmissionDateTime int

    DECLARE @WorkGroup nvarchar(max)

    DECLARE @ErrorCode nvarchar(max)

    DECLARE @i int
    SELECT @i = 0
    DECLARE @count_i int
    EXEC sp_OAMethod @jResp, 'SizeOfArray', @count_i OUT, 'QueryExecutions'
    WHILE @i < @count_i
      BEGIN
        EXEC sp_OASetProperty @jResp, 'I', @i
        EXEC sp_OAMethod @jResp, 'StringOf', @EffectiveEngineVersion OUT, 'QueryExecutions[i].EngineVersion.EffectiveEngineVersion'
        EXEC sp_OAMethod @jResp, 'StringOf', @SelectedEngineVersion OUT, 'QueryExecutions[i].EngineVersion.SelectedEngineVersion'
        EXEC sp_OAMethod @jResp, 'StringOf', @Query OUT, 'QueryExecutions[i].Query'
        EXEC sp_OAMethod @jResp, 'StringOf', @Catalog OUT, 'QueryExecutions[i].QueryExecutionContext.Catalog'
        EXEC sp_OAMethod @jResp, 'StringOf', @Database OUT, 'QueryExecutions[i].QueryExecutionContext.Database'
        EXEC sp_OAMethod @jResp, 'StringOf', @QueryExecutionId OUT, 'QueryExecutions[i].QueryExecutionId'
        EXEC sp_OAMethod @jResp, 'StringOf', @S3AclOption OUT, 'QueryExecutions[i].ResultConfiguration.AclConfiguration.S3AclOption'
        EXEC sp_OAMethod @jResp, 'StringOf', @EncryptionOption OUT, 'QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.EncryptionOption'
        EXEC sp_OAMethod @jResp, 'StringOf', @KmsKey OUT, 'QueryExecutions[i].ResultConfiguration.EncryptionConfiguration.KmsKey'
        EXEC sp_OAMethod @jResp, 'StringOf', @ExpectedBucketOwner OUT, 'QueryExecutions[i].ResultConfiguration.ExpectedBucketOwner'
        EXEC sp_OAMethod @jResp, 'StringOf', @OutputLocation OUT, 'QueryExecutions[i].ResultConfiguration.OutputLocation'
        EXEC sp_OAMethod @jResp, 'StringOf', @StatementType OUT, 'QueryExecutions[i].StatementType'
        EXEC sp_OAMethod @jResp, 'StringOf', @DataManifestLocation OUT, 'QueryExecutions[i].Statistics.DataManifestLocation'
        EXEC sp_OAMethod @jResp, 'IntOf', @DataScannedInBytes OUT, 'QueryExecutions[i].Statistics.DataScannedInBytes'
        EXEC sp_OAMethod @jResp, 'IntOf', @EngineExecutionTimeInMillis OUT, 'QueryExecutions[i].Statistics.EngineExecutionTimeInMillis'
        EXEC sp_OAMethod @jResp, 'IntOf', @QueryPlanningTimeInMillis OUT, 'QueryExecutions[i].Statistics.QueryPlanningTimeInMillis'
        EXEC sp_OAMethod @jResp, 'IntOf', @QueryQueueTimeInMillis OUT, 'QueryExecutions[i].Statistics.QueryQueueTimeInMillis'
        EXEC sp_OAMethod @jResp, 'IntOf', @ServiceProcessingTimeInMillis OUT, 'QueryExecutions[i].Statistics.ServiceProcessingTimeInMillis'
        EXEC sp_OAMethod @jResp, 'IntOf', @TotalExecutionTimeInMillis OUT, 'QueryExecutions[i].Statistics.TotalExecutionTimeInMillis'
        EXEC sp_OAMethod @jResp, 'IntOf', @ErrorCategory OUT, 'QueryExecutions[i].Status.AthenaError.ErrorCategory'
        EXEC sp_OAMethod @jResp, 'StringOf', @ErrorMessage OUT, 'QueryExecutions[i].Status.AthenaError.ErrorMessage'
        EXEC sp_OAMethod @jResp, 'IntOf', @ErrorType OUT, 'QueryExecutions[i].Status.AthenaError.ErrorType'
        EXEC sp_OAMethod @jResp, 'IntOf', @Retryable OUT, 'QueryExecutions[i].Status.AthenaError.Retryable'
        EXEC sp_OAMethod @jResp, 'IntOf', @CompletionDateTime OUT, 'QueryExecutions[i].Status.CompletionDateTime'
        EXEC sp_OAMethod @jResp, 'StringOf', @State OUT, 'QueryExecutions[i].Status.State'
        EXEC sp_OAMethod @jResp, 'StringOf', @StateChangeReason OUT, 'QueryExecutions[i].Status.StateChangeReason'
        EXEC sp_OAMethod @jResp, 'IntOf', @SubmissionDateTime OUT, 'QueryExecutions[i].Status.SubmissionDateTime'
        EXEC sp_OAMethod @jResp, 'StringOf', @WorkGroup OUT, 'QueryExecutions[i].WorkGroup'
        SELECT @i = @i + 1
      END
    SELECT @i = 0
    EXEC sp_OAMethod @jResp, 'SizeOfArray', @count_i OUT, 'UnprocessedQueryExecutionIds'
    WHILE @i < @count_i
      BEGIN
        EXEC sp_OASetProperty @jResp, 'I', @i
        EXEC sp_OAMethod @jResp, 'StringOf', @ErrorCode OUT, 'UnprocessedQueryExecutionIds[i].ErrorCode'
        EXEC sp_OAMethod @jResp, 'StringOf', @ErrorMessage OUT, 'UnprocessedQueryExecutionIds[i].ErrorMessage'
        EXEC sp_OAMethod @jResp, 'StringOf', @QueryExecutionId OUT, 'UnprocessedQueryExecutionIds[i].QueryExecutionId'
        SELECT @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"
    --     }
    --   ]
    -- }

    EXEC @hr = sp_OADestroy @rest
    EXEC @hr = sp_OADestroy @authAws
    EXEC @hr = sp_OADestroy @json
    EXEC @hr = sp_OADestroy @sbRequestBody
    EXEC @hr = sp_OADestroy @sbResponseBody
    EXEC @hr = sp_OADestroy @jResp


END
GO