mirror of
https://github.com/berkeleydb/libdb.git
synced 2024-11-17 09:36:24 +00:00
876 lines
22 KiB
Perl
876 lines
22 KiB
Perl
|
#!./perl -w
|
||
|
|
||
|
# ID: %I%, %G%
|
||
|
|
||
|
use strict ;
|
||
|
|
||
|
use lib 't' ;
|
||
|
use BerkeleyDB;
|
||
|
use Test::More;
|
||
|
use util;
|
||
|
|
||
|
plan(skip_all => "Queue needs Berkeley DB 3.3.x or better\n" )
|
||
|
if $BerkeleyDB::db_version < 3.3;
|
||
|
|
||
|
plan tests => 257;
|
||
|
|
||
|
|
||
|
my $Dfile = "dbhash.tmp";
|
||
|
my $Dfile2 = "dbhash2.tmp";
|
||
|
my $Dfile3 = "dbhash3.tmp";
|
||
|
unlink $Dfile;
|
||
|
|
||
|
umask(0) ;
|
||
|
|
||
|
|
||
|
# Check for invalid parameters
|
||
|
{
|
||
|
# Check for invalid parameters
|
||
|
my $db ;
|
||
|
eval ' $db = new BerkeleyDB::Queue -Stupid => 3 ; ' ;
|
||
|
ok $@ =~ /unknown key value\(s\) Stupid/ ;
|
||
|
|
||
|
eval ' $db = new BerkeleyDB::Queue -Bad => 2, -Mode => 0345, -Stupid => 3; ' ;
|
||
|
ok $@ =~ /unknown key value\(s\) / ;
|
||
|
|
||
|
eval ' $db = new BerkeleyDB::Queue -Env => 2 ' ;
|
||
|
ok $@ =~ /^Env not of type BerkeleyDB::Env/ ;
|
||
|
|
||
|
eval ' $db = new BerkeleyDB::Queue -Txn => "x" ' ;
|
||
|
ok $@ =~ /^Txn not of type BerkeleyDB::Txn/ ;
|
||
|
|
||
|
my $obj = bless [], "main" ;
|
||
|
eval ' $db = new BerkeleyDB::Queue -Env => $obj ' ;
|
||
|
ok $@ =~ /^Env not of type BerkeleyDB::Env/ ;
|
||
|
}
|
||
|
|
||
|
# Now check the interface to Queue
|
||
|
|
||
|
{
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my $rec_len = 10 ;
|
||
|
my $pad = "x" ;
|
||
|
|
||
|
ok my $db = new BerkeleyDB::Queue -Filename => $Dfile,
|
||
|
-Flags => DB_CREATE,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => $pad;
|
||
|
|
||
|
# Add a k/v pair
|
||
|
my $value ;
|
||
|
my $status ;
|
||
|
is $db->Env, undef;
|
||
|
ok $db->db_put(1, "some value") == 0 ;
|
||
|
ok $db->status() == 0 ;
|
||
|
ok $db->db_get(1, $value) == 0 ;
|
||
|
ok $value eq fillout("some value", $rec_len, $pad) ;
|
||
|
ok $db->db_put(2, "value") == 0 ;
|
||
|
ok $db->db_get(2, $value) == 0 ;
|
||
|
ok $value eq fillout("value", $rec_len, $pad) ;
|
||
|
ok $db->db_put(3, "value") == 0 ;
|
||
|
ok $db->db_get(3, $value) == 0 ;
|
||
|
ok $value eq fillout("value", $rec_len, $pad) ;
|
||
|
ok $db->db_del(2) == 0 ;
|
||
|
ok $db->db_get(2, $value) == DB_KEYEMPTY ;
|
||
|
ok $db->status() == DB_KEYEMPTY ;
|
||
|
ok $db->status() =~ $DB_errors{'DB_KEYEMPTY'} ;
|
||
|
|
||
|
ok $db->db_get(7, $value) == DB_NOTFOUND ;
|
||
|
ok $db->status() == DB_NOTFOUND ;
|
||
|
ok $db->status() =~ $DB_errors{'DB_NOTFOUND'} ;
|
||
|
|
||
|
ok $db->db_sync() == 0 ;
|
||
|
|
||
|
# Check NOOVERWRITE will make put fail when attempting to overwrite
|
||
|
# an existing record.
|
||
|
|
||
|
ok $db->db_put( 1, 'x', DB_NOOVERWRITE) == DB_KEYEXIST ;
|
||
|
ok $db->status() =~ $DB_errors{'DB_KEYEXIST'} ;
|
||
|
ok $db->status() == DB_KEYEXIST ;
|
||
|
|
||
|
|
||
|
# check that the value of the key has not been changed by the
|
||
|
# previous test
|
||
|
ok $db->db_get(1, $value) == 0 ;
|
||
|
ok $value eq fillout("some value", $rec_len, $pad) ;
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
{
|
||
|
# Check simple env works with a array.
|
||
|
# and pad defaults to space
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
|
||
|
my $home = "./fred" ;
|
||
|
my $rec_len = 11 ;
|
||
|
ok my $lexD = new LexDir($home);
|
||
|
|
||
|
ok my $env = new BerkeleyDB::Env -Flags => DB_CREATE|DB_INIT_MPOOL,@StdErrFile,
|
||
|
-Home => $home ;
|
||
|
ok my $db = new BerkeleyDB::Queue -Filename => $Dfile,
|
||
|
-Env => $env,
|
||
|
-Flags => DB_CREATE,
|
||
|
-Len => $rec_len;
|
||
|
|
||
|
isa_ok $db->Env, 'BerkeleyDB::Env';
|
||
|
|
||
|
# Add a k/v pair
|
||
|
my $value ;
|
||
|
ok $db->db_put(1, "some value") == 0 ;
|
||
|
ok $db->db_get(1, $value) == 0 ;
|
||
|
ok $value eq fillout("some value", $rec_len) ;
|
||
|
undef $db ;
|
||
|
undef $env ;
|
||
|
}
|
||
|
|
||
|
|
||
|
{
|
||
|
# cursors
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my ($k, $v) ;
|
||
|
my $rec_len = 5 ;
|
||
|
ok my $db = new BerkeleyDB::Queue -Filename => $Dfile,
|
||
|
-ArrayBase => 0,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Len => $rec_len;
|
||
|
|
||
|
# create some data
|
||
|
my @data = (
|
||
|
"red" ,
|
||
|
"green" ,
|
||
|
"blue" ,
|
||
|
) ;
|
||
|
|
||
|
my $i ;
|
||
|
my %data ;
|
||
|
my $ret = 0 ;
|
||
|
for ($i = 0 ; $i < @data ; ++$i) {
|
||
|
$ret += $db->db_put($i, $data[$i]) ;
|
||
|
$data{$i} = $data[$i] ;
|
||
|
}
|
||
|
ok $ret == 0 ;
|
||
|
|
||
|
# create the cursor
|
||
|
ok my $cursor = $db->db_cursor() ;
|
||
|
|
||
|
$k = 0 ; $v = "" ;
|
||
|
my %copy = %data;
|
||
|
my $extras = 0 ;
|
||
|
# sequence forwards
|
||
|
while ($cursor->c_get($k, $v, DB_NEXT) == 0)
|
||
|
{
|
||
|
if ( fillout($copy{$k}, $rec_len) eq $v )
|
||
|
{ delete $copy{$k} }
|
||
|
else
|
||
|
{ ++ $extras }
|
||
|
}
|
||
|
|
||
|
ok $cursor->status() == DB_NOTFOUND ;
|
||
|
ok $cursor->status() =~ $DB_errors{'DB_NOTFOUND'} ;
|
||
|
ok keys %copy == 0 ;
|
||
|
ok $extras == 0 ;
|
||
|
|
||
|
# sequence backwards
|
||
|
%copy = %data ;
|
||
|
$extras = 0 ;
|
||
|
my $status ;
|
||
|
for ( $status = $cursor->c_get($k, $v, DB_LAST) ;
|
||
|
$status == 0 ;
|
||
|
$status = $cursor->c_get($k, $v, DB_PREV)) {
|
||
|
if ( fillout($copy{$k}, $rec_len) eq $v )
|
||
|
{ delete $copy{$k} }
|
||
|
else
|
||
|
{ ++ $extras }
|
||
|
}
|
||
|
ok $status == DB_NOTFOUND ;
|
||
|
ok $status =~ $DB_errors{'DB_NOTFOUND'} ;
|
||
|
ok $cursor->status() == $status ;
|
||
|
ok $cursor->status() eq $status ;
|
||
|
ok keys %copy == 0 ;
|
||
|
ok $extras == 0 ;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# Tied Array interface
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my $db ;
|
||
|
my $rec_len = 10 ;
|
||
|
ok $db = tie @array, 'BerkeleyDB::Queue', -Filename => $Dfile,
|
||
|
-ArrayBase => 0,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Len => $rec_len;
|
||
|
|
||
|
ok my $cursor = (tied @array)->db_cursor() ;
|
||
|
# check the database is empty
|
||
|
my $count = 0 ;
|
||
|
my ($k, $v) = (0,"") ;
|
||
|
while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $cursor->status() == DB_NOTFOUND ;
|
||
|
ok $count == 0 ;
|
||
|
|
||
|
ok @array == 0 ;
|
||
|
|
||
|
# Add a k/v pair
|
||
|
my $value ;
|
||
|
$array[1] = "some value";
|
||
|
ok ((tied @array)->status() == 0) ;
|
||
|
ok $array[1] eq fillout("some value", $rec_len);
|
||
|
ok defined $array[1];
|
||
|
ok ((tied @array)->status() == 0) ;
|
||
|
ok !defined $array[3];
|
||
|
ok ((tied @array)->status() == DB_NOTFOUND) ;
|
||
|
|
||
|
$array[1] = 2 ;
|
||
|
$array[10] = 20 ;
|
||
|
$array[100] = 200 ;
|
||
|
|
||
|
my ($keys, $values) = (0,0);
|
||
|
$count = 0 ;
|
||
|
for ( my $status = $cursor->c_get($k, $v, DB_FIRST) ;
|
||
|
$status == 0 ;
|
||
|
$status = $cursor->c_get($k, $v, DB_NEXT)) {
|
||
|
$keys += $k ;
|
||
|
$values += $v ;
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $count == 3 ;
|
||
|
ok $keys == 111 ;
|
||
|
ok $values == 222 ;
|
||
|
|
||
|
# unshift isn't allowed
|
||
|
# eval {
|
||
|
# $FA ? unshift @array, "red", "green", "blue"
|
||
|
# : $db->unshift("red", "green", "blue" ) ;
|
||
|
# } ;
|
||
|
# ok $@ =~ /^unshift is unsupported with Queue databases/ ;
|
||
|
$array[0] = "red" ;
|
||
|
$array[1] = "green" ;
|
||
|
$array[2] = "blue" ;
|
||
|
$array[4] = 2 ;
|
||
|
ok $array[0] eq fillout("red", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_FIRST) == 0 ;
|
||
|
ok $k == 0 ;
|
||
|
ok $v eq fillout("red", $rec_len) ;
|
||
|
ok $array[1] eq fillout("green", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 1 ;
|
||
|
ok $v eq fillout("green", $rec_len) ;
|
||
|
ok $array[2] eq fillout("blue", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 2 ;
|
||
|
ok $v eq fillout("blue", $rec_len) ;
|
||
|
ok $array[4] == 2 ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 4 ;
|
||
|
ok $v == 2 ;
|
||
|
|
||
|
# shift
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("red", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("green", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("blue", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) == 2) ;
|
||
|
|
||
|
# push
|
||
|
$FA ? push @array, "the", "end"
|
||
|
: $db->push("the", "end") ;
|
||
|
ok $cursor->c_get($k, $v, DB_LAST) == 0 ;
|
||
|
ok $k == 102 ;
|
||
|
ok $v eq fillout("end", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_PREV) == 0 ;
|
||
|
ok $k == 101 ;
|
||
|
ok $v eq fillout("the", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_PREV) == 0 ;
|
||
|
ok $k == 100 ;
|
||
|
ok $v == 200 ;
|
||
|
|
||
|
# pop
|
||
|
ok (( $FA ? pop @array : $db->pop ) eq fillout("end", $rec_len)) ;
|
||
|
ok (( $FA ? pop @array : $db->pop ) eq fillout("the", $rec_len)) ;
|
||
|
ok (( $FA ? pop @array : $db->pop ) == 200) ;
|
||
|
|
||
|
undef $cursor;
|
||
|
|
||
|
# now clear the array
|
||
|
$FA ? @array = ()
|
||
|
: $db->clear() ;
|
||
|
ok $cursor = (tied @array)->db_cursor() ;
|
||
|
ok $cursor->c_get($k, $v, DB_FIRST) == DB_NOTFOUND ;
|
||
|
|
||
|
undef $cursor ;
|
||
|
undef $db ;
|
||
|
untie @array ;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# in-memory file
|
||
|
|
||
|
my @array ;
|
||
|
my $fd ;
|
||
|
my $value ;
|
||
|
my $rec_len = 15 ;
|
||
|
ok my $db = tie @array, 'BerkeleyDB::Queue',
|
||
|
-Len => $rec_len;
|
||
|
|
||
|
ok $db->db_put(1, "some value") == 0 ;
|
||
|
ok $db->db_get(1, $value) == 0 ;
|
||
|
ok $value eq fillout("some value", $rec_len) ;
|
||
|
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# partial
|
||
|
# check works via API
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my $value ;
|
||
|
my $rec_len = 8 ;
|
||
|
ok my $db = new BerkeleyDB::Queue -Filename => $Dfile,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " " ;
|
||
|
|
||
|
# create some data
|
||
|
my @data = (
|
||
|
"",
|
||
|
"boat",
|
||
|
"house",
|
||
|
"sea",
|
||
|
) ;
|
||
|
|
||
|
my $ret = 0 ;
|
||
|
my $i ;
|
||
|
for ($i = 0 ; $i < @data ; ++$i) {
|
||
|
my $r = $db->db_put($i, $data[$i]) ;
|
||
|
$ret += $r ;
|
||
|
}
|
||
|
ok $ret == 0 ;
|
||
|
|
||
|
# do a partial get
|
||
|
my ($pon, $off, $len) = $db->partial_set(0,2) ;
|
||
|
ok ! $pon && $off == 0 && $len == 0 ;
|
||
|
ok $db->db_get(1, $value) == 0 && $value eq "bo" ;
|
||
|
ok $db->db_get(2, $value) == 0 && $value eq "ho" ;
|
||
|
ok $db->db_get(3, $value) == 0 && $value eq "se" ;
|
||
|
|
||
|
# do a partial get, off end of data
|
||
|
($pon, $off, $len) = $db->partial_set(3,2) ;
|
||
|
ok $pon ;
|
||
|
ok $off == 0 ;
|
||
|
ok $len == 2 ;
|
||
|
ok $db->db_get(1, $value) == 0 && $value eq fillout("t", 2) ;
|
||
|
ok $db->db_get(2, $value) == 0 && $value eq "se" ;
|
||
|
ok $db->db_get(3, $value) == 0 && $value eq " " ;
|
||
|
|
||
|
# switch of partial mode
|
||
|
($pon, $off, $len) = $db->partial_clear() ;
|
||
|
ok $pon ;
|
||
|
ok $off == 3 ;
|
||
|
ok $len == 2 ;
|
||
|
ok $db->db_get(1, $value) == 0 && $value eq fillout("boat", $rec_len) ;
|
||
|
ok $db->db_get(2, $value) == 0 && $value eq fillout("house", $rec_len) ;
|
||
|
ok $db->db_get(3, $value) == 0 && $value eq fillout("sea", $rec_len) ;
|
||
|
|
||
|
# now partial put
|
||
|
$db->partial_set(0,2) ;
|
||
|
ok $db->db_put(1, "") != 0 ;
|
||
|
ok $db->db_put(2, "AB") == 0 ;
|
||
|
ok $db->db_put(3, "XY") == 0 ;
|
||
|
ok $db->db_put(4, "KLM") != 0 ;
|
||
|
ok $db->db_put(4, "KL") == 0 ;
|
||
|
|
||
|
($pon, $off, $len) = $db->partial_clear() ;
|
||
|
ok $pon ;
|
||
|
ok $off == 0 ;
|
||
|
ok $len == 2 ;
|
||
|
ok $db->db_get(1, $value) == 0 && $value eq fillout("boat", $rec_len) ;
|
||
|
ok $db->db_get(2, $value) == 0 && $value eq fillout("ABuse", $rec_len) ;
|
||
|
ok $db->db_get(3, $value) == 0 && $value eq fillout("XYa", $rec_len) ;
|
||
|
ok $db->db_get(4, $value) == 0 && $value eq fillout("KL", $rec_len) ;
|
||
|
|
||
|
# now partial put
|
||
|
($pon, $off, $len) = $db->partial_set(3,2) ;
|
||
|
ok ! $pon ;
|
||
|
ok $off == 0 ;
|
||
|
ok $len == 0 ;
|
||
|
ok $db->db_put(1, "PP") == 0 ;
|
||
|
ok $db->db_put(2, "Q") != 0 ;
|
||
|
ok $db->db_put(3, "XY") == 0 ;
|
||
|
ok $db->db_put(4, "TU") == 0 ;
|
||
|
|
||
|
$db->partial_clear() ;
|
||
|
ok $db->db_get(1, $value) == 0 && $value eq fillout("boaPP", $rec_len) ;
|
||
|
ok $db->db_get(2, $value) == 0 && $value eq fillout("ABuse",$rec_len) ;
|
||
|
ok $db->db_get(3, $value) == 0 && $value eq fillout("XYaXY", $rec_len) ;
|
||
|
ok $db->db_get(4, $value) == 0 && $value eq fillout("KL TU", $rec_len) ;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# partial
|
||
|
# check works via tied array
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my $value ;
|
||
|
my $rec_len = 8 ;
|
||
|
ok my $db = tie @array, 'BerkeleyDB::Queue', -Filename => $Dfile,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " " ;
|
||
|
|
||
|
# create some data
|
||
|
my @data = (
|
||
|
"",
|
||
|
"boat",
|
||
|
"house",
|
||
|
"sea",
|
||
|
) ;
|
||
|
|
||
|
my $i ;
|
||
|
my $status = 0 ;
|
||
|
for ($i = 1 ; $i < @data ; ++$i) {
|
||
|
$array[$i] = $data[$i] ;
|
||
|
$status += $db->status() ;
|
||
|
}
|
||
|
|
||
|
ok $status == 0 ;
|
||
|
|
||
|
# do a partial get
|
||
|
$db->partial_set(0,2) ;
|
||
|
ok $array[1] eq fillout("bo", 2) ;
|
||
|
ok $array[2] eq fillout("ho", 2) ;
|
||
|
ok $array[3] eq fillout("se", 2) ;
|
||
|
|
||
|
# do a partial get, off end of data
|
||
|
$db->partial_set(3,2) ;
|
||
|
ok $array[1] eq fillout("t", 2) ;
|
||
|
ok $array[2] eq fillout("se", 2) ;
|
||
|
ok $array[3] eq fillout("", 2) ;
|
||
|
|
||
|
# switch of partial mode
|
||
|
$db->partial_clear() ;
|
||
|
ok $array[1] eq fillout("boat", $rec_len) ;
|
||
|
ok $array[2] eq fillout("house", $rec_len) ;
|
||
|
ok $array[3] eq fillout("sea", $rec_len) ;
|
||
|
|
||
|
# now partial put
|
||
|
$db->partial_set(0,2) ;
|
||
|
$array[1] = "" ;
|
||
|
ok $db->status() != 0 ;
|
||
|
$array[2] = "AB" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
$array[3] = "XY" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
$array[4] = "KL" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
|
||
|
$db->partial_clear() ;
|
||
|
ok $array[1] eq fillout("boat", $rec_len) ;
|
||
|
ok $array[2] eq fillout("ABuse", $rec_len) ;
|
||
|
ok $array[3] eq fillout("XYa", $rec_len) ;
|
||
|
ok $array[4] eq fillout("KL", $rec_len) ;
|
||
|
|
||
|
# now partial put
|
||
|
$db->partial_set(3,2) ;
|
||
|
$array[1] = "PP" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
$array[2] = "Q" ;
|
||
|
ok $db->status() != 0 ;
|
||
|
$array[3] = "XY" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
$array[4] = "TU" ;
|
||
|
ok $db->status() == 0 ;
|
||
|
|
||
|
$db->partial_clear() ;
|
||
|
ok $array[1] eq fillout("boaPP", $rec_len) ;
|
||
|
ok $array[2] eq fillout("ABuse", $rec_len) ;
|
||
|
ok $array[3] eq fillout("XYaXY", $rec_len) ;
|
||
|
ok $array[4] eq fillout("KL TU", $rec_len) ;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# transaction
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my $value ;
|
||
|
|
||
|
my $home = "./fred" ;
|
||
|
ok my $lexD = new LexDir($home);
|
||
|
my $rec_len = 9 ;
|
||
|
ok my $env = new BerkeleyDB::Env -Home => $home,@StdErrFile,
|
||
|
-Flags => DB_CREATE|DB_INIT_TXN|
|
||
|
DB_INIT_MPOOL|DB_INIT_LOCK ;
|
||
|
ok my $txn = $env->txn_begin() ;
|
||
|
ok my $db1 = tie @array, 'BerkeleyDB::Queue',
|
||
|
-Filename => $Dfile,
|
||
|
-ArrayBase => 0,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Env => $env,
|
||
|
-Txn => $txn ,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " " ;
|
||
|
|
||
|
|
||
|
ok $txn->txn_commit() == 0 ;
|
||
|
ok $txn = $env->txn_begin() ;
|
||
|
$db1->Txn($txn);
|
||
|
|
||
|
# create some data
|
||
|
my @data = (
|
||
|
"boat",
|
||
|
"house",
|
||
|
"sea",
|
||
|
) ;
|
||
|
|
||
|
my $ret = 0 ;
|
||
|
my $i ;
|
||
|
for ($i = 0 ; $i < @data ; ++$i) {
|
||
|
$ret += $db1->db_put($i, $data[$i]) ;
|
||
|
}
|
||
|
ok $ret == 0 ;
|
||
|
|
||
|
# should be able to see all the records
|
||
|
|
||
|
ok my $cursor = $db1->db_cursor() ;
|
||
|
my ($k, $v) = (0, "") ;
|
||
|
my $count = 0 ;
|
||
|
# sequence forwards
|
||
|
while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $count == 3 ;
|
||
|
undef $cursor ;
|
||
|
|
||
|
# now abort the transaction
|
||
|
ok $txn->txn_abort() == 0 ;
|
||
|
|
||
|
# there shouldn't be any records in the database
|
||
|
$count = 0 ;
|
||
|
# sequence forwards
|
||
|
ok $cursor = $db1->db_cursor() ;
|
||
|
while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $count == 0 ;
|
||
|
|
||
|
undef $txn ;
|
||
|
undef $cursor ;
|
||
|
undef $db1 ;
|
||
|
undef $env ;
|
||
|
untie @array ;
|
||
|
}
|
||
|
|
||
|
|
||
|
{
|
||
|
# db_stat
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my $recs = ($BerkeleyDB::db_version >= 3.1 ? "qs_ndata" : "qs_nrecs") ;
|
||
|
my @array ;
|
||
|
my ($k, $v) ;
|
||
|
my $rec_len = 7 ;
|
||
|
ok my $db = new BerkeleyDB::Queue -Filename => $Dfile,
|
||
|
-Flags => DB_CREATE,
|
||
|
-Pagesize => 4 * 1024,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " "
|
||
|
;
|
||
|
|
||
|
my $ref = $db->db_stat() ;
|
||
|
ok $ref->{$recs} == 0;
|
||
|
ok $ref->{'qs_pagesize'} == 4 * 1024;
|
||
|
|
||
|
# create some data
|
||
|
my @data = (
|
||
|
2,
|
||
|
"house",
|
||
|
"sea",
|
||
|
) ;
|
||
|
|
||
|
my $ret = 0 ;
|
||
|
my $i ;
|
||
|
for ($i = $db->ArrayOffset ; @data ; ++$i) {
|
||
|
$ret += $db->db_put($i, shift @data) ;
|
||
|
}
|
||
|
ok $ret == 0 ;
|
||
|
|
||
|
$ref = $db->db_stat() ;
|
||
|
ok $ref->{$recs} == 3;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# sub-class test
|
||
|
|
||
|
package Another ;
|
||
|
|
||
|
use strict ;
|
||
|
|
||
|
open(FILE, ">SubDB.pm") or die "Cannot open SubDB.pm: $!\n" ;
|
||
|
print FILE <<'EOM' ;
|
||
|
|
||
|
package SubDB ;
|
||
|
|
||
|
use strict ;
|
||
|
use vars qw( @ISA @EXPORT) ;
|
||
|
|
||
|
require Exporter ;
|
||
|
use BerkeleyDB;
|
||
|
@ISA=qw(BerkeleyDB BerkeleyDB::Queue);
|
||
|
@EXPORT = @BerkeleyDB::EXPORT ;
|
||
|
|
||
|
sub db_put {
|
||
|
my $self = shift ;
|
||
|
my $key = shift ;
|
||
|
my $value = shift ;
|
||
|
$self->SUPER::db_put($key, $value * 3) ;
|
||
|
}
|
||
|
|
||
|
sub db_get {
|
||
|
my $self = shift ;
|
||
|
$self->SUPER::db_get($_[0], $_[1]) ;
|
||
|
$_[1] -= 2 ;
|
||
|
}
|
||
|
|
||
|
sub A_new_method
|
||
|
{
|
||
|
my $self = shift ;
|
||
|
my $key = shift ;
|
||
|
my $value = $self->FETCH($key) ;
|
||
|
return "[[$value]]" ;
|
||
|
}
|
||
|
|
||
|
1 ;
|
||
|
EOM
|
||
|
|
||
|
close FILE ;
|
||
|
|
||
|
use Test::More;
|
||
|
BEGIN { push @INC, '.'; }
|
||
|
eval 'use SubDB ; ';
|
||
|
ok $@ eq "" ;
|
||
|
my @h ;
|
||
|
my $X ;
|
||
|
my $rec_len = 34 ;
|
||
|
eval '
|
||
|
$X = tie(@h, "SubDB", -Filename => "dbqueue.tmp",
|
||
|
-Flags => DB_CREATE,
|
||
|
-Mode => 0640 ,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " "
|
||
|
);
|
||
|
' ;
|
||
|
|
||
|
ok $@ eq "" ;
|
||
|
|
||
|
my $ret = eval '$h[1] = 3 ; return $h[1] ' ;
|
||
|
ok $@ eq "" ;
|
||
|
ok $ret == 7 ;
|
||
|
|
||
|
my $value = 0;
|
||
|
$ret = eval '$X->db_put(1, 4) ; $X->db_get(1, $value) ; return $value' ;
|
||
|
ok $@ eq "" ;
|
||
|
ok $ret == 10 ;
|
||
|
|
||
|
$ret = eval ' DB_NEXT eq main::DB_NEXT ' ;
|
||
|
ok $@ eq "" ;
|
||
|
ok $ret == 1 ;
|
||
|
|
||
|
$ret = eval '$X->A_new_method(1) ' ;
|
||
|
ok $@ eq "" ;
|
||
|
ok $ret eq "[[10]]" ;
|
||
|
|
||
|
undef $X ;
|
||
|
untie @h ;
|
||
|
unlink "SubDB.pm", "dbqueue.tmp" ;
|
||
|
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# DB_APPEND
|
||
|
|
||
|
my $lex = new LexFile $Dfile;
|
||
|
my @array ;
|
||
|
my $value ;
|
||
|
my $rec_len = 21 ;
|
||
|
ok my $db = tie @array, 'BerkeleyDB::Queue',
|
||
|
-Filename => $Dfile,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " " ;
|
||
|
|
||
|
# create a few records
|
||
|
$array[1] = "def" ;
|
||
|
$array[3] = "ghi" ;
|
||
|
|
||
|
my $k = 0 ;
|
||
|
ok $db->db_put($k, "fred", DB_APPEND) == 0 ;
|
||
|
ok $k == 4 ;
|
||
|
ok $array[4] eq fillout("fred", $rec_len) ;
|
||
|
|
||
|
undef $db ;
|
||
|
untie @array ;
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# 23 Sept 2001 -- push into an empty array
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my $db ;
|
||
|
my $rec_len = 21 ;
|
||
|
ok $db = tie @array, 'BerkeleyDB::Queue',
|
||
|
-Flags => DB_CREATE ,
|
||
|
-ArrayBase => 0,
|
||
|
-Len => $rec_len,
|
||
|
-Pad => " " ,
|
||
|
-Filename => $Dfile ;
|
||
|
$FA ? push @array, "first"
|
||
|
: $db->push("first") ;
|
||
|
|
||
|
ok (($FA ? pop @array : $db->pop()) eq fillout("first", $rec_len)) ;
|
||
|
|
||
|
undef $db;
|
||
|
untie @array ;
|
||
|
|
||
|
}
|
||
|
|
||
|
{
|
||
|
# Tied Array interface with transactions
|
||
|
|
||
|
my $lex = new LexFile $Dfile ;
|
||
|
my @array ;
|
||
|
my $db ;
|
||
|
my $rec_len = 10 ;
|
||
|
my $home = "./fred" ;
|
||
|
ok my $lexD = new LexDir($home);
|
||
|
ok my $env = new BerkeleyDB::Env -Home => $home,@StdErrFile,
|
||
|
-Flags => DB_CREATE|DB_INIT_TXN|
|
||
|
DB_INIT_MPOOL|DB_INIT_LOCK ;
|
||
|
ok my $txn = $env->txn_begin() ;
|
||
|
ok $db = tie @array, 'BerkeleyDB::Queue', -Filename => $Dfile,
|
||
|
-ArrayBase => 0,
|
||
|
-Flags => DB_CREATE ,
|
||
|
-Env => $env ,
|
||
|
-Txn => $txn ,
|
||
|
-Len => $rec_len;
|
||
|
|
||
|
ok $txn->txn_commit() == 0 ;
|
||
|
ok $txn = $env->txn_begin() ;
|
||
|
$db->Txn($txn);
|
||
|
|
||
|
ok my $cursor = (tied @array)->db_cursor() ;
|
||
|
# check the database is empty
|
||
|
my $count = 0 ;
|
||
|
my ($k, $v) = (0,"") ;
|
||
|
while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $cursor->status() == DB_NOTFOUND ;
|
||
|
ok $count == 0 ;
|
||
|
|
||
|
ok @array == 0 ;
|
||
|
|
||
|
# Add a k/v pair
|
||
|
my $value ;
|
||
|
$array[1] = "some value";
|
||
|
ok ((tied @array)->status() == 0) ;
|
||
|
ok $array[1] eq fillout("some value", $rec_len);
|
||
|
ok defined $array[1];
|
||
|
ok ((tied @array)->status() == 0) ;
|
||
|
ok !defined $array[3];
|
||
|
ok ((tied @array)->status() == DB_NOTFOUND) ;
|
||
|
|
||
|
$array[1] = 2 ;
|
||
|
$array[10] = 20 ;
|
||
|
$array[100] = 200 ;
|
||
|
|
||
|
my ($keys, $values) = (0,0);
|
||
|
$count = 0 ;
|
||
|
for ( my $status = $cursor->c_get($k, $v, DB_FIRST) ;
|
||
|
$status == 0 ;
|
||
|
$status = $cursor->c_get($k, $v, DB_NEXT)) {
|
||
|
$keys += $k ;
|
||
|
$values += $v ;
|
||
|
++ $count ;
|
||
|
}
|
||
|
ok $count == 3 ;
|
||
|
ok $keys == 111 ;
|
||
|
ok $values == 222 ;
|
||
|
|
||
|
# unshift isn't allowed
|
||
|
# eval {
|
||
|
# $FA ? unshift @array, "red", "green", "blue"
|
||
|
# : $db->unshift("red", "green", "blue" ) ;
|
||
|
# } ;
|
||
|
# ok $@ =~ /^unshift is unsupported with Queue databases/ ;
|
||
|
$array[0] = "red" ;
|
||
|
$array[1] = "green" ;
|
||
|
$array[2] = "blue" ;
|
||
|
$array[4] = 2 ;
|
||
|
ok $array[0] eq fillout("red", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_FIRST) == 0 ;
|
||
|
ok $k == 0 ;
|
||
|
ok $v eq fillout("red", $rec_len) ;
|
||
|
ok $array[1] eq fillout("green", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 1 ;
|
||
|
ok $v eq fillout("green", $rec_len) ;
|
||
|
ok $array[2] eq fillout("blue", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 2 ;
|
||
|
ok $v eq fillout("blue", $rec_len) ;
|
||
|
ok $array[4] == 2 ;
|
||
|
ok $cursor->c_get($k, $v, DB_NEXT) == 0 ;
|
||
|
ok $k == 4 ;
|
||
|
ok $v == 2 ;
|
||
|
|
||
|
# shift
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("red", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("green", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) eq fillout("blue", $rec_len)) ;
|
||
|
ok (($FA ? shift @array : $db->shift()) == 2) ;
|
||
|
|
||
|
# push
|
||
|
$FA ? push @array, "the", "end"
|
||
|
: $db->push("the", "end") ;
|
||
|
ok $cursor->c_get($k, $v, DB_LAST) == 0 ;
|
||
|
ok $k == 102 ;
|
||
|
ok $v eq fillout("end", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_PREV) == 0 ;
|
||
|
ok $k == 101 ;
|
||
|
ok $v eq fillout("the", $rec_len) ;
|
||
|
ok $cursor->c_get($k, $v, DB_PREV) == 0 ;
|
||
|
ok $k == 100 ;
|
||
|
ok $v == 200 ;
|
||
|
|
||
|
# pop
|
||
|
ok (( $FA ? pop @array : $db->pop ) eq fillout("end", $rec_len)) ;
|
||
|
ok (( $FA ? pop @array : $db->pop ) eq fillout("the", $rec_len)) ;
|
||
|
ok (( $FA ? pop @array : $db->pop ) == 200 ) ;
|
||
|
|
||
|
undef $cursor ;
|
||
|
# now clear the array
|
||
|
$FA ? @array = ()
|
||
|
: $db->clear() ;
|
||
|
ok $cursor = (tied @array)->db_cursor() ;
|
||
|
ok $cursor->c_get($k, $v, DB_FIRST) == DB_NOTFOUND ;
|
||
|
undef $cursor ;
|
||
|
ok $txn->txn_commit() == 0 ;
|
||
|
|
||
|
undef $db ;
|
||
|
untie @array ;
|
||
|
}
|
||
|
__END__
|
||
|
|
||
|
|
||
|
# TODO
|
||
|
#
|
||
|
# DB_DELIMETER DB_FIXEDLEN DB_PAD DB_SNAPSHOT with partial records
|