<?php
/* ================================================================
   Copyright (C) 2014 BizStation Corp All rights reserved.

   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., 59 Temple Place - Suite 330, Boston, MA  
   02111-1307, USA.
================================================================ */
mb_internal_encoding('UTF-8');

require_once("transactd.php");
use BizStation\Transactd as Bz;
define('TABLENAME', 'users');
define('FDI_ID', 0);
define('FDI_NAME', 1);
define('BULKBUFSIZE', 65535 - 1000);
define('RECORD_COUNT', 20000);
define('RECORD_UNIT', 20);


function assertEquals($a, $b, $msg = '') {
    if ($a !== $b) {
        echo("$a !== $b $msg");
        return false;
    }
    return true;
}
function assertNotEquals($a, $b, $msg = '') {
    if ($a === $b) {
        echo("$a === $b $msg");
        return false;
    }
    return true;
}

function create($URI) {
    // create database
    $db = new Bz\database();
    $db->create($URI);
    if ($db->stat() == Bz\transactd::STATUS_TABLE_EXISTS_ERROR) {
        $db->open($URI);
        assertEquals($db->stat(), 0);
        $db->drop();
        assertEquals($db->stat(), 0);
        $db->create($URI);
    }
    assertEquals($db->stat(), 0);
    $db->open($URI, Bz\transactd::TYPE_SCHEMA_BDF, Bz\transactd::TD_OPEN_NORMAL);
    assertEquals($db->stat(), 0);
    // create table
    $dbdef = $db->dbDef();
    assertNotEquals($dbdef, NULL);
    $td = new Bz\tabledef();
    // Set table schema codepage to UTF-8
    //   - codepage for field NAME and tableNAME
    $td->schemaCodePage = Bz\transactd::CP_UTF8;
    $td->setTableName(TABLENAME);
    $td->setFileName(TABLENAME . '.dat');
    // Set table default charaset index
    //    - default charset for field VALUE
    $td->charsetIndex = Bz\transactd::charsetIndex(Bz\transactd::CP_UTF8);
    $td->id = 1;
    $td->pageSize = 2048;
    $dbdef->insertTable($td);
    assertEquals($dbdef->stat(), 0);
    // id
    $fd = $dbdef->insertField($td->id, FDI_ID);
    $fd->setName('id');
    $fd->type = Bz\transactd::ft_autoinc;
    $fd->len = 4;
    $dbdef->updateTableDef($td->id);
    assertEquals($dbdef->stat(), 0);
    // user_id
    $fd = $dbdef->insertField($td->id, FDI_NAME);
    $fd->setName('name');
    $fd->type = Bz\transactd::ft_myvarchar;
    $fd->len = 100;
    $dbdef->updateTableDef($td->id);
    assertEquals($dbdef->stat(), 0);
    // key
    $kd = $dbdef->insertKey($td->id, 0);
    $kd->segment(0)->fieldNum = 0;
    $kd->segment(0)->flags->bit8 = 1;
    $kd->segment(0)->flags->bit1 = 1;
    $kd->segmentCount = 1;
    $td->primaryKeyNum = 0;
    $dbdef->updateTableDef($td->id);
    assertEquals($dbdef->stat(), 0);
    // test open
    $tb = $db->openTable(TABLENAME);
    assertEquals($db->stat(), 0);
}

function deleteAll($db, $tb) {
    $db->beginTrn();
    $tb->clearBuffer();
    for ($i = 1; $i <= RECORD_COUNT; $i++) {
        $tb->setFV(FDI_ID, $i);
        $tb->seek();
        if ($tb->stat() === 0) {
            $tb->del();
            if (!assertEquals($tb->stat(), 0, 'deleteAll')) {
                $db->endTrn();
                return false;
            }
        }
    }
    $db->endTrn();
    return true;
}

function insert($db, $tb) {
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    for ($i = 1; $i <= RECORD_COUNT; $i++) {
        $tb->setFV(FDI_ID, $i);
        $tb->setFV(FDI_NAME, $i);
        $tb->insert();
        if (!assertEquals($tb->stat(), 0, 'insert'))
            return false;
    }
    return true;
}

function insertTransaction($db, $tb) {
    $tb->setKeyNum(0);
    $start = 1;
    $tb->clearBuffer();
    while ($start < RECORD_COUNT) {
        $db->beginTrn();
        for ($i = $start; $i < $start + RECORD_UNIT; $i++) {
            $tb->setFV(FDI_ID, $i);
            $tb->setFV(FDI_NAME, $i);
            $tb->insert();
            if (!assertEquals($tb->stat(), 0, 'insertTransaction')) {
                $db->abortTrn();
                return false;
            }
        }
        $db->endTrn();
        $start += RECORD_UNIT;
    }
    return true;
}

function insertBulk($db, $tb) {
    $tb->setKeyNum(0);
    $start = 1;
    $tb->clearBuffer();
    while ($start < RECORD_COUNT) {
        $tb->beginBulkInsert(BULKBUFSIZE);
        for ($i = $start; $i < $start + RECORD_UNIT; $i++) {
            $tb->setFV(FDI_ID, $i);
            $tb->setFV(FDI_NAME, $i);
            $tb->insert();
            if (!assertEquals($tb->stat(), 0, 'insertBulk')) {
                $tb->commitBulkInsert();
                return false;
            }
        }
        $tb->commitBulkInsert();
        $start += RECORD_UNIT;
    }
    return true;
}

function read($db, $tb) {
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    for ($i = 1; $i <= RECORD_COUNT; $i++) {
        $tb->setFV(FDI_ID, $i);
        $tb->seek();
        if (($tb->stat() !==0) || ($tb->getFVint(FDI_ID) !== $i))
        {
            echo("seek stat = ".$tb->stat()." value ".$i." = ".$tb->getFVint(FDI_ID));
            return false;
        }
    }
    return true;
}

function readSnapshot($db, $tb) {
    $tb->setKeyNum(0);
    $db->beginSnapshot();
    $tb->clearBuffer();
    for ($i = 1; $i <= RECORD_COUNT; $i++) {
        $tb->setFV(FDI_ID, $i);
        $tb->seek();
        if (($tb->stat() !==0) || ($tb->getFVint(FDI_ID) !== $i))
        {
            echo("seek stat = ".$tb->stat()." value ".$i." = ".$tb->getFVint(FDI_ID));
            $db->endSnapshot();
            return false;
        }
    }
    $db->endSnapshot();
    return true;
}

function readRange($db, $tb) {
    $start = 1;
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    $tb->setFilter('*', 1, RECORD_UNIT);
    $tb->setFV(FDI_ID, $start);
    $tb->find(Bz\table::findForword);
    while ($start < RECORD_COUNT) {
        for ($i = $start; $i < $start + RECORD_UNIT; $i++) {
            if (/*($tb->stat() !==0) || */($tb->getFVint(FDI_ID) !== $i))
            {
                echo("find stat = ".$tb->stat()." value ".$i." = ".$tb->getFVint(FDI_ID));
                return false;
            }
            $tb->findNext();
        }
        $start += RECORD_UNIT;
    }
    return true;
}

function readRangeSnapshot($db, $tb) {
    $start = 1;
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    $tb->setFilter('*', 1, RECORD_UNIT);
    $tb->setFV(FDI_ID, $start);
    $db->beginSnapshot();
    $tb->find(Bz\table::findForword);
    while ($start < RECORD_COUNT) {
        for ($i = $start; $i < $start + RECORD_UNIT; $i++) {
            if (/*($tb->stat() !==0) || */($tb->getFVint(FDI_ID) !== $i))
            {
                echo("find stat = ".$tb->stat()." value ".$i." = ".$tb->getFVint(FDI_ID));
                $db->endSnapshot();
                return false;
            }
            $tb->findNext();
        }
        $start += RECORD_UNIT;
    }
    $db->endSnapshot();
    return true;
}

function update($db, $tb) {
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    for ($i = 1; $i <= RECORD_COUNT; $i++) {
        $tb->setFV(FDI_ID, $i);
        $tb->setFV(FDI_NAME, ($i + 1));
        $tb->update(Bz\table::changeInKey);
        if ($tb->stat() !==0)
        {
            echo("update stat = ".$tb->stat());
            return false;
        }
    }
    return true;
}

function updateTransaction($db, $tb) {
    $tb->setKeyNum(0);
    $start = 1;
    $tb->clearBuffer();
    while ($start < RECORD_COUNT) {
        $db->beginTrn();
        for ($i = $start; $i < $start + RECORD_UNIT; $i++) {
            $tb->setFV(FDI_ID, $i);
            $tb->setFV(FDI_NAME, ($i + 2));
            $tb->update(Bz\table::changeInKey);
            if ($tb->stat() !==0)
            {
                echo("update stat = ".$tb->stat());
                $db->abortTrn();
                return false;
            }
        }
        $db->endTrn();
        $start += RECORD_UNIT;
    }
    return true;
}

function main($argc, $argv) {
   if ($argc < 4) {
        echo("usage: php bench.php databaseUri processNumber functionNumber\n");
        echo("\t --- functionNumber list ---\n");
        echo("\t-1: all function\n");
        echo("\t 0: Insert\n");
        echo("\t 1: Insert in transaction. 20rec x 1000times\n");
        echo("\t 2: Insert by bulkmode. 20rec x 1000times\n");
        echo("\t 3: read each record\n");
        echo("\t 4: read each record with snapshot\n");
        echo("\t 5: read range. 20rec x 1000times\n");
        echo("\t 6: read range with snapshpot. 20rec x 1000times\n");
        echo("\t 7: update\n");
        echo("\t 8: update in transaction. 20rec x 1000times\n");
        echo("example : php bench.php \"tdap://localhost/test?dbfile=test.bdf\" 0 -1\n");
        return;
   }

    $URI = $argv[1];
    // Currenty $proc is ignored.
    $proc = (int)$argv[2];
    $funcNum = (int)$argv[3];

    create($URI);
    echo("CreateDataBase success.\n");
    echo("Start Bench mark Insert Items = ".RECORD_COUNT."\n");
    echo(date(DATE_ATOM)."\n");
    echo($URI."\n");
    echo("----------------------------------------\n");
    
    $db = new Bz\database();
    $db->open($URI, Bz\transactd::TYPE_SCHEMA_BDF, Bz\transactd::TD_OPEN_NORMAL);
    assertEquals($db->stat(), 0);
    $tb = $db->openTable(TABLENAME);
    assertEquals($db->stat(), 0);
    
    if (($funcNum === -1) || ($funcNum === 0)) {
        deleteAll($db, $tb);
        Bz\benchmark::start();
        $ret = insert($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': insert ' . RECORD_COUNT);
    }
    
    if (($funcNum === -1) || ($funcNum === 1)) {
        deleteAll($db, $tb);
        Bz\benchmark::start();
        $ret = insertTransaction($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': insertTransaction ' . RECORD_COUNT . ' with transaction per ' . RECORD_UNIT);
    }
    
    if (($funcNum === -1) || ($funcNum === 2)) {
        deleteAll($db, $tb);
        Bz\benchmark::start();
        $ret = insertBulk($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': insertBulk ' . RECORD_COUNT . ' with bulkmode per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 3)) {
        Bz\benchmark::start();
        $ret = read($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 4)) {
        Bz\benchmark::start();
        $ret = readSnapshot($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': read with snapshot ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 5)) {
        Bz\benchmark::start();
        $ret = readRange($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT . ' with range per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 6)) {
        Bz\benchmark::start();
        $ret = readRangeSnapshot($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT . ' with snapshot and range per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 7)) {
        Bz\benchmark::start();
        $ret = update($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': update ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 8)) {
        Bz\benchmark::start();
        $ret = updateTransaction($db, $tb);
        Bz\benchmark::showTimeSec($ret, ': updateTransaction ' . RECORD_COUNT . ' with transaction per ' . RECORD_UNIT);
    }
}
main($argc, $argv);