package com.sunyard.sge.database;

import com.sunyard.SydApiException;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.Callable;
import java.util.concurrent.FutureTask;

import static com.sunyard.sge.database.SydApiBaseFunction.toBoolArray;
import static com.sunyard.sge.database.SydApiBaseFunction.toIntArray;
import static com.sunyard.sge.database.SydApiConfig.BATCH_SIZE;

/**
 * 长连接的真正执行过程转发到长连接线程池中执行。
 */
public class SydApiLongLinkWrap implements SydApi {

    private SydApiBuilder builder;

    public SydApiLongLinkWrap(SydApiBuilder builder) {
        this.builder = builder;
    }

    @Override
    public SydApi SYD_Short_Connect_Ex(String[] pcIpList, int[] iPortList, int iConnectTimeOut, int iDealTimeOut) {
        throw new UnsupportedOperationException("此对象为长连接专用");
    }

    @Override
    public void SYD_Disconnect_Ex() {
        // pass
    }

    @Override
    public String SYD_SM3_Hash_ShortData(byte[] pcData) {
        FutureTask<String> task = builder.getLongLinkWorks().fTask(new Callable<String>() {
            @Override
            public String call() throws Exception {
                return builder.get().SYD_SM3_Hash_ShortData(pcData);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public LongDataReturn<String, String> SYD_SM3_Hash_LongData(byte[] pcData, int iPkgNum, String pcProcData) {
        FutureTask<LongDataReturn<String, String>> task = builder.getLongLinkWorks().fTask(new Callable<LongDataReturn<String, String>>() {
            @Override
            public LongDataReturn<String, String> call() throws Exception {
                return builder.get().SYD_SM3_Hash_LongData(pcData, iPkgNum, pcProcData);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public byte[] SYD_SM4_ShortData(int iKeyIndex, byte[] pcData, int iFlag) {
        FutureTask<byte[]> task = builder.getLongLinkWorks().fTask(new Callable<byte[]>() {
            @Override
            public byte[] call() throws Exception {
                return builder.get().SYD_SM4_ShortData(iKeyIndex, pcData, iFlag);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public byte[] SYD_SM4_LongData(int iKeyIndex, byte[] pcData, int iPkgNum, int iFlag) {
        FutureTask<byte[]> task = builder.getLongLinkWorks().fTask(new Callable<byte[]>() {
            @Override
            public byte[] call() throws Exception {
                return builder.get().SYD_SM4_LongData(iKeyIndex, pcData, iPkgNum, iFlag);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }


    @Override
    public LongDataReturn<String, String> SYD_SM4Mac_LongData(int iKeyIndex, byte[] pcData, int iPkgNum, String pcProcData) {
        FutureTask<LongDataReturn<String, String>> task = builder.getLongLinkWorks().fTask(new Callable<LongDataReturn<String, String>>() {
            @Override
            public LongDataReturn<String, String> call() throws Exception {
                return builder.get().SYD_SM4Mac_LongData(iKeyIndex, pcData, iPkgNum, pcProcData);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public LongDataReturn<String, Boolean> SYD_SM4Mac_LongData(int iKeyIndex, byte[] pcData, int iPkgNum, String pcProcData, String pcMac) {
        FutureTask<LongDataReturn<String, Boolean>> task = builder.getLongLinkWorks().fTask(new Callable<LongDataReturn<String, Boolean>>() {
            @Override
            public LongDataReturn<String, Boolean> call() throws Exception {
                return builder.get().SYD_SM4Mac_LongData(iKeyIndex, pcData, iPkgNum, pcProcData, pcMac);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public String SYD_SM4Mac_ShortData(int iKeyIndex, byte[] pcData) {
        FutureTask<String> task = builder.getLongLinkWorks().fTask(new Callable<String>() {
            @Override
            public String call() throws Exception {
                return builder.get().SYD_SM4Mac_ShortData(iKeyIndex, pcData);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public boolean SYD_SM4Mac_ShortData(int iKeyIndex, byte[] pcData, String pcMac) {
        FutureTask<Boolean> task = builder.getLongLinkWorks().fTask(new Callable<Boolean>() {
            @Override
            public Boolean call() throws Exception {
                return builder.get().SYD_SM4Mac_ShortData(iKeyIndex, pcData, pcMac);
            }
        });
        try {
            return task.get();
        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public ISM4 initSM4(int iKeyIndex, int ende) {
        return new SydSM4(this, iKeyIndex, ende);
    }

    @Override
    public ISM3 initSM3() {
        return new SydSM3(this);
    }

    @Override
    public ISM4Mac initSM4Mac(int iKeyIndex) {
        return new SydSM4Mac(this, iKeyIndex);
    }


    @Override
    public List<byte[]> SYD_SM4_BatchData(int[] iKeyIndex, List<byte[]> pcData, int iFlag) {
        // 将数据进行分组，防止超长。
        List<List<byte[]>> dataGroups = new ArrayList<>();
        List<List<Integer>> keyGroups = new ArrayList<>();
        int totalLen = 0;

        List<byte[]> curDataGroup = new ArrayList<>();
        List<Integer> curKeyGroup = new ArrayList<>();

        for (int i = 0; i < pcData.size(); i++) {
            byte[] data = pcData.get(i);
            if (null == data) {
                data = new byte[0];
            }
            if (
                    curDataGroup.size() > 9                    // 分组小于等于 9 个
                            || (totalLen + data.length > BATCH_SIZE)   // 总大小小于 BATCH_SIZE
            ) {                // 扩展包
                dataGroups.add(curDataGroup);
                keyGroups.add(curKeyGroup);

                curDataGroup = new ArrayList<>();
                curKeyGroup = new ArrayList<>();
            }


            curDataGroup.add(data);
            curKeyGroup.add(iKeyIndex[i]);

            totalLen = data.length;

        }

        dataGroups.add(curDataGroup);
        keyGroups.add(curKeyGroup);

        // 并发处理
        List<FutureTask<List<byte[]>>> tasks = new ArrayList<>();
        for (int i = 0; i < dataGroups.size(); i++) {

            int finalI = i;
            FutureTask<List<byte[]>> task = builder.getLongLinkWorks().fTask(new Callable<List<byte[]>>() {
                @Override
                public List<byte[]> call() throws Exception {
                    SydApi api = builder.get();
                    return api.SYD_SM4_BatchData(toIntArray(keyGroups.get(finalI)), dataGroups.get(finalI), iFlag);
                }
            });
            tasks.add(task);
        }

        try {
            List<byte[]> ret = new ArrayList<>();
            for (FutureTask<List<byte[]>> t : tasks) {
                ret.addAll(t.get());
            }
            return ret;

        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public String[] SYD_SM4Mac_BatchData(int[] iKeyIndex, List<byte[]> pcData) {
        // 将数据进行分组，防止超长。
        List<List<byte[]>> dataGroups = new ArrayList<>();
        List<List<Integer>> keyGroups = new ArrayList<>();
        int totalLen = 0;

        List<byte[]> curDataGroup = new ArrayList<>();
        List<Integer> curKeyGroup = new ArrayList<>();

        for (int i = 0; i < pcData.size(); i++) {
            byte[] data = pcData.get(i);
            if (null == data) {
                data = new byte[0];
            }
            if (
                    curDataGroup.size() > 9                    // 分组小于等于 9 个
                            || (totalLen + data.length > BATCH_SIZE)   // 总大小小于 BATCH_SIZE
            ) {// 扩展包
                dataGroups.add(curDataGroup);
                keyGroups.add(curKeyGroup);

                curDataGroup = new ArrayList<>();
                curKeyGroup = new ArrayList<>();
            }

            curDataGroup.add(data);
            curKeyGroup.add(iKeyIndex[i]);

            totalLen = data.length;
        }

        dataGroups.add(curDataGroup);
        keyGroups.add(curKeyGroup);

        // 并发处理
        List<FutureTask<String[]>> tasks = new ArrayList<>();
        for (int i = 0; i < dataGroups.size(); i++) {

            int finalI = i;
            FutureTask<String[]> task = builder.getLongLinkWorks().fTask(new Callable<String[]>() {
                @Override
                public String[] call() throws Exception {
                    SydApi api = builder.get();
                    return api.SYD_SM4Mac_BatchData(toIntArray(keyGroups.get(finalI)), dataGroups.get(finalI));
                }
            });
            tasks.add(task);
        }

        try {
            List<String> ret = new ArrayList<>();
            for (FutureTask<String[]> t : tasks) {
                String[] macs = t.get();
                ret.addAll(Arrays.asList(macs));
            }
            return ret.toArray(new String[0]);

        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }

    @Override
    public boolean[] SYD_SM4Mac_BatchData(int[] iKeyIndex, List<byte[]> pcData, String[] pcMac) {
        // 将数据进行分组，防止超长。
        List<List<byte[]>> dataGroups = new ArrayList<>();
        List<List<Integer>> keyGroups = new ArrayList<>();
        List<List<String>> macGroups = new ArrayList<>();
        int totalLen = 0;

        List<byte[]> curDataGroup = new ArrayList<>();
        List<Integer> curKeyGroup = new ArrayList<>();
        List<String> curMacGroup = new ArrayList<>();

        for (int i = 0; i < pcData.size(); i++) {
            byte[] data = pcData.get(i);
            if (null == data) {
                data = new byte[0];
            }
            if (
                    curDataGroup.size() > 9                            // 分组小于等于 9 个
                            || (totalLen + data.length > BATCH_SIZE)    // 总大小小于 BATCH_SIZE
            ) {
                // 扩展包
                dataGroups.add(curDataGroup);
                keyGroups.add(curKeyGroup);
                macGroups.add(curMacGroup);

                curDataGroup = new ArrayList<>();
                curKeyGroup = new ArrayList<>();
                curMacGroup = new ArrayList<>();
            }


            curDataGroup.add(data);
            curKeyGroup.add(iKeyIndex[i]);
            curMacGroup.add(pcMac[i]);
            totalLen = data.length;
        }

        dataGroups.add(curDataGroup);
        keyGroups.add(curKeyGroup);
        macGroups.add(curMacGroup);

        // 并发处理
        List<FutureTask<boolean[]>> tasks = new ArrayList<>();
        for (int i = 0; i < dataGroups.size(); i++) {

            int finalI = i;
            FutureTask<boolean[]> task = builder.getLongLinkWorks().fTask(new Callable<boolean[]>() {
                @Override
                public boolean[] call() throws Exception {
                    SydApi api = builder.get();
                    return api.SYD_SM4Mac_BatchData(
                            toIntArray(keyGroups.get(finalI)),
                            dataGroups.get(finalI),
                            macGroups.get(finalI).toArray(new String[0])
                    );
                }
            });
            tasks.add(task);
        }

        try {
            List<Boolean> ret = new ArrayList<>();
            for (FutureTask<boolean[]> t : tasks) {
                boolean[] rets = t.get();
                for (boolean r : rets) {
                    ret.add(r);
                }
            }
            return toBoolArray(ret);

        } catch (Exception e) {
            throw new SydApiException("并行任务执行失败", 0xC101, e);
        }
    }
}