X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~yarrgweb/git?a=blobdiff_plain;f=yarrg%2FCommodsDatabase.pm;h=1f977d40b7cca3d40e8dd030b3a253165793512f;hb=HEAD;hp=2ea674d6963e489647b779fb880b464931fc1684;hpb=a9583fa8a29e889f435ee4d6cf3dc6a5979a6bd5;p=ypp-sc-tools.db-test.git diff --git a/yarrg/CommodsDatabase.pm b/yarrg/CommodsDatabase.pm index 2ea674d..1f977d4 100644 --- a/yarrg/CommodsDatabase.pm +++ b/yarrg/CommodsDatabase.pm @@ -35,6 +35,7 @@ use warnings; use DBI; use POSIX; +use DBD::SQLite; use Commods; @@ -44,22 +45,66 @@ BEGIN { $VERSION = 1.00; @ISA = qw(Exporter); @EXPORT = qw(&db_setocean &db_writer &db_connect $dbh - &db_filename &db_doall &db_onconflict &db_setdatadir); + &db_filename &db_doall &db_onconflict + &dbr_filename &dbr_connect &db_connect_core + &dumptab_head &dumptab_row_hashref + &db_chkcommit &db_check_referential_integrity); %EXPORT_TAGS = ( ); @EXPORT_OK = qw(); } +sub dbr_filename ($$) { + my ($datadir,$oceanname) = @_; + return "$datadir/OCEAN-$oceanname.db"; +} +sub dbr_connect ($$) { + my ($datadir,$ocean) = @_; + return db_connect_core(dbr_filename($datadir,$ocean)); +} + +sub db_connect_core ($) { + my ($fn)= @_; + my $opts = { AutoCommit=>0, + RaiseError=>1, ShowErrorStatement=>1, + sqlite_unicode=>1 }; + + # DBI now wants to start a transaction whenever we even say + # SELECT. But this doesn't work if the DB is readonly. We can + # work around this by setting autocommit, in which case there is + # no need for a transaction for read-only db commands. Autocommit + # is (obviously) safe with readonly operations. But callers in + # yarrg do not specify to us whether they intend to write. So we + # decide, by looking at the file mode. And as belt-and-braces we + # set sqlite's own readonly flag as well. + # http://stackoverflow.com/questions/30082008/attempt-to-write-a-readonly-database-but-im-not + # http://stackoverflow.com/questions/35208727/can-sqlite-db-files-be-made-read-only + # http://cpansearch.perl.org/src/ISHIGAKI/DBD-SQLite-1.39/Changes + # (see entry for 1.38_01) + # http://stackoverflow.com/questions/17793672/perl-dbi-treats-setting-sqlite-db-cache-size-as-a-write-operation-when-subclassi + # https://rt.cpan.org/Public/Bug/Display.html?id=56444# + my $readonly = + (access $fn, POSIX::W_OK) ? 0 : + ($! == EACCES) ? 1 : + ($! == ENOENT) ? 0 : + die "$fn access(,W_OK) $!"; + if ($readonly) { + $opts->{sqlite_open_flags} = DBD::SQLite::OPEN_READONLY; + $opts->{AutoCommit}=1; + } + + my $h= DBI->connect("dbi:SQLite:$fn",'','',$opts) + or die "$fn $DBI::errstr ?"; + return $h; + # default timeout is 30s which is plenty +} + our $dbfn; our $dbh; -our $datadir= '.'; -sub db_setdatadir ($) { - $datadir= $_[0]; -} sub db_setocean ($) { my ($oceanname) = @_; - $dbfn= "$datadir/OCEAN-$oceanname.db"; + $dbfn= dbr_filename('.',$oceanname); } sub db_filename () { return $dbfn; @@ -93,13 +138,7 @@ sub db_writer () { } sub db_connect () { - return if $dbh; - $dbh= DBI->connect("dbi:SQLite:$dbfn",'','', - { AutoCommit=>0, - RaiseError=>1, ShowErrorStatement=>1, - unicode=>1 }) - or die "$dbfn $DBI::errstr ?"; - # default timeout is 30s which is plenty + $dbh= db_connect_core($dbfn); } sub db_doall ($) { @@ -108,4 +147,153 @@ sub db_doall ($) { } } +#---------- table dump helper ---------- + +sub dumptab_head ($$$) { + my ($fh,$w,$cols) = @_; + printf $fh "|%-${w}s", $_ foreach @$cols; print $fh "|\n"; + print $fh "+",('-'x$w) foreach @$cols; print $fh "+\n"; +} + +sub dumptab_row_hashref ($$$$) { + my ($fh,$w,$cols,$row) = @_; + printf $fh "|%-$w.${w}s", + (defined $row->{$_} ? $row->{$_} : 'NULL') + foreach @$cols; + print $fh "\n"; +} + +#---------- referential integrity constraints ---------- + +# SQLite doesn't support foreign key constraints so we do it by steam: + +sub nooutput ($) { + my ($stmts) = @_; + my $ekindcount= 0; + my $letxt= ''; + foreach my $stmt (split /\;/, $stmts) { + next unless $stmt =~ /\S/; + + my $etxt= ''; + $stmt =~ s/^([ \t]*\#.*)$/ $etxt .= $1."\n"; ''; /mge; + $etxt= $letxt unless length $etxt; + $letxt= $etxt; + + $stmt =~ s/^\s+//; $stmt =~ s/\s+$//; + my $sth= $dbh->prepare($stmt); + $sth->execute(); + my $row; + my $ecount= 0; + my @cols= @{ $sth->{NAME_lc} }; + my $w= 11; + while ($row= $sth->fetchrow_hashref) { + if (!$ecount++) { + print STDERR "REFERENTIAL INTEGRITY ERROR\n"; + print STDERR "\n$etxt\n $stmt\n\n"; + dumptab_head(\*STDERR,$w,\@cols); + } + if ($ecount>5) { print STDERR "...\n"; last; } + dumptab_row_hashref(\*STDERR,$w,\@cols,$row); + } + next unless $ecount; + + $ekindcount++; + print STDERR "\n\n"; + } + die "REFERENTIAL INTEGRITY ERRORS $ekindcount\n" + if $ekindcount; +} + +sub db_check_referential_integrity ($) { + my ($full) = @_; + # non-full is done only for market data updates; it avoids + # detecting errors which are essentially missing metadata and + # old schemas, etc. + + foreach my $bs (qw(buy sell)) { + nooutput(< 0 + GROUP BY ordval + HAVING count(*) > 1; + + # For every class, posinclass is dense from 1 to maxposinclass, + # apart from the commods for which it is zero. + SELECT commodclass,commodclassid,posinclass,count(*) + FROM commods JOIN commodclasses USING (commodclassid) + WHERE posinclass > 0 + GROUP BY commodclassid,posinclass + HAVING count(*) > 1; + SELECT commodclass,commodclassid,count(*) + FROM commods JOIN commodclasses USING (commodclassid) + WHERE posinclass > 0 + GROUP BY commodclassid + HAVING count(*) != maxposinclass; + SELECT * + FROM commods JOIN commodclasses USING (commodclassid) + WHERE posinclass < 0 OR posinclass > maxposinclass; + +END + } +} + +sub db_chkcommit ($) { + my ($full) = @_; + db_check_referential_integrity($full); + $dbh->commit(); +} + 1;