package examples;

import com.xxdb.DBConnection;
import com.xxdb.comm.ErrorCodeInfo;
import com.xxdb.data.*;
import com.xxdb.data.Vector;
import com.xxdb.multithreadedtablewriter.MultithreadedTableWriter;

import java.time.LocalDate;
import java.util.*;


/**
 *  MTW (MultihreadedTableWriter) is used to write streaming data into dfs table or streamTable.
 *  when write dfs table, we recommend that the thread number equals to the partition number of the dfs table.
 */
public class MultithreadedTableWriterDemo {

    private static String host = "localhost";
    private static int port = 8848;
    private static String username = "admin";
    private static String password = "123456";

    public static void main(String[] args) throws Exception {
        DBConnection conn= new DBConnection();
        conn.connect(host, port, "admin", "123456");
        Random random = new Random();
        String script =
                "dbName = 'dfs://valuedb3'" +
                        "if (exists(dbName))" +
                        "{" +
                        "dropDatabase(dbName);" +
                        "}" +
                        "datetest = table(1000:0,`date`symbol`id,[DATE, SYMBOL, LONG]);" +
                        "db = database(directory= dbName, partitionType= HASH, partitionScheme=[INT, 10]);" +
                        "pt = db.createPartitionedTable(datetest,'pdatetest','id');";
        conn.run(script);
        MultithreadedTableWriter multithreadedTableWriter_ = new MultithreadedTableWriter(host, port, "admin", "123456", "dfs://valuedb3", "pdatetest",
                false, false, null, 10000, 1,
                5, "id", new int[]{Vector.COMPRESS_LZ4, Vector.COMPRESS_LZ4, Vector.COMPRESS_DELTA});
        ErrorCodeInfo ret;
        try
        {
            for (int i = 0; i < 10000; ++i)
            {
                ret = multithreadedTableWriter_.insert(LocalDate.of(2022, 3, 23), "AAAAAAAB", random.nextInt() % 10000);
                if(ret.hasError()){
                    System.out.println("insert failed: " + ret.getErrorInfo());
                }
            }
        }
        catch (Exception e) // mtw may throw exception
        {
            System.out.println("MTW exit with exception {0}" + e.getMessage());
        }

        //wait for the Completion by join the thread.
        multithreadedTableWriter_.waitForThreadCompletion();
        MultithreadedTableWriter.Status writeStatus = new MultithreadedTableWriter.Status();
        writeStatus = multithreadedTableWriter_.getStatus();

        // check if mtw has errors.
        if (writeStatus.hasError())
        {
            System.out.println("error in writing !");
        }
        System.out.println("writeStatus: {0}\n" + writeStatus.toString());
        System.out.println(((BasicLong)conn.run("exec count(*) from pt")).getLong());
    }
}