minus5_mssql 0.1.0

data/Rakefile

@@ -0,0 +1,14 @@
+require "rake/testtask"
+require 'rubygems/package_task'
+load 'minus5_mssql.gemspec'
+
+task :default => [:test]
+
+Rake::TestTask.new do |test|
+  test.libs << "test"
+  test.test_files = Dir[ "test/test_*.rb" ]
+  test.verbose = true
+end
+
+Gem::PackageTask.new(GEMSPEC) do |pkg|
+end

data/lib/minus5_mssql.rb

@@ -0,0 +1,6 @@
+require 'rubygems'
+require 'tiny_tds'
+require 'yaml'
+
+$LOAD_PATH.unshift "#{File.dirname(__FILE__)}/minus5_mssql/"
+require 'adapter.rb'

data/lib/minus5_mssql/adapter.rb

@@ -0,0 +1,132 @@
+module Minus5
+
+  module Mssql
+
+    class Adapter
+
+      # params - tiny_tds connection params: https://github.com/rails-sqlserver/tiny_tds
+      # with additon of mirror_host
+      # Example:
+      #    SqlBase.new({  :username    => "rails",
+      #                   :password    => "",
+      #                   :host        => "bedem",
+      #                   :mirror_host => "mssql",
+      #                   :database    => "activerecord_unittest_mirroring"
+      #                })
+      def initialize(params)
+        params = YAML.load_file(params).symbolize_keys if params.kind_of?(String)
+        @params = params
+        @params_cache = {}
+        connect
+      end
+
+      # Insert row into table_name.
+      # Data is hash {column_name => value, ...}
+      # Acutal column names will be discovered from database.
+      def insert(table_name, data)
+        columns = get_params(table_name).reject{|c| c == "id"}
+        values = hash_to_values columns, data
+        sql = "insert into #{table_name} (#{columns.join(',')}) values (#{values.join(',')})"
+        execute(sql).insert
+      end
+
+      # Delete rows from table_name.
+      # Data is hash with keys eg. {:id => 123}
+      def delete(table_name, data)
+        columns, values = hash_to_columns_values(data)
+        keys = []
+        for i in (0..columns.size-1)
+          keys << "#{columns[i]} = #{values[i]}"
+        end
+        sql = "delete from #{table_name} where #{keys.join(' and ')}"
+        execute(sql).cancel
+      end
+
+      # Send query to the database. With reconnect in case of db mirroring failover.
+      def execute(sql)
+        @connection.execute(sql)
+      rescue TinyTds::Error => e
+        print "execute error #{e}\n"
+        connect
+        execute(sql)
+      end
+
+      # Returns results first column of the first row.
+      def select_value(sql)
+        rows = execute(sql).each
+        return nil if rows.size == 0
+        row = rows[0]
+        row[row.keys[0]]
+      end
+
+      def select(sql)
+        rows = execute(sql).each
+        rows.size == 1 ? rows[0] : rows
+      end
+
+      private
+
+      def connect
+        print "connecting to #{@params[:host]} "
+        @connection = TinyTds::Client.new(@params)
+        print "successful\n"
+      rescue TinyTds::Error => e
+        print "#{e.to_s}\n"
+        throw unless @params[:mirror_host]
+        to_mirror
+        connect
+      end
+
+      # Switch host and mirror_host in @params
+      def to_mirror
+        host = @params[:host]
+        @params[:host] = @params[:mirror_host]
+        @params[:mirror_host] = host
+        @params[:dataserver] = "#{@params[:host]}:#{@params[:port] || 1433}"
+      end
+
+      # Read table or stored procedure param names from database.
+      # Returns array of table column names or stored procedure param
+      def get_params(name)
+        @params_cache[name] ||=
+        begin
+          sql = "select name from sys.syscolumns where id = object_id('#{name}')"
+          @connection.execute(sql).each.map{|row| row["name"]}
+        end
+      end
+
+      def hash_to_values(columns, data)
+        values = columns.map do |column|
+          value = data[column.to_sym]
+          value = data[column.to_s]  if value.nil?
+          if column == "time" && value.kind_of?(String)
+            value = Time.parse(value)
+          end
+          if value.nil?
+            'null'
+          elsif value.kind_of?(String)
+            #"'#{value.gsub("\'","''")}'"
+            "'#{@connection.escape(value)}'"
+          elsif value.kind_of?(Date)
+            "'#{value.strftime("%Y-%m-%d")}'"
+          elsif value.kind_of?(Time) || value.kind_of?(DateTime)
+            "'#{value.strftime("%Y-%m-%d %H:%M:%S")}'"
+          elsif value.kind_of?(Integer) || value.kind_of?(Fixnum) || value.kind_of?(Bignum) || value.kind_of?(Float) || value.kind_of?(Rational)
+            "#{value.to_s}"
+          else
+            "'#{@connection.escape(value.to_s)}'"
+          end
+        end
+        values
+      end
+
+      def hash_to_columns_values(data)
+        columns = data.each_key.map{|key| key.to_s}
+        [columns, hash_to_values(columns, data)]
+      end
+
+    end
+
+  end
+
+end

data/test/test_db_mirroring.rb

@@ -0,0 +1,83 @@
+require 'rubygems'
+require 'test/unit'
+require 'pp'
+
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../lib"
+require 'minus5_mssql.rb'
+
+class Reader < Minus5::Mssql::Adapter
+
+  def read
+    execute("select * from people").each(:symbolize_keys => true)
+  end
+
+  def failover
+    execute("use master; ALTER DATABASE activerecord_unittest_mirroring SET PARTNER FAILOVER")
+  end
+
+end
+
+class DbMirroring < Test::Unit::TestCase
+
+  def setup
+    @reader = Reader.new({:username => "rails",
+                           :password => "",
+                           :host => "bedem",
+                           :mirror_host => "mssql",
+                           :database => "activerecord_unittest_mirroring"})
+  end
+
+  def test_get_params
+    columns = @reader.send(:get_params, 'people')
+    assert_equal 3, columns.size
+    assert columns.include?("id")
+    assert columns.include?("first_name")
+    assert columns.include?("last_name")
+  end
+
+  def test_insert
+    rollback_transaction(@reader) do
+      id = @reader.insert('people', {:first_name => "Igor", :last_name => "Anic"})
+      assert_equal 3, @reader.select_value("select count(*) from people")
+      @reader.delete('people', {:id => id})
+      assert_equal 2, @reader.select_value("select count(*) from people")
+    end
+  end
+
+  def test_read
+    rows = @reader.read
+    data_test rows
+    pp rows
+    @reader.failover
+    rows = @reader.read
+    data_test rows
+  end
+
+  def data_test(rows)
+    assert_equal 2, rows.size
+    assert "Sasa", rows[0][:first_name]
+    assert "Goran", rows[1][:first_name]
+  end
+
+  private
+
+  def setup_table
+    @reader.execute "
+    if not exists(select * from sys.objects where name = 'people')
+    create table people (id int identity, first_name varchar(255), last_name varchar(255))
+    "
+
+    @reader.execute "truncate table people"
+    @reader.execute "insert into people (first_name, last_name) values ('Sasa', 'Juric')"
+    @reader.execute "insert into people (first_name, last_name) values ('Goran', 'Pizent')"
+  end
+
+  def rollback_transaction(client)
+    client.execute("BEGIN TRANSACTION").do
+    yield
+  ensure
+    client.execute("ROLLBACK TRANSACTION").do
+  end
+
+
+end

metadata

@@ -0,0 +1,85 @@
+--- !ruby/object:Gem::Specification 
+name: minus5_mssql
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Igor Anic
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-31 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: tiny_tds
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 0
+        - 4
+        - 5
+        version: 0.4.5
+  type: :runtime
+  version_requirements: *id001
+description: "  minus5_mssql is a simple lib for working with Microsoft Sql Server\n  it is built on top of tiny_tds (https://github.com/rails-sqlserver/tiny_tds)\n"
+email: ianic@minus5.hr
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/minus5_mssql.rb
+- lib/minus5_mssql/adapter.rb
+- test/test_db_mirroring.rb
+- Rakefile
+has_rdoc: true
+homepage: http://www.minus5.hr
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.3
+signing_key: 
+specification_version: 3
+summary: minus5 mssql library
+test_files: []
+