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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
require 'dbi'
module Sequel
module DBI
class Database < Sequel::Database
set_adapter_scheme :dbi
DBI_ADAPTERS = {
:ado => "ADO",
:db2 => "DB2",
:frontbase => "FrontBase",
:interbase => "InterBase",
:msql => "Msql",
:mysql => "Mysql",
:odbc => "ODBC",
:oracle => "Oracle",
:pg => "Pg",
:proxy => "Proxy",
:sqlite => "SQLite",
:sqlrelay => "SQLRelay"
}
# Converts a uri to an options hash. These options are then passed
# to a newly created database object.
def self.uri_to_options(uri)
database = (uri.path =~ /\/(.*)/) && ($1)
if uri.scheme =~ /dbi-(.+)/
adapter = DBI_ADAPTERS[$1.to_sym] || $1
database = "#{adapter}:#{database}"
end
{
:user => uri.user,
:password => uri.password,
:host => uri.host,
:port => uri.port,
:database => database
}
end
def connect
dbname = @opts[:database]
dbname = 'DBI:' + dbname unless dbname =~ /^DBI:/
::DBI.connect(dbname, @opts[:user], @opts[:password])
end
def disconnect
@pool.disconnect {|c| c.disconnect}
end
def dataset(opts = nil)
DBI::Dataset.new(self, opts)
end
def execute(sql)
@logger.info(sql) if @logger
@pool.hold do |conn|
conn.execute(sql)
end
end
def do(sql)
@logger.info(sql) if @logger
@pool.hold do |conn|
conn.do(sql)
end
end
end
class Dataset < Sequel::Dataset
def literal(v)
case v
when Time
literal(v.iso8601)
else
super
end
end
def fetch_rows(sql, &block)
@db.synchronize do
s = @db.execute sql
begin
@columns = s.column_names.map {|c| c.to_sym}
s.fetch {|r| yield hash_row(s, r)}
ensure
s.finish rescue nil
end
end
self
end
def hash_row(stmt, row)
@columns.inject({}) do |m, c|
m[c] = row.shift
m
end
end
def insert(*values)
@db.do insert_sql(*values)
end
def update(*args, &block)
@db.do update_sql(*args, &block)
end
def delete(opts = nil)
@db.do delete_sql(opts)
end
end
end
end
|
