athens 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 29669376828c5d70eff208ef500bb16c16a6b8a3d20a2015fcbe4da7fc7a94c2
-  data.tar.gz: 890e8479a158b000eed78784064c4e7773c06eca9ddcc1b8223edc4f0d96a03f
+  metadata.gz: aeee2d9b7817273521ffe4d0f54865b8f5ce21d185175381ce9d8f013b4efca6
+  data.tar.gz: c7ed5eacc1231fb5d81061bdfd700ca96bcab9f2e6635f3a9a24512de2f27107
 SHA512:
-  metadata.gz: 9741ed71a51d70125c76392cc6eff06eaf2d2ba403f525a9cc953f1152e392fe2c1f8d76508e53e5ffd6a9e1b52f194fd05eaabfa9d907ba2a5de2509d5ac826
-  data.tar.gz: ff41822fb3ae5bf379c36a7901150db1ee130f4c21315b5ec2b11c254574b55bd9d0f29acd28c169706737eaacedcaefb582bf42df2434de07adfea58b55b736
+  metadata.gz: 78c927b03a064c1e9d8ce3ae471aa6fc6f1852ba192abe69a14deae61bf8e020cebda5b3dcc6e8607a9944e373192df9f3ccb3ba23d9b8277ac4ac8b4043f006
+  data.tar.gz: a0cc740c6bf5a6238e3ecf192257c187d94f8e05a34422661d538949a4207abb856d1930263cf489d2a1950770724e1e047f4b30185cdbd7a7e0fc8193d85f54

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.3.2 / 2020-11-24
+
+* Added optional `request_token` and `work_group` parameters to the query execute method (thanks [mediafinger](https://github.com/mediafinger))
+
 ## 0.3.1 / 2020-07-20
 
 * Bumped development rake version from 0.10 to 0.13 for security fixes

data/README.md

@@ -146,6 +146,13 @@ query.cancel   # Attempts to cancel an in-progress query, returns true or false
 query.to_a(header_row: false)  # If you want your query results returned without a header row of column names
 ```
 
+The execute method also optionally supports the `request_token` and `work_group` [parameters](https://docs.aws.amazon.com/athena/latest/APIReference/API_StartQueryExecution.html#API_StartQueryExecution_RequestSyntax):
+
+```ruby
+conn = Athens::Connection.new(database: 'sample')
+query = conn.execute("SELECT * FROM mytable", request_token: single_use_token, work_group: my_work_group)
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/Rakefile

@@ -1,2 +1 @@
 require "bundler/gem_tasks"
-task :default => :spec

data/lib/athens/connection.rb

@@ -8,7 +8,7 @@ module Athens
     def initialize(database: nil, aws_client_override: {})
       @database_name = database
 
-      client_config = { 
+      client_config = {
         access_key_id: Athens.configuration.aws_access_key,
         secret_access_key: Athens.configuration.aws_secret_key,
         region: Athens.configuration.aws_region
@@ -19,26 +19,28 @@ module Athens
 
     # Runs a query against Athena, returning an Athens::Query object
     # that you can use to wait for it to finish or get the results
-    def execute(query)
+    def execute(query, request_token: nil, work_group: nil)
       if @database_name
         resp = @client.start_query_execution(
           query_string: query,
           query_execution_context: context,
-          result_configuration: result_config
+          result_configuration: result_config,
+          client_request_token: request_token,
+          work_group: work_group
         )
       else
         resp = @client.start_query_execution(
           query_string: query,
           result_configuration: result_config
         )
-      end        
+      end
 
       return Athens::Query.new(self, resp.query_execution_id)
     end
 
     private
 
-      def context 
+      def context
         Aws::Athena::Types::QueryExecutionContext.new(database: @database_name)
       end
 

data/lib/athens/version.rb

@@ -1,3 +1,3 @@
 module Athens
-  VERSION = "0.3.1"
+  VERSION = "0.3.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: athens
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Chris Schulte
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-20 00:00:00.000000000 Z
+date: 2020-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-athena