#     ndb-connectors: Wrappers for the NDBAPI
#       Copyright (C) 2006 MySQL, Inc.
    
#       This program is free software; you can redistribute it and/or modify
#     it under the terms of the GNU General Public License as published by
#     the Free Software Foundation; either version 2 of the License, or 
#     (at your option) any later version.
    
#     This program is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#     GNU General Public License for more details.

#     You should have received a copy of the GNU General Public License
#     along with this program; if not, write to the Free Software
#     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA


use lib 'blib/arch';
use lib 'blib';
use lib 'blib/lib';
use ndbapi;
use DBI;
use POSIX qw( ceil floor );
use Time::HiRes qw( gettimeofday tv_interval sleep );

$num_iter = shift;
$INSERT_NUM = shift;
do { print "Usage:\n\ttest.pl NUM_OF_ITERATIONS NUM_OF_ROWS\n";
  exit -1; }  unless $INSERT_NUM;
#$BATCH_SIZE = 100;
$BATCH_SIZE = 1;

### Connect to cluster

print "Dropping and recreating scehma\n\n";

$dbh = DBI->connect("dbi:mysql:host=localhost;read_default_group=client","root","" );


$dbh->do("CREATE DATABASE if not exists test");
$dbh->do("USE test");

$dbh->do("DROP TABLE if exists mytablename");
$dbh->do("CREATE TABLE
  mytablename
   (ATTR1 INT UNSIGNED not null,
    ATTR2 INT UNSIGNED NOT NULL,
    PRIMARY KEY USING HASH (ATTR1)
   )
ENGINE=NDBCLUSTER
");

print "connecting to cluster\n";

$connection = ndbapi::NdbFactory::createNdbClusterConnection();


if ($connection->connect(5,3,1)) {

    print "Connect to cluster management server failed.\n";
    exit -1;
}


if ($connection->wait_until_ready(30,30)) {
    print "Cluster was not ready within 30 secs.\n";
    exit -1;
} else { print "cluster ready\n"; }

$myNdb = $connection->createNdb("test");


if ($myNdb->init(4) == -1) {
    print $myNdb->getNdbError()->getMessage();

    exit -1;
}

print "running tests: \n";


my $before_time = [gettimeofday];

my $auto_id=0;

my $done=0;
foreach my $t (1..(ceil($INSERT_NUM/$BATCH_SIZE))) {

    my $myTransaction = $myNdb->startTransaction();
    while (! $myTransaction ) {
	sleep 1/1000;
	$myNdb->sendPollNdb(0);
    	$myTransaction = $myNdb->startTransaction();
    }

    my $val = $t*$BATCH_SIZE-$INSERT_NUM;
    my $offset = ($val > 0 ) ? $val : 0;

    foreach my $i (1..($BATCH_SIZE-$offset)) {

      my $myOperation = $myTransaction->getNdbOperation("mytablename");
      $myOperation->insertTuple();

      #my $auto_id = $myNdb->getAutoIncrementValue("mytablename",$BATCH_SIZE);
      $auto_id++;
      #print "$t - $i - $auto_id - $myOperation\n";

      $myOperation->equal("ATTR1", $auto_id);
      $myOperation->setVal("ATTR2",$i);

    }

    #my $ret = $myTransaction->execute($ndb::Commit);
    my $ret;
    {
	my $trans=$myTransaction;
    	$ret = $myTransaction->executeAsynchPrepare($ndb::Commit,
		sub { $done--; 
		print "ASYNC CALLED\n";
    		# We want to close transactions after the callback
    	         } );
    }

    if ($ret == -1) {
      print $myTransaction->getNdbError()->getMessage(), "\n";
    }
    $done++;
}

while ( $done>0 ) {
	$myNdb->sendPollNdb(3000,1);
	sleep 1/1000;
}

my $elapsed = tv_interval ( $before_time );

print "Insert time for $INSERT_NUM:\t$elapsed\n";


## Fetch list of ids. We could do this when we build it
## But then we couldn't test Scan Operations

my @ids;

my $myTransaction = $myNdb->startTransaction();


my $myScanOperation=$myTransaction->getNdbScanOperation("mytablename");

print $myTransaction->getNdbError()->getMessage() unless $myScanOperation;

if ($myScanOperation->readTuples($ndb::NdbOperation::LM_CommittedRead)) {
  print $myScanOperation->getNdbError()->getMessage();
}



my $myRecAttr=$myScanOperation->getValue("ATTR1");

$myTransaction->execute($ndb::NoCommit);

while ($myScanOperation->nextResultMt() == 0) {
  
  my $random_id = $myRecAttr->u_32_value();
  push(@ids,$random_id);
  
}

$myTransaction->close();


#Fetch records using NDBAPI

my $before_time = [gettimeofday];

foreach my $f (0..$num_iter) {

      my $myTransaction = $myNdb->startTransaction();
      if (!$myTransaction) {
        print $myNdb->getNdbError()->getMessage(); }

      my $myOper = $myTransaction->getNdbOperation("mytablename");
      $myOper->readTuple($NdbOperation::LockMode::LM_Read);

      my $id_num = $ids[rand($INSERT_NUM)];

      $myOper->equal("ATTR1",$id_num);

      my $myRecAttr = $myOper->getValue("ATTR2");

      if ($myTransaction->execute( $ndb::Commit ) == -1) {
        print $myTransaction->getNdbError()->getMessage(),"\n"; 
      }
      $foo=$myRecAttr->u_32_value();

      $myTransaction->close();

    }


my $elapsed = tv_interval ( $before_time );
print "Fetch time for NDBAPI - $num_iter:\t$elapsed\n";