db2_query 0.1.0 → 0.3.0

data/lib/db2_query/connection_handling.rb

@@ -1,56 +0,0 @@
-# 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

@@ -1,50 +0,0 @@
-# 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

@@ -1,32 +0,0 @@
-# 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/log_subscriber.rb

@@ -1,50 +0,0 @@
-# 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

@@ -1,38 +0,0 @@
-# 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

@@ -1,17 +0,0 @@
-# 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/schema.rb

@@ -1,113 +0,0 @@
-# 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

@@ -1,26 +0,0 @@
-# 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/tasks/database.rake

@@ -1,56 +0,0 @@
-# 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