activerecord-amazon-timestream-adapter 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 864faadd28644aa8d276fa335661e02ee37b55640cdd776c2e9f99eb2b58ff72
+  data.tar.gz: c38741466d3ea5c6a67acee2bf1ec4f5ac553eac0a840d9a4e66e8b14c8c4e86
+SHA512:
+  metadata.gz: 72c9859c3dcc5a56f732822a85d15369a2652935e5542586ae6200ef9552ed1e29345fb25d90e6382a06f86114e1279e2228b0bf0d06bcf43e6ffadc7db849b8
+  data.tar.gz: 5ebb0ae723a08796cb58a6bc8c1f45792ae4824c8c03021eb7a80e9a36b51c8751897c0d7648ec7d2d9fb9504fe7dbae6773160643aa8a4ab9094261bb73d9d3

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2020 Francois Deschenes
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,3 @@
+# ActiveRecord Amazon Timestream Adapter
+
+This is the ActiveRecord adapter for working with [Amazon Timestream](https://aws.amazon.com/timestream/).

data/lib/active_record/connection_adapters/amazon_timestream/database_statements.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require 'active_record/result'
+
+module ActiveRecord
+  module ConnectionAdapters
+    module AmazonTimestream
+      module DatabaseStatements
+        def execute(sql, name = nil)
+          log(sql, name) do
+            @query.query({ query_string: sql })
+          end
+        end
+
+        def select_rows(sql, name = nil, binds = [])
+          exec_query(sql, name, binds).to_a.map(&:values)
+        end
+
+        def exec_query(sql, name = 'SQL', binds = [])
+          log(sql, name, binds) do
+            response = @query.query query_string: sql
+
+            ActiveRecord::Result.new response.column_info.map(&:name),
+                                     response.rows.map { |r| r.data.map(&:scalar_value) },
+                                     column_types_from(response.column_info)
+          end
+        end
+
+        private
+
+        def column_types_from(column_info)
+          column_info.each_with_object({}) { |c, h| h[c.name] = lookup_cast_type(c.type.scalar_type) }
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/amazon_timestream/quoting.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module ConnectionAdapters
+    module AmazonTimestream
+      module Quoting
+        def quote_table_name(table_name)
+          "\"#{@config[:database]}\".\"#{quote_string(table_name)}\""
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/amazon_timestream/schema_statements.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require 'active_record/connection_adapters/column'
+
+module ActiveRecord
+  module ConnectionAdapters
+    module AmazonTimestream
+      module SchemaStatements
+        def tables(_name = nil)
+          response = @write.list_tables database_name: @config[:database]
+
+          response.tables.map(&:table_name)
+        end
+
+        def columns(table_name, _name = nil)
+          response = @query.query query_string: "SELECT * FROM #{quote_table_name(table_name)} LIMIT 1"
+          response.column_info.map do |column|
+            Column.new(column.name, nil, lookup_cast_type(column.type.scalar_type))
+          end
+        end
+
+        def table_exists?(_table_name)
+          tables.include?(table_name)
+        end
+
+        def primary_key(_table_name)
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/active_record/connection_adapters/amazon_timestream_adapter.rb

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require 'active_record/connection_adapters/abstract_adapter'
+require 'active_record/connection_adapters/amazon_timestream/database_statements'
+require 'active_record/connection_adapters/amazon_timestream/quoting'
+require 'active_record/connection_adapters/amazon_timestream/schema_statements'
+require 'arel/visitors/amazon_timestream'
+require 'aws-sdk-core'
+require 'aws-sdk-timestreamquery'
+require 'aws-sdk-timestreamwrite'
+
+module ActiveRecord
+  class Base
+    def self.amazon_timestream_connection(config)
+      config = config.symbolize_keys
+
+      raise ArgumentError, 'No database specified. Missing argument: database.' unless config.key?(:database)
+
+      ConnectionAdapters::AmazonTimestreamAdapter.new(logger, config)
+    end
+  end
+
+  module ConnectionAdapters
+    class AmazonTimestreamAdapter < AbstractAdapter
+      ADAPTER_NAME = 'Amazon Timestream'
+
+      include AmazonTimestream::SchemaStatements
+      include AmazonTimestream::DatabaseStatements
+      include AmazonTimestream::Quoting
+
+      def initialize(logger, config)
+        super(nil, logger)
+        @config = config
+        @visitor = Arel::Visitors::AmazonTimestream.new self
+        connect!
+      end
+
+      def supports_explain?
+        false
+      end
+
+      def requires_reloading?
+        false
+      end
+
+      def active?
+        @query.query({ query_string: 'SELECT 1' })
+        true
+      rescue StandardError
+        false
+      end
+
+      protected
+
+      def initialize_type_map(mapping)
+        register_class_with_limit mapping, /boolean/i, Type::Boolean
+        register_class_with_limit mapping, /date/i, Type::Date
+        register_class_with_limit mapping, /double/i, Type::Float
+        register_class_with_limit mapping, /int/i, Type::Integer
+        register_class_with_limit mapping, /^time$/i, Type::Time
+        register_class_with_limit mapping, /timestamp/i, Type::DateTime
+        register_class_with_limit mapping, /varchar/i, Type::String
+
+        mapping.alias_type(/unknown/i, 'varchar')
+      end
+
+      private
+
+      def connect!
+        if @config[:username] && @config[:password]
+          credentials = Aws::Credentials.new(@config[:username], @config[:password])
+        end
+
+        @query = Aws::TimestreamQuery::Client.new credentials: credentials
+        @write = Aws::TimestreamWrite::Client.new credentials: credentials
+      end
+    end
+  end
+end

data/lib/activerecord-amazon-timestream-adapter.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require 'active_record/connection_adapters/amazon_timestream_adapter'

data/lib/arel/visitors/amazon_timestream.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'arel/visitors'
+
+module Arel
+  module Visitors
+    class AmazonTimestream < Arel::Visitors::ToSql
+    end
+  end
+end
+
+Arel::Visitors::VISITORS['amazon_timestream'] = Arel::Visitors::AmazonTimestream

metadata

@@ -0,0 +1,134 @@
+--- !ruby/object:Gem::Specification
+name: activerecord-amazon-timestream-adapter
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Francois Deschenes
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-11-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.2'
+- !ruby/object:Gem::Dependency
+  name: arel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-timestreamquery
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-timestreamwrite
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: 
+email: fdeschenes@me.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- lib/active_record/connection_adapters/amazon_timestream/database_statements.rb
+- lib/active_record/connection_adapters/amazon_timestream/quoting.rb
+- lib/active_record/connection_adapters/amazon_timestream/schema_statements.rb
+- lib/active_record/connection_adapters/amazon_timestream_adapter.rb
+- lib/activerecord-amazon-timestream-adapter.rb
+- lib/arel/visitors/amazon_timestream.rb
+homepage: https://github.com/fdeschenes/activerecord-amazon-timestream-adapter
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - "~>"
+    - !ruby/object:Gem::Version
+      version: '2.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: ActiveRecord Amazon Timestream Adapter
+test_files: []