-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
openbase.rb
68 lines (60 loc) · 1.38 KB
/
openbase.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
require 'openbase'
module Sequel
module OpenBase
class Database < Sequel::Database
set_adapter_scheme :openbase
def connect(server)
opts = server_opts(server)
OpenBase.new(
opts[:database],
opts[:host] || 'localhost',
opts[:user],
opts[:password]
)
end
def dataset(opts = nil)
OpenBase::Dataset.new(self, opts)
end
def execute(sql, opts={})
log_info(sql)
synchronize(opts[:server]) do |conn|
r = conn.execute(sql)
yield(r) if block_given?
r
end
end
alias_method :do, :execute
private
def disconnect_connection(c)
c.disconnect
end
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)
execute(sql) do |result|
begin
@columns = result.column_infos.map{|c| output_identifier(c.name)}
result.each do |r|
row = {}
r.each_with_index {|v, i| row[@columns[i]] = v}
yield row
end
ensure
# result.close
end
end
self
end
end
end
end