package com.nuvole.util;

import com.nuvole.common.domain.emnu.CommonResultEmnu;
import com.nuvole.common.domain.result.CommonResult;
import lombok.AllArgsConstructor;
import lombok.Data;

import java.util.LinkedList;
import java.util.List;
import java.util.function.Supplier;

/**
 * 功能描述：CommonResult工具
 * @author xuekang
 * @date 2024-04-16 10:59
 * @version 1.0
 * @param <E> data类型
 */
public class ResultHandler<E> {
    /** CommonResult */
    private final CommonResult<E> result;
    /** 设置空处理函数 */
    private Supplier<E> nullProcess;
    /** 设置data空处理函数 */
    private Supplier<E> dataNullProcess;
    /** 错误码处理函数 */
    private List<ErrorProcess<E>> errorProcessList;

    public ResultHandler(CommonResult<E> result) {
        this.result = result;
    }

    public ResultHandler(CommonResult<E> result, ResultHandler<E> resultHandler) {
        this.result = result;
        this.nullProcess = resultHandler.nullProcess;
        this.dataNullProcess = resultHandler.dataNullProcess;
        this.errorProcessList = resultHandler.errorProcessList;
    }

    /**
     * 设置空处理函数
     * @param nullProcess 空处理函数
     * @return ResultHandler
     */
    public ResultHandler<E> setNullProcess(Supplier<E> nullProcess) {
        this.nullProcess = nullProcess;
        return this;
    }

    /**
     * 设置data空处理函数
     * @param dataNullProcess data 空处理函数
     * @return ResultHandler
     */
    public ResultHandler<E> setDataNullProcess(Supplier<E> dataNullProcess) {
        this.dataNullProcess = dataNullProcess;
        return this;
    }

    /**
     * 错误码处理函数
     * @param code 错误码
     * @param errorProcessList 函数
     * @return ResultHandler
     */
    public ResultHandler<E> appendError(Integer code, Supplier<E> errorProcessList) {
        if (this.errorProcessList == null) {
            this.errorProcessList = new LinkedList<>();
        }
        this.errorProcessList.add(new ErrorProcess<>(code, errorProcessList));
        return this;
    }

    /**
     * CommonResult工具
     * @param result CommonResult
     * @return ResultHandler
     * @param <T> data类型
     */
    public static <T>ResultHandler<T> getInstance(CommonResult<T> result) {
        return new ResultHandler<>(result);
    }

    /**
     * 获取
     * @param result CommonResult
     * @param resultHandler ResultHandler
     * @return ResultHandler
     * @param <T> date类型
     */
    public static <T>ResultHandler<T> getInstance(CommonResult<T> result, ResultHandler<T> resultHandler) {
        return new ResultHandler<>(result, resultHandler);
    }

    /**
     * 获取data
     * @return data
     */
    public E getDate() {
        if (result == null) {
            if (nullProcess != null) {
                return nullProcess.get();
            }
            return null;
        }
        Integer code = result.getCode();
        if (code.equals(CommonResultEmnu.OK.getCode())) {
            E data = result.getData();
            if (data == null && dataNullProcess != null) {
                return this.dataNullProcess.get();
            }
            return data;
        }
        if (errorProcessList == null) {
            return null;
        }
        for (ErrorProcess<E> process : this.errorProcessList) {
            if (code.equals(process.getCode())) {
                return process.getSupplier().get();
            }
        }
        return null;
    }

    public static void main(String[] args) {
//        Maps.EntryTransformer<BaseServiceClient, String, CommonResult<String>> getAllManagerByOrgCode = BaseServiceClient::getAllManagerByOrgCode;
//        ClientUtil.getDate(BaseServiceClient.class, BaseServiceClient::getAllManagerByOrgCode, null);
        ResultHandler.getInstance(new CommonResult<>())
                .setNullProcess(() -> {
                    throw new RuntimeException("");
                })
                .setDataNullProcess(() -> "")
                .appendError(1, () -> "")
                .getDate();
    }

    /**
     * 功能描述：错误处理逻辑
     * @author xuekang
     * @date 2024-04-16 13:26
     * @version 1.0
     */
    @Data
    @AllArgsConstructor
    private static class ErrorProcess<E> {
        /** 错误码 */
        private Integer code;
        /** 处理函数 */
        private Supplier<E> supplier;
    }
}