<?php
/* ================================================================
   Copyright (C) 2014,2016 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.
================================================================ */

require("transactd.php");

use BizStation\Transactd\Transactd;
use BizStation\Transactd\Benchmark;
use BizStation\Transactd\Tabledef;
use BizStation\Transactd\Database;
use BizStation\Transactd\Table;

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 Database();
    $db->create($URI);
    if ($db->stat() == 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, Transactd::TYPE_SCHEMA_BDF, Transactd::TD_OPEN_NORMAL);
    assertEquals($db->stat(), 0);
    // create table
    $dbdef = $db->dbDef();
    assertNotEquals($dbdef, null);
    $td = new Tabledef();
    // Set table schema codepage to UTF-8
    //   - codepage for field NAME and tableNAME
    $td->schemaCodePage = Transactd::CP_UTF8;
    $td->setTableName(TABLENAME);
    $td->setFileName(TABLENAME . '.dat');
    // Set table default charaset index
    //    - default charset for field VALUE
    $td->charsetIndex = Transactd::charsetIndex(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 = 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 = 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
    $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($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($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($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($tb)
{
    $start = 1;
    $tb->setKeyNum(0);
    $tb->clearBuffer();
    $tb->setFilter('*', 1, RECORD_UNIT);
    $tb->setFV(FDI_ID, $start);
    $tb->find(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(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($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(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(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 Database();
    $db->open($URI, Transactd::TYPE_SCHEMA_BDF, Transactd::TD_OPEN_NORMAL);
    assertEquals($db->stat(), 0);
    $tb = $db->openTable(TABLENAME);
    assertEquals($db->stat(), 0);
    
    if (($funcNum === -1) || ($funcNum === 0)) {
        deleteAll($db, $tb);
        Benchmark::start();
        $ret = insert($tb);
        Benchmark::showTimeSec($ret, ': insert ' . RECORD_COUNT);
    }
    
    if (($funcNum === -1) || ($funcNum === 1)) {
        deleteAll($db, $tb);
        Benchmark::start();
        $ret = insertTransaction($db, $tb);
        Benchmark::showTimeSec($ret, ': insertTransaction ' . RECORD_COUNT . ' with transaction per ' . RECORD_UNIT);
    }
    
    if (($funcNum === -1) || ($funcNum === 2)) {
        deleteAll($db, $tb);
        Benchmark::start();
        $ret = insertBulk($tb);
        Benchmark::showTimeSec($ret, ': insertBulk ' . RECORD_COUNT . ' with bulkmode per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 3)) {
        Benchmark::start();
        $ret = read($tb);
        Benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 4)) {
        Benchmark::start();
        $ret = readSnapshot($db, $tb);
        Benchmark::showTimeSec($ret, ': read with snapshot ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 5)) {
        Benchmark::start();
        $ret = readRange($tb);
        Benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT . ' with range per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 6)) {
        Benchmark::start();
        $ret = readRangeSnapshot($db, $tb);
        Benchmark::showTimeSec($ret, ': read ' . RECORD_COUNT . ' with snapshot and range per ' . RECORD_UNIT);
    }
    if (($funcNum === -1) || ($funcNum === 7)) {
        Benchmark::start();
        $ret = update($tb);
        Benchmark::showTimeSec($ret, ': update ' . RECORD_COUNT);
    }
    if (($funcNum === -1) || ($funcNum === 8)) {
        Benchmark::start();
        $ret = updateTransaction($db, $tb);
        Benchmark::showTimeSec($ret, ': updateTransaction ' . RECORD_COUNT . ' with transaction per ' . RECORD_UNIT);
    }
}
main($argc, $argv);