db2_query 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 5d2946448234d56484778d59e3051573c81c2789631776f0ed543a79fdc510fa
+  data.tar.gz: a257fabf1a7de500f7d325c70dd11f494a08163dbf78460f61b3bf51cb626b77
+SHA512:
+  metadata.gz: ca8c1b52138ec97d9f4e9cbbca0d8bb148e95f4ac181a929e3e5715b597ed59b70e25b779156ba45a7fe9f119203e4162a2a932c63d11c94acdc2a1eaeb56db8
+  data.tar.gz: 67f0d7228274d4a7a6e38771cc0743b83da27621d3ce310e0d5d7d323e937316f700ec3b278734312ff27cdc88796321cb09f02d7c2bf48b8c5607a7331a7f27

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright 2020 yohanes_l
+
+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,124 @@
+# Db2Query
+A Rails query plugin to fetch data from Db2 database by using ODBC connection.
+
+## Installation
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'db2_query'
+```
+
+And then execute:
+```bash
+$ bundle
+```
+
+Or install it yourself as:
+```bash
+$ gem install db2-query
+```
+
+## Initialization
+Execute init task at the app root
+```bash
+$ rake db2query:init
+```
+Db2Query will generate two required files:
+- `config/db2query_database.yml`
+- `config/initializers/db2query`
+
+Edit these files according to the requirement.
+
+### Database Configuration
+At `db2query_database.yml` we can use two type of connection:
+1. DSN connection config
+2. Connection String config
+```yml
+development:
+  primary:                         # Connection String Example
+    conn_string:
+      driver: DB2
+      database: SAMPLE
+      dbalias: SAMPLE
+      hostname: LOCALHOST
+      currentschema: LIBTEST
+      port: "0"
+      protocol: IPC
+      uid: <%= ENV["DB2EC_UID"] %>
+      pwd: <%= ENV["DB2EC_PWD"] %>
+  secondary:                       # DSN Example
+    dsn: SAMPLE
+    uid: <%= ENV["DB2EC_UID"] %>
+    pwd: <%= ENV["DB2EC_PWD"] %>
+```
+
+## Usage
+### Basic Usage
+Create query class that inherit from `Db2Query::Base` in `app/queries` folder
+```ruby
+class Users < Db2Query::Base
+  query :find_by, <<-SQL
+    SELECT * FROM LIBTEST.USERS WHERE user_id = ?
+  SQL
+end
+```
+Or use a normal sql method (don't forget the `_sql` suffix)
+```ruby
+class Users < Db2Query::Base 
+  def find_by_sql
+    "SELECT * FROM LIBTEST.USERS WHERE user_id = ?"
+  end
+end
+```
+Check it at rails console
+```bash
+Users.find_by 10000
+Users Load (330.28ms)  SELECT * FROM LIBTEST.USERS WHERE user_id = ? [[10000]]
+=> #<Db2Query::Result [#<Users::FindBy user_id: 10000, first_name: "Alex", last_name: "Jacobi", email: "lula_durgan@dooley.com">]>
+```
+### Formatter
+In order to get different result column format, a query result can be reformatted by add a formatter class that inherit `Db2Query::AbstractFormatter` then register at `config\initializers\db2query.rb`
+```ruby
+require "db2_query/formatter"
+
+# create a formatter class
+class FirstNameFormatter < Db2Query::AbstractFormatter
+  def format(value)
+    "Mr/Mrs. " + value
+  end
+end
+
+# register the formatter class
+Db2Query::Formatter.registration do |format|
+  format.register(:first_name_formatter, FirstNameFormatter)
+end
+```
+Use it at query class
+```ruby
+class Users < Db2Query::Base
+  attributes :first_name, :first_name_formatter
+
+  query :find_by, <<-SQL
+    SELECT * FROM LIBTEST.USERS WHERE user_id = ?
+  SQL
+end
+```
+Check it at rails console
+```bash
+Users.find_by 10000
+Users Load (330.28ms)  SELECT * FROM LIBTEST.USERS WHERE user_id = ? [[10000]]
+=> #<Db2Query::Result [#<Users::FindBy user_id: 10000, first_name: "Mr/Mrs. Alex", last_name: "Jacobi", email: "lula_durgan@dooley.com">]>
+```
+### Available methods
+Db2Query::Result has public methods as follows:
+- to_a
+- to_hash
+- pluck
+- first
+- last
+- size
+- each
+
+
+## License
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+begin
+  require "bundler/setup"
+rescue LoadError
+  puts "You must `gem install bundler` and `bundle install` to run rake tasks"
+end
+
+require "rdoc/task"
+
+RDoc::Task.new(:rdoc) do |rdoc|
+  rdoc.rdoc_dir = "rdoc"
+  rdoc.title    = "Db2Query"
+  rdoc.options << "--line-numbers"
+  rdoc.rdoc_files.include("README.md")
+  rdoc.rdoc_files.include("lib/**/*.rb")
+end
+
+require "bundler/gem_tasks"
+
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.pattern = "test/**/*_test.rb"
+  t.verbose = false
+end
+
+task default: :test

data/lib/db2_query.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal:true
+
+require "odbc"
+require "yaml"
+require "erb"
+require "active_record"
+require "active_support"
+
+module Db2Query
+  extend ActiveSupport::Autoload
+
+  autoload :Version
+  autoload :Error
+  autoload :Path
+  autoload :Schema
+  autoload :DatabaseConfigurations
+  autoload :ODBCConnector
+  autoload :Connection
+  autoload :ConnectionHandling
+  autoload :DatabaseStatements
+  autoload :SQLValidator
+  autoload :LogSubscriber
+  autoload :Formatter
+  autoload :Column
+  autoload :Result
+  autoload :Base
+end
+
+require "db2_query/railtie"

data/lib/db2_query/base.rb

@@ -0,0 +1,84 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Base
+    include DatabaseConfigurations
+    include ConnectionHandling
+
+    class << self
+      include SQLValidator
+
+      def attributes(attr_name, format)
+        attr_format.store(attr_name, format)
+      end
+
+      def query(method_name, sql_statement)
+        unless is_query?(sql_statement)
+          raise Error, "Query only for SQL query commands."
+        end
+
+        if self.class.respond_to?(method_name)
+          raise Error, "Query :#{method_name} has been defined before"
+        end
+
+        self.class.define_method(method_name) do |*args|
+          log(sql_statement, args) do
+            columns, rows = connection.exec_query(sql_statement, *args)
+            Result.new(self, method_name, columns, rows, attr_format)
+          end
+        end
+      end
+
+      private
+        def attr_format
+          @attr_format ||= Hash.new
+        end
+
+        def define_query_method(method_name, sql_statement)
+          if is_query?(sql_statement)
+            query(method_name, sql_statement)
+          else
+            raise NotImplementedError
+          end
+        end
+
+        def method_missing(method_name, *args, &block)
+          sql_methods = self.instance_methods.grep(/_sql/)
+          sql_method = "#{method_name}_sql".to_sym
+
+          if sql_methods.include?(sql_method)
+            sql_statement = allocate.method(sql_method).call
+
+            raise Error, "Query methods must return a SQL statement string!" unless sql_statement.is_a? String
+
+            expected_args = sql_statement.count "?"
+            given_args = args.size
+
+            if expected_args == given_args
+              define_query_method(method_name, sql_statement)
+            else
+              raise ArgumentError, "wrong number of arguments (given #{given_args}, expected #{expected_args})"
+            end
+
+            method(method_name).call(*args)
+          else
+            super
+          end
+        end
+
+        def instrumenter
+          @instrumenter ||= ActiveSupport::Notifications.instrumenter
+        end
+
+        def log(sql_statement, args)
+          instrumenter.instrument(
+            "sql.db2_query",
+            sql:   sql_statement,
+            name:  self,
+            binds: args) do
+            yield
+          end
+        end
+    end
+  end
+end

data/lib/db2_query/column.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Column
+    attr_reader :name, :type, :formatter
+
+    FLOAT_TYPES = [ODBC::SQL_FLOAT, ODBC::SQL_DOUBLE, ODBC::SQL_DECIMAL, ODBC::SQL_REAL]
+    INTEGER_TYPES = [ODBC::SQL_TINYINT, ODBC::SQL_SMALLINT, ODBC::SQL_INTEGER, ODBC::SQL_BIGINT, ODBC::SQL_NUMERIC]
+
+    def initialize(name, type, format = nil)
+      @name = name
+      @type = type.to_i
+
+      @formatter = if custom_format?(format)
+        Formatter.lookup(format)
+      elsif float_type?
+        FloatFormatter.new
+      elsif integer_type?
+        IntegerFormatter.new
+      else
+        BareFormatter.new
+      end
+    end
+
+    def format(value)
+      formatter.format(value)
+    end
+
+    private
+      def custom_format?(format)
+        Formatter.registry.key?(format)
+      end
+
+      def float_type?
+        FLOAT_TYPES.include?(type)
+      end
+
+      def integer_type?
+        INTEGER_TYPES.include?(type)
+      end
+  end
+end

data/lib/db2_query/connection.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Connection
+    include DatabaseStatements
+
+    attr_reader :connector, :db_name, :odbc_conn
+
+    def initialize(connector, db_name)
+      @connector = connector
+      @db_name = db_name.to_sym
+      connect
+    end
+
+    def connect
+      @odbc_conn = connector.connect
+      @odbc_conn.use_time = true
+    end
+
+    def active?
+      @odbc_conn.connected?
+    end
+
+    def disconnect!
+      @odbc_conn.drop_all
+      @odbc_conn.disconnect if active?
+    end
+
+    def reconnect!
+      disconnect!
+      connect
+    end
+    alias reset! reconnect!
+
+    def execute(sql, *args)
+      reset! unless active?
+      @odbc_conn.do(sql, *args)
+    end
+
+    def exec_query(sql, *args)
+      reset! unless active?
+      statement = @odbc_conn.run(sql, *args)
+      columns = statement.columns.values
+      rows = statement.to_a
+      [columns, rows]
+    ensure
+      statement.drop if statement
+    end
+  end
+end

data/lib/db2_query/connection_handling.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module ConnectionHandling
+    extend ActiveSupport::Concern
+
+    DEFAULT_DB = "primary"
+
+    included do |base|
+      def base.inherited(child)
+        child.connection = @connection
+      end
+
+      base.extend ClassMethods
+      base.connection = nil
+    end
+
+    module ClassMethods
+      attr_reader :connection
+
+      def connection=(connection)
+        @connection = connection
+        update_descendants_connection unless self.descendants.empty?
+      end
+
+      def update_descendants_connection
+        self.descendants.each { |child| child.connection = @connection }
+      end
+
+      def establish_connection(db_name = nil)
+        clear_connection unless self.connection.nil?
+        db_name = db_name.nil? ? DEFAULT_DB : db_name.to_s
+
+        self.load_database_configurations if self.configurations.nil?
+
+        if self.configurations[db_name].nil?
+          raise Error, "Database (:#{db_name}) not found at database configurations."
+        end
+
+        conn_type, conn_config = extract_configuration(db_name)
+
+        connector = ODBCConnector.new(conn_type, conn_config)
+        self.connection = Connection.new(connector, db_name)
+      end
+
+      def current_database
+        @connection.db_name
+      end
+
+      def clear_connection
+        @connection.disconnect!
+        @connection = nil
+      end
+    end
+  end
+end

data/lib/db2_query/database_configurations.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module DatabaseConfigurations
+    extend ActiveSupport::Concern
+
+    DEFAULT_ENV = -> { (Rails.env if defined?(Rails.env)) || ENV["RAILS_ENV"].presence || ENV["RACK_ENV"].presence }
+
+    included do |base|
+      @@configurations = Hash.new
+
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def configurations
+        @@configurations
+      end
+
+      def load_database_configurations(path = nil)
+        file_path = path.nil? ? Path.database_config_file : path
+
+        if File.exist?(file_path)
+          file = File.read(file_path)
+          @@configurations = YAML.load(ERB.new(file).result)[DEFAULT_ENV.call]
+        else
+          raise Error, "Could not load db2query database configuration. No such file - #{file_path}"
+        end
+      end
+
+      def configurations_databases
+        self.load_database_configurations if self.configurations.nil?
+        @@configurations.keys
+      end
+
+      private
+        def extract_configuration(db_name)
+          configs = @@configurations[db_name.to_s]
+          conn_config = configs.nil? ? configs : configs.transform_keys(&:to_sym)
+          conn_type = (conn_config.keys & ODBCConnector::CONNECTION_TYPES).first
+
+          if conn_type.nil?
+            raise Error, "No data source name (:dsn) or connection string (:conn_str) provided."
+          end
+
+          [conn_type, conn_config]
+        end
+    end
+  end
+end

data/lib/db2_query/database_statements.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module DatabaseStatements
+    def query_value(sql) # :nodoc:
+      single_value_from_rows(query(sql))
+    end
+
+    def query(sql)
+      exec_query(sql).last
+    end
+
+    def query_values(sql)
+      query(sql).map(&:first)
+    end
+
+    def current_database
+      db_name.to_s
+    end
+
+    def current_schema
+      query_value("select current_schema from sysibm.sysdummy1").strip
+    end
+    alias library current_schema
+
+    private
+      def single_value_from_rows(rows)
+        row = rows.first
+        row && row.first
+      end
+  end
+end

data/lib/db2_query/error.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Error < ArgumentError; end
+end

data/lib/db2_query/formatter.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module Formatter
+    def self.register(name, klass)
+      self.registry.store(name, klass.new)
+    end
+
+    def self.registry
+      @@registry ||= Hash.new
+    end
+
+    def self.lookup(name)
+      @@registry.fetch(name.to_sym)
+    end
+
+    def self.registration(&block)
+      yield self if block_given?
+    end
+  end
+
+  class AbstractFormatter
+    def format(value)
+      raise NotImplementedError, "Implement format method in your subclass."
+    end
+  end
+
+  class BareFormatter < AbstractFormatter
+    def format(value)
+      value
+    end
+  end
+
+  class FloatFormatter < AbstractFormatter
+    def format(value)
+      value.to_f
+    end
+  end
+
+  class IntegerFormatter < AbstractFormatter
+    def format(value)
+      value.to_i
+    end
+  end
+end

data/lib/db2_query/log_subscriber.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class LogSubscriber < ActiveSupport::LogSubscriber
+    # Embed in a String to clear all previous ANSI sequences.
+    CLEAR   = "\e[0m"
+    BOLD    = "\e[1m"
+
+    # Colors
+    BLACK   = "\e[30m"
+    RED     = "\e[31m"
+    GREEN   = "\e[32m"
+    YELLOW  = "\e[33m"
+    BLUE    = "\e[34m"
+    MAGENTA = "\e[35m"
+    CYAN    = "\e[36m"
+    WHITE   = "\e[37m"
+
+    def sql(event)
+      class_load_duration = color("#{event.payload[:name]} Load (#{event.duration.round(2)}ms)", :cyan, true)
+      sql_statement = color("#{event.payload[:sql]}", :blue, true)
+      message = "  #{class_load_duration} #{sql_statement}"
+
+      if event.payload[:binds].size > 0
+        binds = color("#{event.payload[:binds]}", :white)
+        message = "#{message} [#{binds}]"
+      end
+
+      puts message
+    end
+
+    def schema_task(event)
+      puts color("  Done (#{(event.duration).round(2)}ms)", :green, true)
+    end
+
+    def schema_task_perform(event)
+      task_name = color(":#{event.payload[:task_name]}", :white, true)
+      schema = color("#{event.payload[:schema]}", :white, true)
+      puts "- Performing #{task_name} in #{schema} ..."
+    end
+
+    private
+      def color(text, color, bold = false) # :doc:
+        return text unless colorize_logging
+        color = self.class.const_get(color.upcase) if color.is_a?(Symbol)
+        bold  = bold ? BOLD : ""
+        "#{bold}#{color}#{text}#{CLEAR}"
+      end
+  end
+end

data/lib/db2_query/odbc_connector.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class ODBCConnector
+    attr_reader :connector, :conn_type, :conn_config
+
+    CONNECTION_TYPES = %i[dsn conn_string].freeze
+
+    def initialize(type, config)
+      @conn_type, @conn_config = type, config.transform_keys(&:to_sym)
+      @connector = Db2Query.const_get("#{conn_type.to_s.camelize}Connector").new
+    end
+
+    def connect
+      connector.connect(conn_config)
+    end
+  end
+
+  class DsnConnector
+    def connect(config)
+      ::ODBC.connect(config[:dsn], config[:uid], config[:pwd])
+    rescue ::ODBC::Error => e
+      raise Error, "Unable to activate ODBC DSN connection #{e}"
+    end
+  end
+
+  class ConnStringConnector
+    def connect(config)
+      driver = ::ODBC::Driver.new.tap do |d|
+        d.attrs = config[:conn_string].transform_keys(&:to_s)
+        d.name = "odbc"
+      end
+      ::ODBC::Database.new.drvconnect(driver)
+    rescue ::ODBC::Error => e
+      raise Error, "Unable to activate ODBC Conn String connection #{e}"
+    end
+  end
+end

data/lib/db2_query/path.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module Path
+    def self.database_config_file
+      @@database_config_file ||= nil
+    end
+
+    def self.database_config_file=(file_path)
+      @@database_config_file ||= file_path
+    end
+
+    def self.database_config_file_exists?
+      File.exist?(self.database_config_file)
+    end
+  end
+end

data/lib/db2_query/railtie.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Railtie < ::Rails::Railtie
+    railtie_name :db2_query
+
+    rake_tasks do
+      tasks_path = "#{File.expand_path("..", __dir__)}/tasks"
+      Dir.glob("#{tasks_path}/*.rake").each { |file| load file }
+    end
+
+    initializer "db2_query.database_initialization" do
+      Path.database_config_file = "#{Rails.root}/config/db2query_database.yml"
+      Base.load_database_configurations
+      Base.establish_connection
+    end
+
+    initializer "db2_query.attach_log_subscription" do
+      LogSubscriber.attach_to :db2_query
+    end
+  end
+end

data/lib/db2_query/result.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Result
+    attr_reader :core, :klass, :rows, :columns, :column_metadatas, :attr_format
+
+    def initialize(core, klass, columns, rows, attr_format = {})
+      @core = core
+      @klass = klass.to_s.camelize
+      @rows  = rows
+      @attr_format = attr_format
+      @columns = []
+      @column_metadatas = extract_metadatas(columns)
+    end
+
+    def to_a
+      core.const_set(klass, row_class) unless core.const_defined?(klass)
+
+      rows.map do |row|
+        (core.const_get klass).new(row, column_metadatas)
+      end
+    end
+
+    def to_hash
+      rows.map do |row|
+        Hash[columns.zip(row)]
+      end
+    end
+
+    def pluck(*column_names)
+      records.map do |record|
+        column_names.map { |column_name| record.send(column_name) }
+      end
+    end
+
+    def first
+      records.first
+    end
+
+    def last
+      records.last
+    end
+
+    def size
+      records.size
+    end
+    alias length size
+
+    def each(&block)
+      records.each(&block)
+    end
+
+    def inspect
+      entries = records.take(11).map!(&:inspect)
+      entries[10] = "..." if entries.size == 11
+      "#<#{self.class} [#{entries.join(', ')}]>"
+    end
+
+    private
+      def records
+        @records ||= to_a
+      end
+
+      def extract_metadatas(columns)
+        columns.map do |col|
+          @columns << column_name = col.name.downcase
+          Column.new(column_name, col.type, attr_format[column_name.to_sym])
+        end
+      end
+
+      def row_class
+        Class.new do
+          def initialize(row, columns_metadata)
+            columns_metadata.zip(row) do |column, val|
+              self.class.send(:attr_accessor, column.name.to_sym)
+              instance_variable_set("@#{column.name}", column.format(val))
+            end
+          end
+
+          def inspect
+            inspection = if defined?(instance_variables) && instance_variables
+              instance_variables.collect do |attr|
+                value = instance_variable_get(attr)
+                "#{attr[1..-1]}: #{(value.kind_of? String) ? %Q{"#{value}"} : value}"
+              end.compact.join(", ")
+            else
+              "not initialized"
+            end
+
+            "#<#{self.class} #{inspection}>"
+          end
+        end
+      end
+  end
+end

data/lib/db2_query/schema.rb

@@ -0,0 +1,113 @@
+# frozen_string_literal: true
+
+module Db2Query
+  class Schema
+    include SQLValidator
+
+    attr_accessor :config, :current_schema
+
+    def initialize
+      @config = Config.new
+    end
+
+    def connection
+      Base.connection
+    end
+
+    def schema
+      config.schema
+    end
+
+    def sql_files_dir(path)
+      config.sql_files_dir = path
+    end
+
+    def is_valid_schema?
+      connection.current_schema == config.schema
+    end
+
+    def perform_tasks(&block)
+      raise Error, "#{config.main_library} is not connection's current schema" unless is_valid_schema?
+
+      if Base.connection.nil? && config.init_connection
+        Base.establish_connection
+      end
+
+      puts "\n# Perform Schema Tasks: \n\n"
+
+      instance_eval(&block)
+
+      puts "\n"
+    end
+
+    def sql(sql_statement)
+      raise Error, "Task only for SQL execute commands." if query_command?(sql_statement)
+      sql_statement
+    end
+
+    def file(file_name)
+      sql(File.read("#{config.sql_files_dir}/#{file_name}"))
+    end
+
+    def task(task_name, sql = nil, *args)
+      Task.new(schema, task_name).perform do
+        if block_given?
+          yield(sql)
+        else
+          execute(sql, *args)
+        end
+      end
+    end
+
+    def execute(sql, *args)
+      connection.execute(sql, *args)
+    rescue ::ODBC::Error => e
+      raise Error, "Unable to execute SQL - #{e}"
+    end
+
+    def tables_in_schema
+      connection.query_values <<-SQL
+        SELECT table_name FROM SYSIBM.SQLTABLES
+        WHERE table_schem='#{schema}' AND table_type='TABLE'
+      SQL
+    end
+
+    def self.initiation(&block)
+      new.initiation(&block)
+    end
+
+    def initiation(&block)
+      instance_eval(&block)
+    end
+
+    class Task
+      attr_reader :instrumenter, :schema, :task_name, :start_time, :finish_time
+
+      def initialize(schema, task_name)
+        @schema = schema
+        @task_name = task_name
+        @instrumenter = ActiveSupport::Notifications.instrumenter
+      end
+
+      def perform
+        instrumenter.instrument("schema_task_perform.db2_query", payload)
+        instrumenter.start("schema_task.db2_query", payload)
+        yield
+        instrumenter.finish("schema_task.db2_query", payload)
+      end
+
+      private
+        def payload
+          { task_name: task_name, schema: schema }
+        end
+    end
+
+    class Config
+      attr_accessor :initial_tasks, :init_connection, :schema, :sql_files_dir
+
+      def initialize
+        @init_connection = false
+      end
+    end
+  end
+end

data/lib/db2_query/sql_validator.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Db2Query
+  module SQLValidator
+    DEFAULT_TASK_COMMANDS = [:create, :drop, :delete, :insert, :set, :update]
+    COMMENT_REGEX = %r{/\*(?:[^\*]|\*[^/])*\*/}m
+
+    def task_command?(sql_statement)
+      sql_statement.match?(task_commands_regexp)
+    end
+
+    def query_command?(sql_statement)
+      sql_statement.match?(/select/i)
+    end
+
+    def is_query?(sql_statement)
+      query_command?(sql_statement) && !task_command?(sql_statement)
+    end
+
+    private
+      def task_commands_regexp
+        parts = DEFAULT_TASK_COMMANDS.map { |part| /#{part}/i }
+        /\A(?:[\(\s]|#{COMMENT_REGEX})*#{Regexp.union(*parts)}/i
+      end
+  end
+end

data/lib/db2_query/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Db2Query
+  VERSION = "0.1.0"
+end

data/lib/tasks/database.rake

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+DB2_QUERY_DATABASE_TEMPLATE ||= <<-EOF
+# frozen_string_literal: true
+
+# Database configuration example
+development:
+  primary:
+    conn_string:
+      driver: DB2
+      database: SAMPLE
+      dbalias: SAMPLE
+      hostname: LOCALHOST
+      currentschema: LIBTEST
+      port: "0"
+      protocol: IPC
+      uid: <%= ENV["DB2EC_UID"] %>
+      pwd: <%= ENV["DB2EC_PWD"] %>
+  secondary:
+    dsn: SAMPLE
+    uid: <%= ENV["DB2EC_UID"] %>
+    pwd: <%= ENV["DB2EC_PWD"] %>
+
+test:
+  primary:
+    conn_string:
+      driver: DB2
+      database: SAMPLE
+      dbalias: SAMPLE
+      hostname: LOCALHOST
+      currentschema: LIBTEST
+      port: "0"
+      protocol: IPC
+      uid: <%= ENV["DB2EC_UID"] %>
+      pwd: <%= ENV["DB2EC_PWD"] %>
+  secondary:
+    dsn: SAMPLE
+    uid: <%= ENV["DB2EC_UID"] %>
+    pwd: <%= ENV["DB2EC_PWD"] %>
+EOF
+
+namespace :db2query do
+  desc "Create Database configuration file"
+  task :database do
+    database_path = "#{Rails.root}/config/db2query_database.yml"
+    if File.exist?(database_path)
+      raise ArgumentError, "File exists."
+    else
+      puts "  Creating database config file ..."
+      File.open(database_path, "w") do |file|
+        file.puts DB2_QUERY_DATABASE_TEMPLATE
+      end
+      puts "  File '#{database_path}' created."
+    end
+  end
+end

data/lib/tasks/init.rake

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+namespace :db2query do
+  desc "Create Initializer and Database configuration file"
+  task :init do
+    Rake::Task["db2query:initializer"].invoke
+    Rake::Task["db2query:database"].invoke
+  end
+end

data/lib/tasks/initializer.rake

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+DB2_QUERY_INITIALIZER_TEMPLATE ||= <<-EOF
+# frozen_string_literal: true
+
+# Example
+require "db2_query/formatter"
+
+class FirstNameFormatter < Db2Query::AbstractFormatter
+  def format(value)
+    "First Name: " + value
+  end
+end
+
+Db2Query::Formatter.registration do |format|
+  format.register(:first_name_formatter, FirstNameFormatter)
+end
+EOF
+
+namespace :db2query do
+  desc "Create Initializer file"
+  task :initializer do
+    # Create initializer file
+    initializer_path = "#{Rails.root}/config/initializers/db2query.rb"
+    if File.exist?(initializer_path)
+      raise ArgumentError, "File exists."
+    else
+      puts "  Creating initializer file ..."
+      File.open(initializer_path, "w") do |file|
+        file.puts DB2_QUERY_INITIALIZER_TEMPLATE
+      end
+      puts "  File '#{initializer_path}' created."
+    end
+  end
+end

metadata

@@ -0,0 +1,199 @@
+--- !ruby/object:Gem::Specification
+name: db2_query
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- yohanes_l
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-06-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.0.3
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.3.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 6.0.3
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.3.1
+- !ruby/object:Gem::Dependency
+  name: ruby-odbc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.99999'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.99999'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-performance
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A Rails query plugin to fetch data from Db2 database by using ODBC connection.
+email:
+- yohanes.lumentut@yahoo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- README.md
+- Rakefile
+- lib/db2_query.rb
+- lib/db2_query/base.rb
+- lib/db2_query/column.rb
+- lib/db2_query/connection.rb
+- lib/db2_query/connection_handling.rb
+- lib/db2_query/database_configurations.rb
+- lib/db2_query/database_statements.rb
+- lib/db2_query/error.rb
+- lib/db2_query/formatter.rb
+- lib/db2_query/log_subscriber.rb
+- lib/db2_query/odbc_connector.rb
+- lib/db2_query/path.rb
+- lib/db2_query/railtie.rb
+- lib/db2_query/result.rb
+- lib/db2_query/schema.rb
+- lib/db2_query/sql_validator.rb
+- lib/db2_query/version.rb
+- lib/tasks/database.rake
+- lib/tasks/init.rake
+- lib/tasks/initializer.rake
+homepage: https://github.com/yohaneslumentut/db2_query
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.3
+signing_key: 
+specification_version: 4
+summary: Db2Query
+test_files: []