package jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.test;


import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungle;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.Jungle;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.DefaultTraverser;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.*;
import java.nio.ByteBuffer;
import java.util.Random;

/**
 * Created by e115731 on 15/03/20.
 */
public class JungleBenchMark {
    public static void main(String[] args) throws InterruptedException, IOException, ParserConfigurationException, SAXException {


        if (args.length < 1) {
            System.out.println("args[0] = read or find, args[1] = separate or nonSeparate");
            System.exit(0);
        }

        Runtime rt = Runtime.getRuntime();
        int cpuNum = rt.availableProcessors();

        Jungle jungle = new DefaultJungle(null, "hoge", new DefaultTreeEditor(new DefaultTraverser()));
        JungleTree trees[] = new JungleTree[cpuNum];

        JungleTree tree = jungle.createNewTree("tree");
        JungleTreeEditor editor = tree.getTreeEditor();
        editor = editor.putAttribute(new DefaultNodePath(), "key", ByteBuffer.wrap(String.valueOf(0).getBytes())).b();
        JungleTreeEditor newEditor = createTree(0, new DefaultNodePath(), editor);
        if (newEditor.success().isA()) {
            System.out.println("success faild");
            System.exit(1);
        }
        trees[0] = tree;

        File file = new File("./time/" + args[1] + "JungleTree" + args[0] + "Time");
        PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));
        JungleBenchMarkThread readThread[] = null;

        for (int THREAD_COUNT = 1; THREAD_COUNT <= cpuNum; THREAD_COUNT++) {

            readThread = new JungleBenchMarkThread[THREAD_COUNT];

            if (args[0].equals("read")) {
                for (int count = 0; THREAD_COUNT > count; count++) {
                    readThread[count] = new readTreeAttributeThread(tree);
                }
            } else if (args[0].equals("find")) {
                for (int count = 0; THREAD_COUNT > count; count++) {
                    readThread[count] = new findTreeAttributeThread(tree);
                }
            } else {
                System.out.println("not allow args");
                System.exit(0);
            }

            JungleWriteThread writeThread = null;
            if (args[1].equals("write")) {
                writeThread = new JungleWriteThread(tree);
                writeThread.start();
            }
            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count].start();
            }

            System.out.println("StartThread");

            Thread.sleep(1000);
            if (args[1].equals("write")) {
                writeThread.set(false);
                writeThread.get();
            }
            long 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 Integer nodeNum = 1;

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

        Random rnd = new Random();
        String value1 = String.valueOf(nodeNum);
        nodeNum++;
        String value2 = String.valueOf(nodeNum);
        nodeNum++;

        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(1), "key", ByteBuffer.wrap(value2.getBytes())).b();
        deep++;

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