bdb/test.rb

131 lines
2.8 KiB
Ruby
Raw Normal View History

2006-02-14 03:56:36 +01:00
#!/usr/bin/env ruby
require 'bdb2'
50.times {|n|
db=Bdb::Db.new
db.open(nil,"dbtest.db",nil,Bdb::Db::BTREE,Bdb::DB_CREATE,0)
db.put(nil,n.to_s,"ploppy #{n} #{Time.now}",0)
2006-02-14 03:56:36 +01:00
db.close(0)
}
db=Bdb::Db.new
db.open(nil,"dbtest.db",nil,nil,nil,0)
2006-02-14 03:56:36 +01:00
500.times {|n|
db.put(nil,n.to_s,"ploppy #{n}",0)
2006-02-14 03:56:36 +01:00
}
db.close(0)
db=Bdb::Db.new
db.open(nil,"dbtest.db",nil,nil,nil,0)
dbc=db.cursor(nil,0)
2006-02-14 03:56:36 +01:00
puts("cursor is: "+dbc.inspect.to_s)
kv=dbc.get(nil,nil,Bdb::DB_FIRST);
puts("first data is: " + kv.inspect.to_s)
kv=dbc.get(nil,nil,Bdb::DB_LAST);
puts("last data is: " + kv.inspect.to_s)
kv=dbc.get(nil,nil,Bdb::DB_PREV);
puts("prior data is: " + kv.inspect.to_s)
dbc.del;
begin
kv=dbc.get(nil,nil,Bdb::DB_CURRENT);
rescue Bdb::DbError => m
puts("deleted record is gone from current position:" + m.to_s)
puts("code is #{m.code}")
end
puts("current data is: " +kv.inspect.to_s)
dbc.put("elephant","gorilla",Bdb::DB_KEYFIRST)
kv=dbc.get(nil,nil,Bdb::DB_CURRENT);
puts("current data after put: " +kv.inspect.to_s)
kv=dbc.get(nil,nil,Bdb::DB_LAST);
puts("last data is: " + kv.inspect.to_s)
kv=dbc.get(nil,nil,Bdb::DB_PREV);
puts("prior data is: " + kv.inspect.to_s)
puts("duplicates here is: " + dbc.count.to_s)
dbc.close
db.close(0)
db=Bdb::Db.new
db.open(nil,"dbtest.db",nil,nil,nil,0)
2006-02-14 03:56:36 +01:00
5.times {|n|
$stdout.puts(db.get(nil,n.to_s,nil,0))
2006-02-14 03:56:36 +01:00
begin
db.del(n.to_s,0)
rescue
end
}
5.times {|n|
v=db.get(nil,n.to_s,nil,0)
2006-02-14 03:56:36 +01:00
if v
$stdout.puts("For #{n}:" + v)
else
$stdout.puts("-- not in database #{n}")
end
}
db.close(0)
$stderr.puts("All OK!")
File.delete('db1.db') if File.exist?('db1.db')
File.delete('db2.db') if File.exist?('db2.db')
db=Bdb::Db.new
db.open(nil,"db1.db",nil,Bdb::Db::HASH,Bdb::DB_CREATE,0)
2006-02-14 03:56:36 +01:00
db2=Bdb::Db.new
db2.flags=Bdb::DB_DUPSORT
db2.open(nil,"db2.db",nil,Bdb::Db::HASH,Bdb::DB_CREATE,0)
2006-02-14 03:56:36 +01:00
db.associate(nil,db2,0,
2006-02-14 03:56:36 +01:00
proc {|sdb,key,data|
key.split('-')[0]
})
def write_data(db)
c=0
IO.popen("gzip -dc skus.gz") {|fd|
tlen=fd.stat.size
pf=tlen/10
pl=0
fd.each do |line|
c+=1
if c%1000==0
$stderr.print('.')
2006-02-14 03:56:36 +01:00
end
line.chomp!
n=line*50 # multiply the amount of data written
isbn,item=line.split('|')[0..1]
sku="%s-%03d"%[isbn,item]
db.put(nil,sku,line,0)
2006-02-14 03:56:36 +01:00
end
} if File.exist?("skus.gz")
$stderr.print("\ntotal count: #{c}\n")
end
write_data(db)
2006-02-14 03:56:36 +01:00
db2.close(0)
db.close(0)
$stderr.puts("test environment")
env=Bdb::Env.new(0)
2006-02-14 03:56:36 +01:00
env.cachesize=25*1024*1024;
env.open(".",Bdb::DB_INIT_CDB|Bdb::DB_INIT_MPOOL|Bdb::DB_CREATE,0)
db=env.db
db.open(nil,"db1.db",nil,Bdb::Db::HASH,Bdb::DB_CREATE,0)
2006-02-14 03:56:36 +01:00
db2=env.db
db2.flags=Bdb::DB_DUPSORT
db2.open(nil,"db2.db",nil,Bdb::Db::HASH,Bdb::DB_CREATE,0)
2006-02-14 03:56:36 +01:00
db.associate(nil,db2,0,
2006-02-14 03:56:36 +01:00
proc {|sdb,key,data|
key.split('-')[0]
})
write_data(db)
2006-02-14 03:56:36 +01:00
db2.close(0)
db.close(0)
env.close