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
|
# require 'adapter_lib'
module Sequel
module Adapter
class Database < Sequel::Database
set_adapter_scheme :adapter
def connect
AdapterDB.new(@opts[:database], @opts[:user], @opts[:password])
end
def disconnect
@pool.disconnect {|c| c.disconnect}
end
def dataset(opts = nil)
Adapter::Dataset.new(self, opts)
end
def execute(sql)
@logger.info(sql) if @logger
@pool.hold {|conn| conn.exec(sql)}
end
alias_method :do, :execute
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
cursor = @db.execute sql
begin
@columns = cursor.get_col_names.map {|c| c.to_sym}
while r = cursor.fetch
row = {}
r.each_with_index {|v, i| row[@columns[i]] = v}
yield row
end
ensure
cursor.close
end
end
self
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
|
