btree.t

关于Berkelay数据库的共享源码

    ok 108, $db->db_put("some key", "some value") == 0  ;
    ok 109, $db->db_get("some key", $value) == 0 ;
    ok 110, $value eq "some value" ;

}
 
{
    # partial
    # check works via API

    my $lex = new LexFile $Dfile ;
    my $value ;
    ok 111, my $db = new BerkeleyDB::Btree, -Filename => $Dfile,
                                      	       -Flags    => DB_CREATE ;

    # create some data
    my %data =  (
		"red"	=> "boat",
		"green"	=> "house",
		"blue"	=> "sea",
		) ;

    my $ret = 0 ;
    while (my ($k, $v) = each %data) {
        $ret += $db->db_put($k, $v) ;
    }
    ok 112, $ret == 0 ;


    # do a partial get
    my ($pon, $off, $len) = $db->partial_set(0,2) ;
    ok 113, ! $pon && $off == 0 && $len == 0 ;
    ok 114, $db->db_get("red", $value) == 0 && $value eq "bo" ;
    ok 115, $db->db_get("green", $value) == 0 && $value eq "ho" ;
    ok 116, $db->db_get("blue", $value) == 0 && $value eq "se" ;

    # do a partial get, off end of data
    ($pon, $off, $len) = $db->partial_set(3,2) ;
    ok 117, $pon ;
    ok 118, $off == 0 ;
    ok 119, $len == 2 ;
    ok 120, $db->db_get("red", $value) == 0 && $value eq "t" ;
    ok 121, $db->db_get("green", $value) == 0 && $value eq "se" ;
    ok 122, $db->db_get("blue", $value) == 0 && $value eq "" ;

    # switch of partial mode
    ($pon, $off, $len) = $db->partial_clear() ;
    ok 123, $pon ;
    ok 124, $off == 3 ;
    ok 125, $len == 2 ;
    ok 126, $db->db_get("red", $value) == 0 && $value eq "boat" ;
    ok 127, $db->db_get("green", $value) == 0 && $value eq "house" ;
    ok 128, $db->db_get("blue", $value) == 0 && $value eq "sea" ;

    # now partial put
    $db->partial_set(0,2) ;
    ok 129, $db->db_put("red", "") == 0 ;
    ok 130, $db->db_put("green", "AB") == 0 ;
    ok 131, $db->db_put("blue", "XYZ") == 0 ;
    ok 132, $db->db_put("new", "KLM") == 0 ;

    ($pon, $off, $len) = $db->partial_clear() ;
    ok 133, $pon ;
    ok 134, $off == 0 ;
    ok 135, $len == 2 ;
    ok 136, $db->db_get("red", $value) == 0 && $value eq "at" ;
    ok 137, $db->db_get("green", $value) == 0 && $value eq "ABuse" ;
    ok 138, $db->db_get("blue", $value) == 0 && $value eq "XYZa" ;
    ok 139, $db->db_get("new", $value) == 0 && $value eq "KLM" ;

    # now partial put
    ($pon, $off, $len) = $db->partial_set(3,2) ;
    ok 140, ! $pon ;
    ok 141, $off == 0 ;
    ok 142, $len == 0 ;
    ok 143, $db->db_put("red", "PPP") == 0 ;
    ok 144, $db->db_put("green", "Q") == 0 ;
    ok 145, $db->db_put("blue", "XYZ") == 0 ;
    ok 146, $db->db_put("new", "TU") == 0 ;

    $db->partial_clear() ;
    ok 147, $db->db_get("red", $value) == 0 && $value eq "at\0PPP" ;
    ok 148, $db->db_get("green", $value) == 0 && $value eq "ABuQ" ;
    ok 149, $db->db_get("blue", $value) == 0 && $value eq "XYZXYZ" ;
    ok 150, $db->db_get("new", $value) == 0 && $value eq "KLMTU" ;
}

{
    # partial
    # check works via tied hash 

    my $lex = new LexFile $Dfile ;
    my %hash ;
    my $value ;
    ok 151, my $db = tie %hash, 'BerkeleyDB::Btree', -Filename => $Dfile,
                                      	       -Flags    => DB_CREATE ;

    # create some data
    my %data =  (
		"red"	=> "boat",
		"green"	=> "house",
		"blue"	=> "sea",
		) ;

    while (my ($k, $v) = each %data) {
	$hash{$k} = $v ;
    }


    # do a partial get
    $db->partial_set(0,2) ;
    ok 152, $hash{"red"} eq "bo" ;
    ok 153, $hash{"green"} eq "ho" ;
    ok 154, $hash{"blue"}  eq "se" ;

    # do a partial get, off end of data
    $db->partial_set(3,2) ;
    ok 155, $hash{"red"} eq "t" ;
    ok 156, $hash{"green"} eq "se" ;
    ok 157, $hash{"blue"} eq "" ;

    # switch of partial mode
    $db->partial_clear() ;
    ok 158, $hash{"red"} eq "boat" ;
    ok 159, $hash{"green"} eq "house" ;
    ok 160, $hash{"blue"} eq "sea" ;

    # now partial put
    $db->partial_set(0,2) ;
    ok 161, $hash{"red"} = "" ;
    ok 162, $hash{"green"} = "AB" ;
    ok 163, $hash{"blue"} = "XYZ" ;
    ok 164, $hash{"new"} = "KLM" ;

    $db->partial_clear() ;
    ok 165, $hash{"red"} eq "at" ;
    ok 166, $hash{"green"} eq "ABuse" ;
    ok 167, $hash{"blue"} eq "XYZa" ;
    ok 168, $hash{"new"} eq "KLM" ;

    # now partial put
    $db->partial_set(3,2) ;
    ok 169, $hash{"red"} = "PPP" ;
    ok 170, $hash{"green"} = "Q" ;
    ok 171, $hash{"blue"} = "XYZ" ;
    ok 172, $hash{"new"} = "TU" ;

    $db->partial_clear() ;
    ok 173, $hash{"red"} eq "at\0PPP" ;
    ok 174, $hash{"green"} eq "ABuQ" ;
    ok 175, $hash{"blue"} eq "XYZXYZ" ;
    ok 176, $hash{"new"} eq "KLMTU" ;
}

{
    # transaction

    my $lex = new LexFile $Dfile ;
    my %hash ;
    my $value ;

    my $home = "./fred" ;
    ok 177, my $lexD = new LexDir($home) ;
    ok 178, my $env = new BerkeleyDB::Env -Home => $home, @StdErrFile,
				     -Flags => DB_CREATE|DB_INIT_TXN|
					  	DB_INIT_MPOOL|DB_INIT_LOCK ;
    ok 179, my $txn = $env->txn_begin() ;
    ok 180, my $db1 = tie %hash, 'BerkeleyDB::Btree', -Filename => $Dfile,
                                      	       -Flags    =>  DB_CREATE ,
					       -Env 	 => $env,
					       -Txn	 => $txn ;

    ok 181, (my $Z = $txn->txn_commit()) == 0 ;
    ok 182, $txn = $env->txn_begin() ;
    $db1->Txn($txn);
    
    # create some data
    my %data =  (
		"red"	=> "boat",
		"green"	=> "house",
		"blue"	=> "sea",
		) ;

    my $ret = 0 ;
    while (my ($k, $v) = each %data) {
        $ret += $db1->db_put($k, $v) ;
    }
    ok 183, $ret == 0 ;

    # should be able to see all the records

    ok 184, my $cursor = $db1->db_cursor() ;
    my ($k, $v) = ("", "") ;
    my $count = 0 ;
    # sequence forwards
    while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
        ++ $count ;
    }
    ok 185, $count == 3 ;
    undef $cursor ;

    # now abort the transaction
    #ok 151, $txn->txn_abort() == 0 ;
    ok 186, ($Z = $txn->txn_abort()) == 0 ;

    # there shouldn't be any records in the database
    $count = 0 ;
    # sequence forwards
    ok 187, $cursor = $db1->db_cursor() ;
    while ($cursor->c_get($k, $v, DB_NEXT) == 0) {
        ++ $count ;
    }
    ok 188, $count == 0 ;

    undef $txn ;
    undef $cursor ;
    undef $db1 ;
    undef $env ;
    untie %hash ;
}

{
    # DB_DUP

    my $lex = new LexFile $Dfile ;
    my %hash ;
    ok 189, my $db = tie %hash, 'BerkeleyDB::Btree', -Filename => $Dfile,
				      -Property  => DB_DUP,
                                      -Flags    => DB_CREATE ;

    $hash{'Wall'} = 'Larry' ;
    $hash{'Wall'} = 'Stone' ;
    $hash{'Smith'} = 'John' ;
    $hash{'Wall'} = 'Brick' ;
    $hash{'Wall'} = 'Brick' ;
    $hash{'mouse'} = 'mickey' ;

    ok 190, keys %hash == 6 ;

    # create a cursor
    ok 191, my $cursor = $db->db_cursor() ;

    my $key = "Wall" ;
    my $value ;
    ok 192, $cursor->c_get($key, $value, DB_SET) == 0 ;
    ok 193, $key eq "Wall" && $value eq "Larry" ;
    ok 194, $cursor->c_get($key, $value, DB_NEXT) == 0 ;
    ok 195, $key eq "Wall" && $value eq "Stone" ;
    ok 196, $cursor->c_get($key, $value, DB_NEXT) == 0 ;
    ok 197, $key eq "Wall" && $value eq "Brick" ;
    ok 198, $cursor->c_get($key, $value, DB_NEXT) == 0 ;
    ok 199, $key eq "Wall" && $value eq "Brick" ;

    #my $ref = $db->db_stat() ; 
    #ok 200, ($ref->{bt_flags} | DB_DUP) == DB_DUP ;
#print "bt_flags " . $ref->{bt_flags} . " DB_DUP " . DB_DUP ."\n";

    undef $db ;
    undef $cursor ;
    untie %hash ;

}

{
    # db_stat

    my $lex = new LexFile $Dfile ;
    my $recs = ($BerkeleyDB::db_version >= 3.1 ? "bt_ndata" : "bt_nrecs") ;
    my %hash ;
    my ($k, $v) ;
    ok 200, my $db = new BerkeleyDB::Btree -Filename => $Dfile, 
				     -Flags    => DB_CREATE,
				 	-Minkey	=>3 ,
					-Pagesize	=> 2 **12 
					;

    my $ref = $db->db_stat() ; 
    ok 201, $ref->{$recs} == 0;
    ok 202, $ref->{'bt_minkey'} == 3;
    ok 203, $ref->{'bt_pagesize'} == 2 ** 12;

    # create some data
    my %data =  (
		"red"	=> 2,
		"green"	=> "house",
		"blue"	=> "sea",
		) ;

    my $ret = 0 ;
    while (($k, $v) = each %data) {
        $ret += $db->db_put($k, $v) ;
    }
    ok 204, $ret == 0 ;

    $ref = $db->db_stat() ; 
    ok 205, $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::Btree );
   @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 ;

    BEGIN { push @INC, '.'; }    
    eval 'use SubDB ; ';
    main::ok 206, $@ eq "" ;
    my %h ;
    my $X ;
    eval '
	$X = tie(%h, "SubDB", -Filename => "dbbtree.tmp", 
			-Flags => DB_CREATE,
			-Mode => 0640 );
	' ;

    main::ok 207, $@ eq "" && $X ;

    my $ret = eval '$h{"fred"} = 3 ; return $h{"fred"} ' ;
    main::ok 208, $@ eq "" ;
    main::ok 209, $ret == 7 ;

    my $value = 0;
    $ret = eval '$X->db_put("joe", 4) ; $X->db_get("joe", $value) ; return $value' ;
    main::ok 210, $@ eq "" ;
    main::ok 211, $ret == 10 ;

    $ret = eval ' DB_NEXT eq main::DB_NEXT ' ;
    main::ok 212, $@ eq ""  ;
    main::ok 213, $ret == 1 ;

    $ret = eval '$X->A_new_method("joe") ' ;
    main::ok 214, $@ eq "" ;
    main::ok 215, $ret eq "[[10]]" ;

    undef $X;
    untie %h;
    unlink "SubDB.pm", "dbbtree.tmp" ;

}

{
    # DB_RECNUM, DB_SET_RECNO & DB_GET_RECNO

    my $lex = new LexFile $Dfile ;
    my %hash ;
    my ($k, $v) = ("", "");
    ok 216, my $db = new BerkeleyDB::Btree 
				-Filename  => $Dfile, 
			     	-Flags     => DB_CREATE,
			     	-Property  => DB_RECNUM ;


    # create some data
    my @data =  (
		"A zero",
		"B one",
		"C two",
		"D three",
		"E four"
		) ;

    my $ix = 0 ;
    my $ret = 0 ;
    foreach (@data) {
        $ret += $db->db_put($_, $ix) ;
	++ $ix ;
    }
    ok 217, $ret == 0 ;

    # db_get & DB_SET_RECNO
    $k = 1 ;
    ok 218, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 219, $k eq "B one" && $v == 1 ;

    $k = 3 ;
    ok 220, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 221, $k eq "D three" && $v == 3 ;

    $k = 4 ;
    ok 222, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 223, $k eq "E four" && $v == 4 ;

    $k = 0 ;
    ok 224, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 225, $k eq "A zero" && $v == 0 ;

    # cursor & DB_SET_RECNO

    # create the cursor
    ok 226, my $cursor = $db->db_cursor() ;

    $k = 2 ;
    ok 227, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 228, $k eq "C two" && $v == 2 ;

    $k = 0 ;
    ok 229, $cursor->c_get($k, $v, DB_SET_RECNO) == 0;
    ok 230, $k eq "A zero" && $v == 0 ;

    $k = 3 ;
    ok 231, $db->db_get($k, $v, DB_SET_RECNO) == 0;
    ok 232, $k eq "D three" && $v == 3 ;

    # cursor & DB_GET_RECNO
    ok 233, $cursor->c_get($k, $v, DB_FIRST) == 0 ;
    ok 234, $k eq "A zero" && $v == 0 ;
    ok 235, $cursor->c_get($k, $v, DB_GET_RECNO) == 0;
    ok 236, $v == 0 ;

    ok 237, $cursor->c_get($k, $v, DB_NEXT) == 0 ;
    ok 238, $k eq "B one" && $v == 1 ;
    ok 239, $cursor->c_get($k, $v, DB_GET_RECNO) == 0;
    ok 240, $v == 1 ;

    ok 241, $cursor->c_get($k, $v, DB_LAST) == 0 ;
    ok 242, $k eq "E four" && $v == 4 ;
    ok 243, $cursor->c_get($k, $v, DB_GET_RECNO) == 0;
    ok 244, $v == 4 ;

}