package com.wgcloud.service;

import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.wgcloud.dto.HeaderDto;
import com.wgcloud.entity.HeathMonitor;
import com.wgcloud.entity.HeathState;
import com.wgcloud.mapper.HeathMonitorMapper;
import com.wgcloud.util.DateUtil;
import com.wgcloud.util.HostUtil;
import com.wgcloud.util.RestUtil;
import com.wgcloud.util.UUIDUtil;
import com.wgcloud.util.msg.WarnMailUtil;
import com.wgcloud.util.msg.WarnPools;
import com.wgcloud.util.staticvar.BatchData;
import com.wgcloud.util.staticvar.StaticKeys;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.servlet.http.HttpServletRequest;
import java.util.*;

/**
 * @version v3.3
 * @ClassName:HeathMonitorService.java
 * @author: http://www.wgstart.com
 * @date: 2021年1月16日
 * @Description: HeathMonitorService.java
 * @Copyright: 2019-2021 wgcloud. All rights reserved.
 */
@Service
public class HeathMonitorService {

    private static final Logger logger = LoggerFactory.getLogger(HeathMonitorService.class);


    public PageInfo selectByParams(Map<String, Object> params, int currPage, int pageSize) throws Exception {
        PageHelper.startPage(currPage, pageSize);
        List<HeathMonitor> list = heathMonitorMapper.selectByParams(params);
        PageInfo<HeathMonitor> pageInfo = new PageInfo<HeathMonitor>(list);
        return pageInfo;
    }

    public void save(HeathMonitor heathMonitor) throws Exception {
        heathMonitor.setId(UUIDUtil.getUUID());
        heathMonitor.setCreateTime(DateUtil.getNowTime());
        if (StringUtils.isEmpty(heathMonitor.getResNoKeyword())) {
            heathMonitor.setResNoKeyword("");
        }
        if (StringUtils.isEmpty(heathMonitor.getResKeyword())) {
            heathMonitor.setResKeyword("");
        }
        if (StringUtils.isEmpty(heathMonitor.getHeathUrl())) {
            heathMonitor.setHeathUrl(heathMonitor.getHeathUrl().trim());
        }

        //如果是post方式，处理header键值对转为json
        setHeaderJson(heathMonitor);

        heathMonitorMapper.save(heathMonitor);
    }


    @Transactional
    public void saveRecord(List<HeathMonitor> recordList) throws Exception {
        if (recordList.size() < 1) {
            return;
        }
        for (HeathMonitor as : recordList) {
            as.setId(UUIDUtil.getUUID());
        }
        heathMonitorMapper.insertList(recordList);
    }

    public int countByParams(Map<String, Object> params) throws Exception {
        return heathMonitorMapper.countByParams(params);
    }

    @Transactional
    public int deleteById(String[] id) throws Exception {
        return heathMonitorMapper.deleteById(id);
    }

    public void updateById(HeathMonitor heathMonitor)
            throws Exception {
        if (StringUtils.isEmpty(heathMonitor.getHeathUrl())) {
            heathMonitor.setHeathUrl(heathMonitor.getHeathUrl().trim());
        }

        //如果是post方式，处理header键值对转为json
        setHeaderJson(heathMonitor);

        heathMonitorMapper.updateById(heathMonitor);
    }

    public void updateForServerBackup(HeathMonitor heathMonitor)
            throws Exception {
        heathMonitorMapper.updateById(heathMonitor);
    }

    public HeathMonitor selectById(String id) throws Exception {
        return heathMonitorMapper.selectById(id);
    }

    @Transactional
    public void updateRecord(List<HeathMonitor> recordList) throws Exception {
        heathMonitorMapper.updateList(recordList);
    }

    public List<HeathMonitor> selectAllByParams(Map<String, Object> params) throws Exception {
        return heathMonitorMapper.selectAllByParams(params);
    }

    /**
     * 如果是post方式，处理header键值对转为json
     *
     * @param heathMonitor
     */
    private void setHeaderJson(HeathMonitor heathMonitor) {
        if (StaticKeys.METHOD_POST.equals(heathMonitor.getMethod())) {
            List<HeaderDto> dtoList = new ArrayList<>();
            if (!StringUtils.isEmpty(heathMonitor.getHeaderKey())) {
                HeaderDto dto1 = new HeaderDto();
                dto1.setSort("1");
                dto1.setKey(heathMonitor.getHeaderKey());
                dto1.setValue(heathMonitor.getHeaderValue());
                dtoList.add(dto1);

            }
            if (!StringUtils.isEmpty(heathMonitor.getHeaderKey2())) {
                HeaderDto dto2 = new HeaderDto();
                dto2.setSort("2");
                dto2.setKey(heathMonitor.getHeaderKey2());
                dto2.setValue(heathMonitor.getHeaderValue2());
                dtoList.add(dto2);

            }
            if (!StringUtils.isEmpty(heathMonitor.getHeaderKey3())) {
                HeaderDto dto3 = new HeaderDto();
                dto3.setSort("3");
                dto3.setKey(heathMonitor.getHeaderKey3());
                dto3.setValue(heathMonitor.getHeaderValue3());
                dtoList.add(dto3);

            }
            if (!StringUtils.isEmpty(heathMonitor.getHeaderKey4())) {
                HeaderDto dto4 = new HeaderDto();
                dto4.setSort("4");
                dto4.setKey(heathMonitor.getHeaderKey4());
                dto4.setValue(heathMonitor.getHeaderValue4());
                dtoList.add(dto4);

            }
            if (!StringUtils.isEmpty(heathMonitor.getHeaderKey5())) {
                HeaderDto dto5 = new HeaderDto();
                dto5.setSort("5");
                dto5.setKey(heathMonitor.getHeaderKey5());
                dto5.setValue(heathMonitor.getHeaderValue5());
                dtoList.add(dto5);

            }
            heathMonitor.setHeaderJson(JSONUtil.toJsonStr(dtoList));
        }
    }

    /**
     * 如果是post方式，处理header键值对页面回显
     *
     * @param heathMonitor
     */
    public HashMap<String, String> displayHeaderJson(HeathMonitor heathMonitor) {
        HashMap<String, String> headerHashMap = new HashMap<>();
        try {
            if (StaticKeys.METHOD_POST.equals(heathMonitor.getMethod())) {
                if (!StringUtils.isEmpty(heathMonitor.getHeaderJson())) {
                    if (JSONUtil.isJsonObj(heathMonitor.getHeaderJson())) {
                        //此处是为了兼容v3.4.0之前版本，因为之前存贮的是json对象
                        HeaderDto dto = JSONUtil.toBean(heathMonitor.getHeaderJson(), HeaderDto.class);
                        heathMonitor.setHeaderKey(dto.getKey());
                        heathMonitor.setHeaderValue(dto.getValue());
                        headerHashMap.put(dto.getKey(), dto.getValue());
                    } else {
                        JSONArray jsonArray = JSONUtil.parseArray(heathMonitor.getHeaderJson());
                        for (Object jsonObject : jsonArray) {
                            HeaderDto dto = JSONUtil.toBean((JSONObject) jsonObject, HeaderDto.class);
                            headerHashMap.put(dto.getKey(), dto.getValue());
                            if (dto.getSort().equals("1")) {
                                heathMonitor.setHeaderKey(dto.getKey());
                                heathMonitor.setHeaderValue(dto.getValue());
                            }
                            if (dto.getSort().equals("2")) {
                                heathMonitor.setHeaderKey2(dto.getKey());
                                heathMonitor.setHeaderValue2(dto.getValue());
                            }
                            if (dto.getSort().equals("3")) {
                                heathMonitor.setHeaderKey3(dto.getKey());
                                heathMonitor.setHeaderValue3(dto.getValue());
                            }
                            if (dto.getSort().equals("4")) {
                                heathMonitor.setHeaderKey4(dto.getKey());
                                heathMonitor.setHeaderValue4(dto.getValue());
                            }
                            if (dto.getSort().equals("5")) {
                                heathMonitor.setHeaderKey5(dto.getKey());
                                heathMonitor.setHeaderValue5(dto.getValue());
                            }
                        }
                    }
                }
            }
        } catch (Exception e) {
            logger.error("displayHeaderJson错误", e);
        }
        return headerHashMap;
    }

    /**
     * 保存操作日志
     *
     * @param request 获取当前登录用户
     * @param action  操作标识
     */
    public void saveLog(HttpServletRequest request, String action, HeathMonitor heathMonitor) {
        if (null == heathMonitor) {
            return;
        }
        logInfoService.save(HostUtil.getAccountByRequest(request).getAccount() + action + "服务接口监测信息：" + heathMonitor.getAppName(),
                "服务接口URL：" + heathMonitor.getHeathUrl(), StaticKeys.LOG_XTCZ);
    }

    /**
     * 检测服务接口任务具体处理线程
     *
     * @param h
     * @param date
     */
    public void taskThreadHandler(HeathMonitor h, Date date) {
        int status = 500;
        Map<String, Integer> statusMap = new HashMap<>();
        if (StaticKeys.METHOD_POST.equals(h.getMethod())) {
            HashMap<String, String> headerMap = displayHeaderJson(h);
            statusMap = restUtil.post(h.getHeathUrl(), h.getPostStr(), h.getResKeyword(), h.getResNoKeyword(), headerMap);
        } else {
            statusMap = restUtil.get(h.getHeathUrl(), h.getResKeyword(), h.getResNoKeyword());
        }
        status = statusMap.get("heathStatus");
        h.setCreateTime(date);
        //接口下线后，不再更新时间 begin
        if (status != 200) {
            h.setCreateTime(null);
        }
        //接口下线后，不再更新时间 end
        h.setHeathStatus(status + "");
        h.setResTimes(statusMap.get("resTimes"));
        try {
            updateById(h);
        } catch (Exception e) {
            e.printStackTrace();
        }
        HeathState heathState = new HeathState();
        heathState.setResTimes(h.getResTimes());
        heathState.setHeathId(h.getId());
        heathState.setCreateTime(date);
        BatchData.HEATH_STATE_LIST.add(heathState);
        if (!StaticKeys.HEATH_STATUS_200.equals(h.getHeathStatus())) {
            //返回错误处理
            WarnMailUtil.sendHeathInfo(h, true);
        } else {
            //返回成功处理
            if (null != WarnPools.MEM_WARN_MAP.get(h.getId())) {
                WarnMailUtil.sendHeathInfo(h, false);
            }
        }
    }

    @Autowired
    private HeathMonitorMapper heathMonitorMapper;
    @Autowired
    private LogInfoService logInfoService;
    @Autowired
    private RestUtil restUtil;


}