minus5_mssql 0.1.0 → 0.2.0

data/lib/minus5_mssql/adapter.rb

@@ -53,26 +53,71 @@ module Minus5
 
       # 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]]
+        rows = execute(sql).each(:as=>:array)
+        return if rows.size == 0
+        rows[0][0].kind_of?(Array) ? rows[0][0][0] : rows[0][0]
+      end
+      
+      # Returns array of the values from the first column of all returned rows 
+      def select_values(sql)
+        rows = execute(sql).each(:as=>:array)
+        return [] if rows.size == 0
+        (rows[0][0].kind_of?(Array) ? rows[0] : rows).map{|row| row[0] }        
       end
 
-      def select(sql)
-        rows = execute(sql).each
-        rows.size == 1 ? rows[0] : rows
+      def select(options)
+        if options.kind_of?(String)
+          execute(options).each(:symbolize_keys=>true)
+        else
+          options = {:primary_key=>:id}.merge(options)
+          results = execute(options[:sql]).each(:symbolize_keys=>true)
+          return [] if results.size == 0
+          return results if results[0].kind_of?(Hash)
+          data = {} #parent indexed by primary_key
+          results[0].each{ |row| data[row[options[:primary_key]]] = row }
+          options[:relations].each_with_index do |relation, index|index
+            result = results[index+1] #child result
+            result.each do |row|
+              #find parent row by foreign key and insert child row in collection
+              data_row = data[row[relation[:foreign_key]]]
+              next unless data_row
+              if relation[:type] == :one_to_many
+               data_row[relation[:name]] = [] unless data_row[relation[:name]]
+                data_row[relation[:name]] << row
+              elsif relation[:type] == :one_to_one
+                #merge child row with parent
+                if relation[:name]
+                  data_row[relation[:name]] = row
+                else                
+                  data_row.merge!(row.reject{|key, value| key == relation[:foreign_key]})
+                end
+              end
+            end
+          end
+          options[:return_hash] ? data : results[0]
+        end
       end
 
+      def create_table(schema, table, columns_def)
+        execute <<-SQL
+          if not exists(select * 
+                        from sys.tables 
+                        inner join sys.schemas on tables.schema_id = schemas.schema_id 
+                        where 
+                          tables.name = '#{table}' 
+                          and schemas.name = '#{schema}')
+            create table #{schema}.#{table} (#{columns_def})
+        SQL
+      end
       private
 
       def connect
-        print "connecting to #{@params[:host]} "
+        #print "connecting to #{@params[:host]} "
         @connection = TinyTds::Client.new(@params)
-        print "successful\n"
+        #print "successful\n"
       rescue TinyTds::Error => e
-        print "#{e.to_s}\n"
-        throw unless @params[:mirror_host]
+        raise unless @params[:mirror_host]
+        #print "#{e.to_s}\n"
         to_mirror
         connect
       end
@@ -89,10 +134,10 @@ module Minus5
       # 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
+          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)
@@ -129,4 +174,4 @@ module Minus5
 
   end
 
-end
+end

data/test/helper.rb

@@ -0,0 +1,29 @@
+require 'rubygems'
+require 'test/unit'
+require 'pp'
+
+$LOAD_PATH.unshift File.dirname(__FILE__) + "/../lib"
+require 'minus5_mssql.rb'
+
+module Helper 
+
+  protected
+
+  def setup_table(adapter)
+    adapter.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))
+    "
+    adapter.execute "truncate table people"
+    adapter.execute "insert into people (first_name, last_name) values ('Sasa', 'Juric')"
+    adapter.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

data/test/test_select.rb

@@ -0,0 +1,120 @@
+$LOAD_PATH.unshift File.dirname(__FILE__)
+require 'helper.rb'
+
+class Select < Test::Unit::TestCase
+  include Helper
+
+  def setup
+    @reader = Minus5::Mssql::Adapter.new({:username => "rails",
+                                         :password => "",
+                                         :host => "mssql",
+                                         :database => "minus5_mssql_tests"}) 
+    create_tables
+  end
+
+  def test_select_value
+    assert_equal 1, @reader.select_value("select 1")
+    assert_equal 1, @reader.select_value("select 1, 2")
+    assert_equal 1, @reader.select_value("select 1\nunion\nselect 2")
+    assert_equal 1, @reader.select_value("select 1; select 2")
+    assert_nil @reader.select_value("set nocount on")
+  end
+
+  def test_select_values
+    assert_equal [1,2], @reader.select_values("select 1, 11\nunion\nselect 2, 22")
+    assert_equal [1,2], @reader.select_values("select 1, 11\nunion\nselect 2, 22; select 3, 3")
+    assert_equal [], @reader.select_values("set nocount on")
+  end
+  
+  def test_select_simple
+    assert_equal([{:first => 1, :second => 2}], @reader.select("select 1 first, 2 second"))
+    assert_equal([{:first => 1, :second => 2}], @reader.select(:sql=>"select 1 first, 2 second"))
+    assert_equal [], @reader.select("set nocount on")
+  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_delete
+    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
+
+
+  def test_simple_select
+    result = @reader.select("select * from orders")
+    assert result.kind_of?(Array)
+    assert_equal 3, result.size
+    assert_equal 3, result[0].keys.size
+  end
+
+  def test_parent_child_one_to_many_and_one_to_one
+    result = @reader.select(
+        :sql=>"select * from orders; select * from order_details order by no; select order_id, sum(no) sum_no from order_details group by order_id",
+        :primary_key=>:id,
+        :relations=>[{:type=>:one_to_many, :name=>:order_details, :foreign_key=>:order_id},
+                     {:type=>:one_to_one, :foreign_key=>:order_id}]
+    )
+    assert result.kind_of?(Array)
+    assert_equal 3, result.size
+    assert_equal 5, result[0].keys.size
+    3.times do |i|
+      parent = result[i]
+      childs = parent[:order_details]
+      assert childs.kind_of?(Array)
+      assert_equal 5, childs.size
+      childs.each_with_index do |child, j|
+        assert_equal i+1, child[:order_id]
+        assert_equal j, child[:no]
+      end
+      assert_not_nil parent[:sum_no] 
+    end
+    #pp result
+  end
+
+  def test_one_to_one
+    result = @reader.select(
+        :sql=>"select * from orders; select order_id, sum(no) sum_no from order_details group by order_id",
+        :primary_key=>:id,
+        :relations=>[{:type=>:one_to_one, :name=>:sum, :foreign_key=>:order_id}]
+    )
+    #pp result
+    assert_equal 4, result[0].keys.size
+    assert result[0][:sum].kind_of?(Hash)
+  end
+
+  private 
+
+  def create_tables
+    @reader.create_table "dbo", "people", 
+      "id int identity, first_name varchar(255), last_name varchar(255)"
+    @reader.execute "truncate table dbo.people"
+    @reader.insert "dbo.people", {:first_name=>"Sasa", :last_name=>"Juric"}
+    @reader.insert "dbo.people", {:first_name=>"Goran", :last_name=>"Pizent"}
+
+    @reader.create_table "dbo", "orders",
+      "id int identity, created datetime, amount money"
+    @reader.create_table "dbo", "order_details",
+      "id int identity, order_id int, no int" 
+    @reader.execute "truncate table dbo.orders"
+    @reader.execute "truncate table dbo.order_details"
+    3.times do |o|
+      id = @reader.insert "orders", {:created=>Time.now, :amount=>1234.5 * (o + 1)}
+      5.times{|no| @reader.insert "order_details", {:order_id=>id, :no=>no}}
+    end
+    @reader.execute "drop procedure all_orders"
+    @reader.execute <<-SQL
+      create procedure all_orders as
+        select * from orders
+        select order_id parent_id, * from order_details
+    SQL
+  end
+
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: minus5_mssql
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 23
   prerelease: 
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Igor Anic
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-31 00:00:00 +02:00
+date: 2011-07-05 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -45,7 +45,9 @@ extra_rdoc_files: []
 files: 
 - lib/minus5_mssql.rb
 - lib/minus5_mssql/adapter.rb
+- test/helper.rb
 - test/test_db_mirroring.rb
+- test/test_select.rb
 - Rakefile
 has_rdoc: true
 homepage: http://www.minus5.hr