#!/usr/bin/perl
#SQLite mem test
use strict; use warnings;
use Time::HiRes;
use Benchmark qw/cmpthese/;
use DBI;

my $href;
my @t_synsets;
my $DB_DISK='./wnjpn.db';
my $DB_TO_MEM='./wn-synlink.db';

#load @ARGV[0] of synset for lookup test
load_test_synsets();

my %tests = (
    '01_substr' => \&test1,
    '02_sqlite_mem' => \&test2,
#   '03_sqlite_disk' => \&test3,
);
cmpthese(
    -20, #for 20 cpu secs
    \%tests
);
sub test1{ #substr
    $href={};
    open(my $fh, "<", "04.txt") or die $!;
    while(<$fh>){
        chomp;
        push @{ $href->{ substr($_,0,10)} }, [ substr($_,10,10), substr($_,20)];
    }
    close $fh;

    #iterate
    my $cnt=0;
    for( @t_synsets ){
        if (exists $href->{$_} ){
            $cnt += @{ $href->{$_} };
        }
    }
    #print "test1:arrays cnt=$cnt\n";
}
sub test2{ #sqlite mem
    my $mem_dbh = DBI->connect('dbi:SQLite:dbname=:memory:');
    $mem_dbh->sqlite_backup_from_file($DB_TO_MEM);
    my $sth=$mem_dbh->prepare("select synset2 from synlink where synset1=?");

    #iterate
    my $cnt=0;
    for( @t_synsets ){
        $sth->execute($_);
        my $tmp=$sth->fetchall_arrayref;
        $cnt+=@{$tmp};
    }
    $sth->finish;
    #print "test2:arrays cnt=$cnt\n";

    $mem_dbh->disconnect;
}
sub test3{ #sqlite disk
    my $dbh = DBI->connect("dbi:SQLite:dbname=$DB_DISK") or die DBI->errstr;
    my $sth=$dbh->prepare("select synset2 from synlink where synset1=?");

    #iterate
    my $cnt=0;
    for( @t_synsets ){
        $sth->execute($_);
        my $tmp=$sth->fetchall_arrayref;
        $cnt+=@{$tmp};
    }
    $sth->finish;
    #print "test3:arrays cnt=$cnt\n";

    $dbh->disconnect;
}
#$ARGV[0] synsets to lookup
sub load_test_synsets{
    my $dbh=DBI->connect("dbi:SQLite:./wnjpn.db","","", {
        AutoCommit=>0,
        #sqlite_use_immediate_transaction=>1,
        #sqlite_unicode=>1,
        RaiseError=>1,
    }) or die DBI->errstr;

    if ( defined($ARGV[0]) && $ARGV[0] =~ /^\d+$/){
        @t_synsets = map{$_->[0]}
            @{$dbh->selectall_arrayref("select synset FROM synset limit $ARGV[0]")};
    } else {
        @t_synsets = map{$_->[0]}
            @{$dbh->selectall_arrayref("select synset FROM synset")};
    }
    $dbh->disconnect;
}