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
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#-- vim:sw=2:et
#++
#
# :title: Sqlite3 registry implementation
#
require 'sqlite3'
module Irc
class Bot
class Registry
class SqliteAccessor < AbstractAccessor
def initialize(filename)
super filename + '.db'
end
def registry
super
unless @registry
@registry = SQLite3::Database.new(@filename)
begin
@registry.execute('SELECT COUNT(*) FROM data')
rescue
@registry.execute('CREATE TABLE data (key PRIMARY KEY, value)')
end
end
@registry
end
def optimize
return unless @registry
@registry.execute('VACUUM')
end
def [](key)
if dbexists?
begin
value = registry.get_first_row('SELECT value FROM data WHERE key = ?', key.to_s)
return restore(value.first)
rescue
return default
end
else
return default
end
end
def []=(key,value)
value = SQLite3::Blob.new(store(value))
if has_key? key
registry.execute('UPDATE data SET value = ? WHERE key = ?', value, key.to_s)
else
registry.execute('INSERT INTO data VALUES (?, ?)', key.to_s, value)
end
end
def each(&block)
return nil unless dbexists?
res = registry.execute('SELECT * FROM data')
res.each do |row|
key, value = row
block.call(key, restore(value))
end
end
alias each_pair each
def has_key?(key)
return nil unless dbexists?
res = registry.get_first_row('SELECT COUNT(*) FROM data WHERE key = ?', key.to_s)
return res.first > 0
end
alias include? has_key?
alias member? has_key?
alias key? has_key?
def has_value?(value)
return nil unless dbexists?
value = SQLite3::Blob.new(store(value))
res = registry.get_first_row('SELECT COUNT(*) FROM data WHERE value = ?', value)
return res.first > 0
end
def delete(key)
return default unless dbexists?
begin
value = self[key]
registry.execute('DELETE FROM data WHERE key = ?', key.to_s)
value if registry.changes > 0
rescue
nil
end
end
# returns a list of your keys
def keys
return [] unless dbexists?
res = registry.execute('SELECT key FROM data')
res.map { |row| row.first }
end
def clear
return unless dbexists?
registry.execute('DELETE FROM data')
end
alias truncate clear
# returns the number of keys in your registry namespace
def length
return 0 unless dbexists?
res = registry.get_first_row('SELECT COUNT(key) FROM data')
res.first
end
alias size length
end
end # Registry
end # Bot
end # Irc
|
