package jp.ac.u_ryukyu.ie.cr.tatsuki.xml;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.nio.ByteBuffer;
import java.util.Random;

import javax.xml.parsers.ParserConfigurationException;

import org.xml.sax.SAXException;

import jp.ac.u_ryukyu.cr.ie.tatsuki.xmlTestBbs.JuGrix;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;

public class TransactionPerSecondMeasurement {

    public static void main(String[] args) throws InterruptedException, IOException, ParserConfigurationException, SAXException {
        JuGrix jugrix = extendXmlTree.createJuGrix(true, 400);
        System.out.println("endCreatejugrix");
        System.out.println(Runtime.getRuntime().availableProcessors());

        if (args.length > 0) {

            if (args[0].equals("read"))
                readOnly(jugrix);
            if (args[0].equals("write"))
                readAndWrite(jugrix);

        } else
            System.out.println("Please with the argument");
    }

    private static void readAndWrite(JuGrix jugrix) throws InterruptedException, IOException {
        System.out.println("read and write");
        Runtime rt = Runtime.getRuntime();
        int cpuNum = rt.availableProcessors();

        File file = new File("./time/newTransactionReadAndWrite");
        PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));

        for (int THREAD_COUNT = 1; THREAD_COUNT <= cpuNum; THREAD_COUNT++) {
            ReadJungleThread readThread[] = new ReadJungleThread[THREAD_COUNT];
            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count] = new ReadJungleThread(jugrix);
            }

            WriteJungleThread writeThread = new WriteJungleThread(jugrix);
            writeThread.start();
            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count].start();
            }

            System.out.println("StartThread");

            Thread.sleep(1000);
            writeThread.set(false);
            int readCount = 0;

            for (int count = 0; THREAD_COUNT > count; count++) {
                readCount = readCount + readThread[count].getFindCount();
                readThread[count].set(false);
            }


            pw.println(THREAD_COUNT + " " + readCount);
            System.out.println(THREAD_COUNT + "readCount = " + readCount);
            Thread.sleep(1000);
        }
        pw.close();
    }

    private static void readOnly(JuGrix jugrix) throws InterruptedException, IOException {
        System.out.println("read only");
        Runtime rt = Runtime.getRuntime();
        int cpuNum = rt.availableProcessors();

        File file = new File("./time/readOnly");
        PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));
        for (int THREAD_COUNT = 1; THREAD_COUNT <= cpuNum; THREAD_COUNT++) {
            ReadJungleThread readThread[] = new ReadJungleThread[THREAD_COUNT];
            ;
            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count] = new ReadJungleThread(jugrix);
            }

            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count].start();
            }

            System.out.println("StartThread");

            Thread.sleep(1000);
            int readCount = 0;

            for (int count = 0; THREAD_COUNT > count; count++) {
                readCount = readCount + readThread[count].getFindCount();
                readThread[count].set(false);
                readThread[count] = null;
            }

            pw.println(THREAD_COUNT + " " + readCount);
            System.out.println(THREAD_COUNT + "readCount = " + readCount);
            System.gc();
            System.out.println("-------------GC--------------");
            Thread.sleep(1000);
        }

        pw.close();
    }

    static String key = "KEY";

    public static JungleTreeEditor createTree(int deep, NodePath path, JungleTreeEditor editor) {

        Random rnd = new Random();
        String value1 = String.valueOf(rnd.nextInt(1000));
        String value2 = String.valueOf(rnd.nextInt(1000));
        editor = editor.addNewChildAt(path, 0).b();
        editor = editor.putAttribute(path.add(0), key, ByteBuffer.wrap(value1.getBytes())).b();
        editor = editor.addNewChildAt(path, 1).b();
        editor = editor.putAttribute(path.add(0), key, ByteBuffer.wrap(value2.toString().getBytes())).b();
        deep++;

        if (deep < 12) {
            editor = createTree(deep, path.add(0), editor);
            editor = createTree(deep, path.add(1), editor);
        }
        return editor;
    }
}