2010-02-02 00:52:25 +01:00
|
|
|
require 'bdb'
|
|
|
|
require 'sbdb/cursor'
|
|
|
|
|
|
|
|
module SBDB
|
2010-02-06 11:50:52 +01:00
|
|
|
TYPES = []
|
2010-02-02 00:52:25 +01:00
|
|
|
class DB
|
|
|
|
UNKNOWN = Bdb::Db::UNKNOWN
|
|
|
|
BTREE = Bdb::Db::BTREE
|
|
|
|
HASH = Bdb::Db::HASH
|
|
|
|
QUEUE = Bdb::Db::QUEUE
|
|
|
|
ARRAY = RECNO = Bdb::Db::RECNO
|
2010-02-02 18:12:11 +01:00
|
|
|
RDONLY = READLONY = Bdb::DB_RDONLY
|
2010-02-05 19:08:30 +01:00
|
|
|
CONSUME = Bdb::DB_CONSUME
|
|
|
|
CONSUME_WAIT = Bdb::DB_CONSUME_WAIT
|
2010-02-02 00:52:25 +01:00
|
|
|
|
|
|
|
attr_reader :home
|
2010-03-09 19:09:22 +01:00
|
|
|
attr_accessor :txn
|
2010-02-02 00:52:25 +01:00
|
|
|
include Enumerable
|
|
|
|
def bdb_object() @db end
|
|
|
|
def sync() @db.sync end
|
|
|
|
def close( f = nil) @db.close f || 0 end
|
|
|
|
def cursor( &e) Cursor.new self, &e end
|
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def at k, txn = nil
|
|
|
|
@db.get _txn(txn), k.nil? ? nil : k.to_s, nil, 0
|
2010-02-23 19:39:27 +01:00
|
|
|
rescue Bdb::KeyEmpty
|
|
|
|
return nil
|
2010-02-06 00:41:43 +01:00
|
|
|
end
|
2010-03-11 00:18:38 +01:00
|
|
|
alias [] at
|
2010-02-06 00:41:43 +01:00
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def put k, v, txn = nil
|
2010-02-05 19:08:30 +01:00
|
|
|
if v.nil?
|
2010-03-11 00:18:38 +01:00
|
|
|
@db.del _txn(txn), k.to_s, 0
|
2010-02-05 19:08:30 +01:00
|
|
|
else
|
2010-03-11 00:18:38 +01:00
|
|
|
@db.put _txn(txn), k.nil? ? nil : k.to_s, v.to_s, 0
|
2010-02-05 19:08:30 +01:00
|
|
|
end
|
|
|
|
end
|
2010-03-11 00:18:38 +01:00
|
|
|
|
|
|
|
def []= k, v
|
|
|
|
put k, v
|
|
|
|
end
|
|
|
|
|
|
|
|
def delete k, txn = nil
|
|
|
|
@db.del _txn(txn), k.to_s
|
|
|
|
end
|
|
|
|
alias del delete
|
2010-02-05 19:08:30 +01:00
|
|
|
|
2010-02-02 00:52:25 +01:00
|
|
|
class << self
|
|
|
|
def new *p, &e
|
|
|
|
x = super *p
|
|
|
|
return x unless e
|
|
|
|
begin e.call x
|
|
|
|
ensure
|
2010-03-11 00:18:38 +01:00
|
|
|
begin x.sync
|
|
|
|
rescue Object
|
|
|
|
$stderr.puts [$!.class,$!,$!.backtrace].inspect
|
|
|
|
end
|
|
|
|
begin x.close
|
|
|
|
rescue Object
|
|
|
|
$stderr.puts [$!.class,$!,$!.backtrace].inspect
|
|
|
|
end
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
alias open new
|
|
|
|
end
|
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def _txn t
|
|
|
|
t ||= @txn
|
|
|
|
t && t.bdb_object
|
|
|
|
end
|
|
|
|
|
2010-02-23 19:39:27 +01:00
|
|
|
def initialize file, *args
|
2010-02-23 20:20:26 +01:00
|
|
|
opts = ::Hash === args.last ? args.pop : {}
|
|
|
|
opts = {:name => args[0], :type => args[1], :flags => args[2], :mode => args[3], :env => args[4]}.update opts
|
2010-02-06 11:47:41 +01:00
|
|
|
#type = BTREE if type == UNKNOWN and (flags & CREATE) == CREATE
|
2010-02-23 19:39:27 +01:00
|
|
|
@home, @db = opts[:env], opts[:env] ? opts[:env].bdb_object.db : Bdb::Db.new
|
2010-02-23 20:20:26 +01:00
|
|
|
opts[:type] = TYPES.index(self.class) || UNKNOWN
|
|
|
|
@db.re_len = opts[:re_len] if opts[:re_len]
|
2010-03-11 00:18:38 +01:00
|
|
|
txn = opts[:txn]
|
2010-03-07 22:01:05 +01:00
|
|
|
begin
|
2010-03-09 19:09:22 +01:00
|
|
|
@db.open txn && txn.bdb_object, file, opts[:name], opts[:type], opts[:flags] || 0, opts[:mode] || 0
|
2010-02-02 00:52:25 +01:00
|
|
|
rescue Object
|
|
|
|
close
|
|
|
|
raise $!
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def each k = nil, v = nil, &e
|
|
|
|
cursor{|c|c.each k, v, &e}
|
|
|
|
end
|
|
|
|
|
|
|
|
def reverse k = nil, v = nil, &e
|
|
|
|
cursor{|c|c.reverse k, v, &e}
|
|
|
|
end
|
|
|
|
|
|
|
|
def to_hash k = nil, v = nil
|
|
|
|
h = {}
|
|
|
|
each( k, v) {|k, v| h[ k] = v }
|
|
|
|
h
|
|
|
|
end
|
2010-03-11 00:18:38 +01:00
|
|
|
|
|
|
|
def truncate txn = nil
|
|
|
|
@db.truncate _txn(txn)
|
|
|
|
end
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|
|
|
|
|
|
|
|
class Unknown < DB
|
2010-02-23 19:39:27 +01:00
|
|
|
def self.new file, *p, &e
|
|
|
|
dbt = super( file, *p) {|db| db.bdb_object.get_type }
|
|
|
|
TYPES[dbt] ? TYPES[dbt].new( file, *p, &e) : super( file, *p, &e)
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
class Btree < DB
|
|
|
|
end
|
2010-02-06 11:50:52 +01:00
|
|
|
TYPES[DB::BTREE] = Btree
|
2010-02-02 00:52:25 +01:00
|
|
|
|
|
|
|
class Hash < DB
|
|
|
|
end
|
2010-02-06 11:50:52 +01:00
|
|
|
TYPES[DB::HASH] = Hash
|
2010-02-02 00:52:25 +01:00
|
|
|
|
|
|
|
class Recno < DB
|
2010-02-02 15:15:10 +01:00
|
|
|
def [] k
|
|
|
|
super [k].pack('I')
|
|
|
|
end
|
|
|
|
|
|
|
|
def []= k, v
|
|
|
|
super [k].pack('I'), v
|
|
|
|
end
|
2010-02-22 16:17:25 +01:00
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def push v, txn = nil
|
|
|
|
@db.put _txn(txn), "\0\0\0\0", v, Bdb::DB_APPEND
|
2010-02-22 16:17:25 +01:00
|
|
|
end
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|
|
|
|
Array = Recno
|
2010-02-06 11:50:52 +01:00
|
|
|
TYPES[DB::RECNO] = Recno
|
2010-02-02 00:52:25 +01:00
|
|
|
|
|
|
|
class Queue < DB
|
2010-02-02 15:15:10 +01:00
|
|
|
def [] k
|
|
|
|
super [k].pack('I')
|
|
|
|
end
|
|
|
|
|
|
|
|
def []= k, v
|
|
|
|
super [k].pack('I'), v
|
|
|
|
end
|
2010-02-05 19:08:30 +01:00
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def unshift txn = nil
|
|
|
|
@db.get _txn(txn), "\0\0\0\0", nil, Bdb::DB_CONSUME
|
2010-02-05 19:08:30 +01:00
|
|
|
end
|
2010-02-22 16:17:25 +01:00
|
|
|
|
2010-03-11 00:18:38 +01:00
|
|
|
def push v, txn = nil
|
|
|
|
@db.put _txn(txn), "\0\0\0\0", v, Bdb::DB_APPEND
|
2010-02-22 16:17:25 +01:00
|
|
|
end
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|
2010-02-06 11:50:52 +01:00
|
|
|
TYPES[DB::QUEUE] = Queue
|
2010-02-02 00:52:25 +01:00
|
|
|
end
|