File Coverage

blib/lib/DBIx/Recordset/DBSeq.pm

Criterion	Covered	Total	%
statement	3	24	12.5
branch	0	14	0.0
condition	0	7	0.0
subroutine	1	3	33.3
pod	0	2	0.0
total	4	50	8.0

###################################################################################
#
#   DBIx::Recordset - Copyright (c) 1997-2000 Gerald Richter / ECOS
#
#   You may distribute under the terms of either the GNU General Public
#   License or the Artistic License, as specified in the Perl README file.
#
#   THIS IS BETA SOFTWARE!
#
#   THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR
#   IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
#   WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
#
#   $Id: DBSeq.pm,v 1.5 2000/06/26 05:16:18 richter Exp $
#
###################################################################################


package DBIx::Recordset::DBSeq ;

use strict 'vars' ;



## ----------------------------------------------------------------------------
##
## new
##
## creates a new DBIx::Recordset::DBSeq object.
##
## $dbh          = Database handle
## $table        = table where to keep sequences
##

sub new

    {
    my ($class, $dbh, $table, $min, $max) = @_ ;



    my $self = {
                '*Debug'      => $DBIx::Recordset::Debug,
                '*dbh'        => $dbh,
                '*table'      => $table,
                '*DefaultMin' => $min || 1,
                '*DefaultMax' => $max || 'NULL',
               } ;

    bless ($self, $class) ;

    return $self ;
    }



## ----------------------------------------------------------------------------
##
## NextVal
##
## get next value from counter
##
## in   $name = counter name
##


sub NextVal

    {
    my ($self, $name) = @_ ;

    my $dbh = $self -> {'*dbh'} ;

    $dbh -> do ("lock table $self->{'*table'} write") or die "Cannot lock $self->{'*table'} ($DBI::errstr)" ;



    my $sth = $dbh -> prepare ("select cnt,maxcnt from $self->{'*table'} where name=?") or die "Cannot prepare select for $self->{'*table'} ($DBI::errstr)" ;

    $sth -> execute ($name) ;

    my $row = $sth -> fetchrow_arrayref ;
    my $cnt ;
    my $max ;

    if (!$row)
        {
        $cnt = $self->{'*DefaultMin'} ;
        $max = $self->{'*DefaultMax'} ;
        my $cnt1 = $cnt + 1 ;
        $dbh -> do ("insert into $self->{'*table'} (name,cnt,maxcnt) values ('$name',$cnt1,$max)") or die "Cannot insert $self->{'*table'} ($DBI::errstr)" ;
        }
    else
        {
        $cnt = $row -> [0] ;
        die "Max count reached for sequence $name" if (defined ($row->[1]) && $cnt+1 > $row->[1]) ;
        $dbh -> do ("update $self->{'*table'} set cnt=cnt+1 where name='$name'") or die "Cannot update $self->{'*table'} ($DBI::errstr)" ;
        }

    $dbh -> do ("unlock table") or die "Cannot unlock $self->{'*table'} ($DBI::errstr)" ;

    return $cnt ;
    }

1;

__END__