package jnpf.scheduletask.task;

import com.xxl.job.core.context.XxlJobHelper;
import com.xxl.job.core.handler.annotation.XxlJob;
import jnpf.base.DataInterFaceApi;
import jnpf.base.UserInfo;
import jnpf.base.entity.DataInterfaceEntity;
import jnpf.base.model.datainterface.DataInterfaceInvokeModel;
import jnpf.database.util.TenantDataSourceUtil;
import jnpf.emnus.TemplateEnum;
import jnpf.exception.TenantInvalidException;
import jnpf.model.visualJson.TemplateJsonModel;
import jnpf.scheduletask.entity.TimeTaskEntity;
import jnpf.scheduletask.model.ContentNewModel;
import jnpf.scheduletask.model.UpdateTaskModel;
import jnpf.scheduletask.rest.RestScheduleTaskUtil;
import jnpf.util.AuthUtil;
import jnpf.util.JsonUtil;
import jnpf.util.StringUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * Api和数据接口使用
 *
 * @author ：JNPF开发平台组
 * @version: V3.1.0
 * @copyright 引迈信息技术有限公司
 * @date ：2022/3/23 9:25
 */
@Slf4j
@Component
public class ScheduleTaskHandler {

    @Autowired
    private DataInterFaceApi dataInterFaceApi;


    @XxlJob("defaultHandler")
    public void defaultHandler() {
        long time = System.currentTimeMillis();
        // 获取参数
        String param = XxlJobHelper.getJobParam();
        // 转换成模型
        TimeTaskEntity entity = JsonUtil.getJsonToBean(param, TimeTaskEntity.class);
        ContentNewModel model = JsonUtil.getJsonToBean(param, ContentNewModel.class);
        // 得到数据接口信息
        String tenantId = StringUtil.isNotEmpty(model.getUserInfo().getTenantId()) ? model.getUserInfo().getTenantId() : "";
        String userId = StringUtil.isNotEmpty(model.getUserInfo().getUserId()) ? model.getUserInfo().getUserId() : "";
        String token = AuthUtil.loginTempUser(userId, tenantId, true);
        // 切换租户
        UserInfo userInfo = model.getUserInfo();
        // 切换数据源
        if (userInfo != null && StringUtil.isNotEmpty(userInfo.getTenantId())) {
            try {
                TenantDataSourceUtil.switchTenant(userInfo.getTenantId());
            }catch (TenantInvalidException e){
                // 租户无效 禁用任务
                log.error("ScheduleTaskHandler, 租户无效, 禁用任务：{}", userInfo.getTenantId());
                entity = RestScheduleTaskUtil.getInfo(entity.getId(), model.getUserInfo());
                entity.setEnabledMark(0);
                UpdateTaskModel updateTaskModel = new UpdateTaskModel();
                updateTaskModel.setEntity(entity);
                updateTaskModel.setUserInfo(model.getUserInfo());
                RestScheduleTaskUtil.updateTask(updateTaskModel);
            }
        }
        DataInterfaceEntity dataInterfaceEntity = dataInterFaceApi.getDataInterfaceInfo(model.getInterfaceId(), tenantId);
        // 如果是http
        if (entity != null && "3".equals(String.valueOf(dataInterfaceEntity.getType()))) {
            dataInterfaceEntity.setTenantId(tenantId);
            boolean callHttp = callHttp(model, dataInterfaceEntity, token);
        } else if (entity != null && "1".equals(String.valueOf(dataInterfaceEntity.getType()))) {
            dataInterfaceEntity.setTenantId(tenantId);
            boolean callSql = callSql(model, dataInterfaceEntity, token);
        }
    }

    // ---------------START callSQL

    /**
     * 调用SQL
     *
     * @param model 系统调度参数
     */
    private boolean callSql(ContentNewModel model, DataInterfaceEntity entity, String token) {
        try {
            if (entity != null) {
                Map<String, String> map = null;
                if (model.getParameter() != null && model.getParameter().size() > 0) {
                    map = new HashMap<>(16);
                    for (TemplateJsonModel parameterModel : model.getParameter()) {
                        String value = Objects.equals(parameterModel.getSourceType(), TemplateEnum.Empty.getCode()) ? "" : parameterModel.getRelationField();
                        map.put(parameterModel.getField(), value);
                    }
                }
                dataInterFaceApi.invokeById(new DataInterfaceInvokeModel(entity.getId(), entity.getTenantId(), map, token));
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

    // ---------------START callHttp

    /**
     * HTTP调用
     *
     * @param model 系统调度参数
     * @return
     */
    private Boolean callHttp(ContentNewModel model, DataInterfaceEntity entity, String token) {
        try {
            if (entity != null) {
                Map<String, String> map = null;
                if (model.getParameter() != null && model.getParameter().size() > 0) {
                    map = new HashMap<>(16);
                    for (TemplateJsonModel parameterModel : model.getParameter()) {
                        String value = Objects.equals(parameterModel.getSourceType(), TemplateEnum.Empty.getCode()) ? "" : parameterModel.getRelationField();
                        map.put(parameterModel.getField(), value);
                    }
                }
                dataInterFaceApi.invokeById(new DataInterfaceInvokeModel(entity.getId(), entity.getTenantId(), map, token));
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return false;
    }

}