-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
ado.rb
80 lines (67 loc) · 1.79 KB
/
ado.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
require 'win32ole'
module Sequel
# The ADO adapter provides connectivity to ADO databases in Windows. ADO
# databases can be opened using a URL with the ado schema:
#
# DB = Sequel.open('ado://mydb')
#
# or using the Sequel.ado method:
#
# DB = Sequel.ado('mydb')
#
module ADO
class Database < Sequel::Database
set_adapter_scheme :ado
AUTO_INCREMENT = 'IDENTITY(1,1)'.freeze
def auto_increment_sql
AUTO_INCREMENT
end
def connect
s = "driver=#{@opts[:driver] || 'SQL Server'};server=#{@opts[:host]};database=#{@opts[:database]}#{";uid=#{@opts[:user]};pwd=#{@opts[:password]}" if @opts[:user]}"
handle = WIN32OLE.new('ADODB.Connection')
handle.Open(s)
handle
end
def disconnect
@pool.disconnect {|conn| conn.Close}
end
def dataset(opts = nil)
ADO::Dataset.new(self, opts)
end
def execute(sql)
log_info(sql)
@pool.hold {|conn| conn.Execute(sql)}
end
alias_method :do, :execute
end
class Dataset < Sequel::Dataset
def literal(v)
case v
when Time
literal(v.iso8601)
when Date, DateTime
literal(v.to_s)
else
super
end
end
def fetch_rows(sql, &block)
@db.synchronize do
s = @db.execute sql
@columns = s.Fields.extend(Enumerable).map {|x| x.Name.to_sym}
unless s.eof
s.moveFirst
s.getRows.transpose.each {|r| yield hash_row(r)}
end
end
self
end
def hash_row(row)
@columns.inject({}) do |m, c|
m[c] = row.shift
m
end
end
end
end
end