kotovalexarian-likes-github/rails--rails
1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
Code Releases Activity

Fixed the MS SQL adapter to work with the new limit/offset approach and with binary data (still suffering from 7KB limit, though) #901 [delynnb]

Browse source 
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@982 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
This commit is contained in:
David Heinemeier Hansson 2005-03-23 00:56:13 +00:00
parent 4043b5cdb3
commit f30b7a31bd
7 changed files with 185 additions and 184 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
activerecord/CHANGELOG
View file

@ -1,3 +1,8 @@
*SVN*
* Fixed the MS SQL adapter to work with the new limit/offset approach and with binary data (still suffering from 7KB limit, though) #901 [delynnb]
*1.9.0* (22th March, 2005)
* Added adapter independent limit clause as a two-element array with the first being the limit, the second being the offset #795 [Sam Stephenson]. Example:

260
activerecord/lib/active_record/connection_adapters/sqlserver_adapter.rb
View file

@ -5,6 +5,27 @@ require 'active_record/connection_adapters/abstract_adapter'
# Author: Joey Gibson <joey@joeygibson.com>
# Date: 10/14/2004
#
# Modifications: DeLynn Berry <delynnb@megastarfinancial.com>
# Date: 3/22/2005
#
# This adapter will ONLY work on Windows systems, since it relies on Win32OLE, which,
# to my knowledge, is only available on Window.
#
# It relies on the ADO support in the DBI module. If you are using the
# one-click installer of Ruby, then you already have DBI installed, but
# the ADO module is *NOT* installed. You will need to get the latest
# source distribution of Ruby-DBI from http://ruby-dbi.sourceforge.net/
# unzip it, and copy the file <tt>src/lib/dbd_ado/ADO.rb</tt> to
# <tt>X:/Ruby/lib/ruby/site_ruby/1.8/DBD/ADO/ADO.rb</tt> (you will need to create
# the ADO directory). Once you've installed that file, you are ready to go.
#
# Options:
#
# * <tt>:host</tt> -- Defaults to localhost
# * <tt>:username</tt> -- Defaults to sa
# * <tt>:password</tt> -- Defaults to nothing
# * <tt>:database</tt> -- The name of the database. No default, must be provided.
#
# I have tested this code on a WindowsXP Pro SP1 system,
# ruby 1.8.2 (2004-07-29) [i386-mswin32], SQL Server 2000.
#
@ -19,7 +40,7 @@ module ActiveRecord
username = config[:username] ? config[:username].to_s : 'sa'
password = config[:password].to_s
if config.has_key? :database
if config.has_key?(:database)
database = config[:database]
else
raise ArgumentError, "No database specified. Missing argument: database."
@ -38,92 +59,110 @@ module ActiveRecord
def initialize(name, default, sql_type = nil, is_identity = false, scale_value = 0)
super(name, default, sql_type)
@scale = scale_value
@identity = is_identity
end
def binary_to_string(value)
value
end
def string_to_binary(value)
value
end
def simplified_type(field_type)
case field_type
when /int/i
:integer
when /float|double|decimal|numeric/i
if @scale == 0
:integer
else
:float
nil
end
when /datetime/i
:datetime
when /timestamp/i
:timestamp
when /time/i
:time
when /date/i
:date
when /clob|text|ntext/i
:text
when /blob|binary|image/i
:binary
when /char|string/i
:string
when /boolean|bit/i
:boolean
when /int|bigint|smallint|tinyint/i : :integer
when /float|double|decimal|money|numeric|real|smallmoney/i : @scale == 0 ? :integer : :float
when /datetime|smalldatetime/i : :datetime
when /timestamp/i : :timestamp
when /time/i : :time
when /text|ntext/i : :text
when /binary|image|varbinary/i : :binary
when /char|nchar|nvarchar|string|varchar/i : :string
when /bit/i : :boolean
end
end
def string_to_time(string)
return string if string.is_a?(Time)
time_array = ParseDate.parsedate(string, true)
time_array.each_index do |i|
case i
when 0
time_array[i] = time_array[i].nil? ? "2000" : time_array[i].to_s
when 1
time_array[i] = time_array[i].nil? ? "Jan" : time_array[i].to_s
when 2
time_array[i] = time_array[i].nil? ? "1" : time_array[i].to_s
when 3
time_array[i] = time_array[i].nil? ? "0" : time_array[i].to_s
when 4
time_array[i] = time_array[i].nil? ? "0" : time_array[i].to_s
when 5
time_array[i] = time_array[i].nil? ? "0" : time_array[i].to_s
def type_cast(value)
return nil if value.nil? || value =~ /^\s*null\s*$/i
case type
when :string then value
when :integer then value == true || value == false ? value == true ? '1' : '0' : value.to_i
when :float then value.to_f
when :datetime then cast_to_date_or_time(value)
when :timestamp then cast_to_time(value)
when :time then cast_to_time(value)
else value
end
end
def cast_to_date_or_time(value)
return value if value.is_a?(Date)
guess_date_or_time (value.is_a?(Time)) ? value : cast_to_time(value)
end
def cast_to_time(value)
return value if value.is_a?(Time)
time_array = ParseDate.parsedate value
time_array[0] ||= 2000; time_array[1] ||= 1; time_array[2] ||= 1;
Time.send Base.default_timezone, *time_array
end
def guess_date_or_time(value)
(value.hour == 0 and value.min == 0 and value.sec == 0) ?
Date.new(value.year, value.month, value.day) : value
end
# These methods will only allow the adapter to insert binary data with a length of 7K or less
# because of a SQL Server statement length policy.
def string_to_binary(value)
value.gsub(/(\r|\n|\0|\x1a)/) do
case $1
when "\r"
"%00"
when "\n"
"%01"
when "\0"
"%02"
when "\x1a"
"%03"
end
end
end
def binary_to_string(value)
value.gsub(/(%00|%01|%02|%03)/) do
case $1
when "%00"
"\r"
when "%01"
"\n"
when "%02\0"
"\0"
when "%03"
"\x1a"
end
end
# treat 0000-00-00 00:00:00 as nil
Time.send(Base.default_timezone, *time_array) rescue nil
end
end
# This adapter will ONLY work on Windows systems, since it relies on Win32OLE, which,
# to my knowledge, is only available on Window.
#
# It relies on the ADO support in the DBI module. If you are using the
# one-click installer of Ruby, then you already have DBI installed, but
# the ADO module is *NOT* installed. You will need to get the latest
# source distribution of Ruby-DBI from http://ruby-dbi.sourceforge.net/
# unzip it, and copy the file <tt>src/lib/dbd_ado/ADO.rb</tt> to
# <tt>X:/Ruby/lib/ruby/site_ruby/1.8/DBD/ADO/ADO.rb</tt> (you will need to create
# the ADO directory). Once you've installed that file, you are ready to go.
#
# Options:
#
# * <tt>:host</tt> -- Defaults to localhost
# * <tt>:username</tt> -- Defaults to sa
# * <tt>:password</tt> -- Defaults to nothing
# * <tt>:database</tt> -- The name of the database. No default, must be provided.
class SQLServerAdapter < AbstractAdapter
def native_database_types
{
:primary_key => "int NOT NULL IDENTITY(1, 1) PRIMARY KEY",
:string => { :name => "varchar(255)" },
:text => { :name => "text(16)" },
:integer => { :name => "int(4)", :limit => 11 },
:float => { :name => "float(8)" },
:datetime => { :name => "datetime(8)" },
:timestamp => { :name => "datetime(8)" },
:time => { :name => "datetime(8)" },
:date => { :name => "datetime(8)" },
:binary => { :name => "image(16)" },
:boolean => { :name => "bit(1)" }
}
end
def adapter_name
'SQLServer'
end
def select_all(sql, name = nil)
add_limit!(sql, nil)
select(sql, name)
@ -136,36 +175,22 @@ module ActiveRecord
end
def columns(table_name, name = nil)
sql = <<EOL
SELECT
COLUMN_NAME as ColName,
COLUMN_DEFAULT as DefaultValue,
DATA_TYPE as ColType,
COL_LENGTH('#{table_name}', COLUMN_NAME) as Length,
COLUMNPROPERTY(OBJECT_ID('#{table_name}'), COLUMN_NAME, 'IsIdentity') as IsIdentity,
NUMERIC_SCALE as Scale
FROM INFORMATION_SCHEMA.columns
WHERE TABLE_NAME = '#{table_name}'
EOL
sql = "SELECT COLUMN_NAME as ColName, COLUMN_DEFAULT as DefaultValue, DATA_TYPE as ColType, COL_LENGTH('#{table_name}', COLUMN_NAME) as Length, COLUMNPROPERTY(OBJECT_ID('#{table_name}'), COLUMN_NAME, 'IsIdentity') as IsIdentity, NUMERIC_SCALE as Scale FROM INFORMATION_SCHEMA.columns WHERE TABLE_NAME = '#{table_name}'"
result = nil
# Uncomment if you want to have the Columns select statment logged.
# Personnally, I think it adds unneccessary SQL statement bloat to the log.
# Personnally, I think it adds unneccessary bloat to the log.
# If you do uncomment, make sure to comment the "result" line that follows
log(sql, name, @connection) { |conn| result = conn.select_all(sql) }
#result = @connection.select_all(sql)
columns = []
result.each { |field| columns << ColumnWithIdentity.new(field[:ColName], field[:DefaultValue].to_s.gsub!(/[()\']/,"") =~ /null/ ? nil : field[:DefaultValue], "#{field[:ColType]}(#{field[:Length]})", field[:IsIdentity] == 1 ? true : false, field[:Scale]) }
columns
end
def insert(sql, name = nil, pk = nil, id_value = nil)
begin
table_name = get_table_name(sql)
col = get_identity_column(table_name)
ii_enabled = false
if col != nil
@ -181,14 +206,12 @@ EOL
log(sql, name, @connection) do |conn|
conn.execute(sql)
select_one("SELECT @@IDENTITY AS Ident")["Ident"]
end
ensure
if ii_enabled
begin
execute enable_identity_insert(table_name, false)
rescue Exception => e
# Couldn't turn off IDENTITY_INSERT
end
@ -196,11 +219,24 @@ EOL
end
end
def update(sql, name = nil)
execute(sql, name)
affected_rows(name)
def execute(sql, name = nil)
if sql =~ /^INSERT/i
insert(sql, name)
elsif sql =~ /^UPDATE|DELETE/i
log(sql, name, @connection) do |conn|
conn.execute(sql)
retVal = select_one("SELECT @@ROWCOUNT AS AffectedRows")["AffectedRows"]
end
else
log(sql, name, @connection) do |conn|
conn.execute(sql)
end
end
end
def update(sql, name = nil)
execute(sql, name)
end
alias_method :delete, :update
def begin_db_transaction
@ -236,8 +272,8 @@ EOL
"'#{quote_string(value)}'"
end
when NilClass then "NULL"
when TrueClass then (column && column.type == :boolean ? "'t'" : "1")
when FalseClass then (column && column.type == :boolean ? "'f'" : "0")
when TrueClass then '1'
when FalseClass then '0'
when Float, Fixnum, Bignum then value.to_s
when Date then "'#{value.to_s}'"
when Time, DateTime then "'#{value.strftime("%Y-%m-%d %H:%M:%S")}'"
@ -254,11 +290,12 @@ EOL
end
def add_limit_with_offset!(sql, limit, offset)
raise ArgumentError, 'add_limit_with_offset! not implemented'
order_by = sql.include?("ORDER BY") ? get_order_by(sql.sub(/.*ORDER\sBY./, "")) : nil
sql.gsub!(/SELECT/i, "SELECT * FROM ( SELECT TOP #{limit} * FROM ( SELECT TOP #{limit + offset}")<<" ) AS tmp1 ORDER BY #{order_by[1]} ) AS tmp2 ORDER BY #{order_by[0]}"
end
def add_limit_without_offset!(sql, limit)
raise ArgumentError, 'add_limit_without_offset! not implemented'
limit.nil? ? sql : sql.gsub!(/SELECT/i, "SELECT TOP #{limit}")
end
def recreate_database(name)
@ -274,36 +311,18 @@ EOL
execute "CREATE DATABASE #{name}"
end
def execute(sql, name = nil)
if sql =~ /^INSERT/i
insert(sql, name)
else
log(sql, name, @connection) do |conn|
conn.execute(sql)
end
end
end
def adapter_name()
'SqlServer'
end
private
def select(sql, name = nil)
rows = []
log(sql, name, @connection) do |conn|
conn.select_all(sql) do |row|
record = {}
row.column_names.each do |col|
record[col] = row[col]
end
rows << record
end
end
rows
end
@ -346,10 +365,6 @@ EOL
return sql =~ /[\(\.\,]\s*#{col}/
end
def query_contains_text_column(sql, col)
end
def get_order_by(sql)
return sql, sql.gsub(/\s*DESC\s*/, "").gsub(/\s*ASC\s*/, " DESC")
end
@ -358,15 +373,6 @@ EOL
limit = limit.gsub!(/.OFFSET./i, ",").split(',')
return limit[0].to_i, limit[0].to_i+limit[1].to_i
end
def affected_rows(name = nil)
sql = "SELECT @@ROWCOUNT AS AffectedRows"
log(sql, name, @connection) do |conn|
conn.select_all(sql) do |row|
return row[:AffectedRows].to_i
end
end
end
end
end
end

2
activerecord/test/column_alias_test.rb
View file

@ -11,7 +11,7 @@ class TestColumnAlias < Test::Unit::TestCase
assert_equal(records[0].keys[0], "pk")
end
else
records = topic.connection.select_all("SELECT id AS pk FROM topics LIMIT 1")
records = topic.connection.select_all("SELECT id AS pk FROM topics")
assert_equal(records[0].keys[0], "pk")
end
end

3
activerecord/test/connections/native_sqlserver/connection.rb
View file

@ -4,6 +4,9 @@ require 'logger'
ActiveRecord::Base.logger = Logger.new("debug.log")
db1 = 'activerecord_unittest'
db2 = 'activerecord_unittest2'
ActiveRecord::Base.establish_connection(
:adapter => "sqlserver",
:host => "localhost",

7
activerecord/test/finder_test.rb
View file

@ -2,6 +2,7 @@ require 'abstract_unit'
require 'fixtures/company'
require 'fixtures/topic'
require 'fixtures/entrant'
require 'fixtures/developer'
class FinderTest < Test::Unit::TestCase
fixtures :companies, :topics, :entrants, :developers
@ -43,10 +44,10 @@ class FinderTest < Test::Unit::TestCase
def test_find_all_with_prepared_limit_and_offset
if ActiveRecord::ConnectionAdapters.const_defined? :OracleAdapter
if ActiveRecord::Base.connection.instance_of?(ActiveRecord::ConnectionAdapters::OracleAdapter)
assert_raises(ArgumentError) { Entrant.find_all nil, "id ASC", ["? OFFSET ?", 2, 1] }
assert_raises(ArgumentError) { Entrant.find_all nil, "id ASC", [2, 1] }
end
else
entrants = Entrant.find_all nil, "id ASC", ["? OFFSET ?", 2, 1]
entrants = Entrant.find_all nil, "id ASC", [2, 1]
assert_equal(2, entrants.size)
assert_equal(@entrants["second"]["name"], entrants.first.name)
@ -260,7 +261,7 @@ class FinderTest < Test::Unit::TestCase
def test_find_all_with_limit_and_offset
first_three_developers = Developer.find_all nil, 'id ASC', [3, 0]
second_three_developers = Developer.find_all nil, 'id ASC', [3, 3]
last_two_developers = Developer.find_all nil, 'id ASC', [3, 8]
last_two_developers = Developer.find_all nil, 'id ASC', [2, 8]
assert_equal 3, first_three_developers.length
assert_equal 3, second_three_developers.length

1
activerecord/test/fixtures/db_definitions/sqlserver.drop.sql vendored
View file

@ -15,4 +15,3 @@ DROP TABLE mixins;
DROP TABLE people;
DROP TABLE binaries;
DROP TABLE computers;

79
activerecord/test/fixtures/db_definitions/sqlserver.sql vendored
View file

@ -1,88 +1,79 @@
CREATE TABLE accounts (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
firm_id int default NULL,
credit_limit int default NULL,
PRIMARY KEY (id)
credit_limit int default NULL
)
CREATE TABLE companies (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
type varchar(50) default NULL,
ruby_type varchar(50) default NULL,
firm_id int default NULL,
name varchar(50) default NULL,
client_of int default NULL,
companies_count int default 0,
rating int default 1,
PRIMARY KEY (id)
rating int default 1
)
CREATE TABLE topics (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
title varchar(255) default NULL,
author_name varchar(255) default NULL,
author_email_address varchar(255) default NULL,
written_on datetime default NULL,
bonus_time datetime default NULL,
last_read datetime default NULL,
content text,
approved tinyint default 1,
replies_count int default 0,
parent_id int default NULL,
type varchar(50) default NULL,
PRIMARY KEY (id)
type varchar(50) default NULL
)
CREATE TABLE developers (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
name varchar(100) default NULL,
PRIMARY KEY (id)
salary int default 70000
);
CREATE TABLE projects (
id int NOT NULL IDENTITY(1, 1),
name varchar(100) default NULL,
salary int default 70000,
PRIMARY KEY (id)
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
name varchar(100) default NULL
);
CREATE TABLE developers_projects (
developer_id int NOT NULL,
project_id int NOT NULL
project_id int NOT NULL,
joined_on datetime default NULL
);
CREATE TABLE customers (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
name varchar(100) default NULL,
balance int default 0,
address_street varchar(100) default NULL,
address_city varchar(100) default NULL,
address_country varchar(100) default NULL,
gps_location varchar(100) default NULL,
PRIMARY KEY (id)
gps_location varchar(100) default NULL
);
CREATE TABLE movies (
movieid int NOT NULL IDENTITY(1, 1),
name varchar(100) default NULL,
PRIMARY KEY (movieid)
movieid int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
name varchar(100) default NULL
);
CREATE TABLE subscribers (
nick varchar(100) NOT NULL,
name varchar(100) default NULL,
PRIMARY KEY (nick)
nick varchar(100) NOT NULL PRIMARY KEY,
name varchar(100) default NULL
);
CREATE TABLE booleantests (
id int NOT NULL IDENTITY(1, 1),
value integer default NULL,
PRIMARY KEY (id)
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
value bit default NULL
);
CREATE TABLE auto_id_tests (
auto_id int NOT NULL IDENTITY(1, 1),
value int default NULL,
PRIMARY KEY (auto_id)
auto_id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
value int default NULL
);
CREATE TABLE entrants (
@ -92,22 +83,20 @@ CREATE TABLE entrants (
);
CREATE TABLE colnametests (
id int NOT NULL IDENTITY(1, 1),
[references] int NOT NULL,
PRIMARY KEY (id)
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
[references] int NOT NULL
);
CREATE TABLE mixins (
id int NOT NULL IDENTITY(1, 1),
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
parent_id int default NULL,
type varchar(40) default NULL,
pos int default NULL,
created_at datetime default NULL,
updated_at datetime default NULL,
lft int default NULL,
rgt int default NULL,
root_id int default NULL,
created_at datetime default NULL,
updated_at datetime default NULL,
PRIMARY KEY (id)
type varchar(40) default NULL
);
CREATE TABLE people (
@ -118,14 +107,12 @@ CREATE TABLE people (
);
CREATE TABLE binaries (
id int NOT NULL IDENTITY(1, 1),
data image NULL,
PRIMARY KEY (id)
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
data image NULL
);
CREATE TABLE computers (
id int NOT NULL IDENTITY(1, 1),
developer int NOT NULL,
PRIMARY KEY (id)
id int NOT NULL IDENTITY(1, 1) PRIMARY KEY,
developer int NOT NULL
);