package com.wgcloud.service;

import cn.hutool.core.bean.BeanUtil;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.wgcloud.entity.DceInfo;
import com.wgcloud.entity.DceState;
import com.wgcloud.entity.HostGroup;
import com.wgcloud.mapper.DceInfoMapper;
import com.wgcloud.util.DateUtil;
import com.wgcloud.util.HostUtil;
import com.wgcloud.util.PingUtil;
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.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

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

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

    public void save(DceInfo DceInfo, HttpServletRequest request) throws Exception {
        DceInfo.setId(UUIDUtil.getUUID());
        DceInfo.setResTimes(1);
        Date nowDate = DateUtil.getNowTime();
        DceInfo.setCreateTime(nowDate);
        if (StringUtils.isEmpty(DceInfo.getHostname())) {
            DceInfo.setHostname(DceInfo.getHostname().trim());
        }
        dceInfoMapper.save(DceInfo);
        //保存页面动态添加的端口数据
        addExtDataForm(DceInfo, request, nowDate);
    }

    /**
     * 保存页面动态添加的PING设备数据
     *
     * @param dceInfo
     * @param request
     */
    private void addExtDataForm(DceInfo dceInfo, HttpServletRequest request, Date nowDate) throws Exception {
        String dataFromIndex = request.getParameter("dataFromIndex");
        int rowsLen = 0;
        if (!StringUtils.isEmpty(dataFromIndex)) {
            for (int i = 0; i <= Integer.valueOf(dataFromIndex); i++) {


                String hostname = request.getParameter("hostname_" + i);
                String remark = request.getParameter("remark_" + i);
                if (StringUtils.isEmpty(hostname)) {
                    continue;
                }
                DceInfo dceInfoExt = new DceInfo();
                BeanUtil.copyProperties(dceInfo, dceInfoExt, true);
                dceInfoExt.setId(UUIDUtil.getUUID());
                dceInfoExt.setCreateTime(nowDate);
                dceInfoExt.setHostname(hostname);
                dceInfoExt.setRemark(remark);
                dceInfoMapper.save(dceInfoExt);

                saveLog(request, StaticKeys.LOG_ADD, dceInfoExt);
                rowsLen += 1;
            }
        }
    }


    @Transactional
    public void saveRecord(List<DceInfo> recordList) throws Exception {
        if (recordList.size() < 1) {
            return;
        }
        Date now = DateUtil.getNowTime();
        for (DceInfo as : recordList) {
            as.setResTimes(1);
            as.setId(UUIDUtil.getUUID());
            as.setCreateTime(now);
        }
        dceInfoMapper.insertList(recordList);
    }

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

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

    public void updateById(DceInfo dceInfo)
            throws Exception {
        if (StringUtils.isEmpty(dceInfo.getHostname())) {
            dceInfo.setHostname(dceInfo.getHostname().trim());
        }
        dceInfoMapper.updateById(dceInfo);
    }

    public DceInfo selectById(String id) throws Exception {
        return dceInfoMapper.selectById(id);
    }

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

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

    /**
     * 设置资源分组 begin
     *
     * @param recordList
     */
    public List<HostGroup> setGroupInList(List<DceInfo> recordList) throws Exception {
        Map<String, Object> params = new HashMap<String, Object>();
        params.put("groupType", StaticKeys.HOST_GROUP_2);
        List<HostGroup> hostGroupList = hostGroupService.selectAllByParams(params);
        //查询所有主机分组列表，封装为map
        Map<String, String> hostGroupMap = new HashMap<>();
        for (HostGroup hostGroup : hostGroupList) {
            hostGroupMap.put(hostGroup.getId(), hostGroup.getGroupName());
        }
        for (DceInfo dceInfo1 : recordList) {
            if (!StringUtils.isEmpty(dceInfo1.getGroupId())) {
                dceInfo1.setGroupId(hostGroupMap.get(dceInfo1.getGroupId()));
            }
        }
        return hostGroupList;
    }

    /**
     * 保存操作日志
     *
     * @param request 获取当前登录用户
     * @param action  操作标识
     */
    public void saveLog(HttpServletRequest request, String action, DceInfo dceInfo) {
        if (null == dceInfo) {
            return;
        }
        logInfoService.save(HostUtil.getAccountByRequest(request).getAccount() + action + "数通设备PING监测信息：" + dceInfo.getHostname(),
                "数通设备：" + dceInfo.getRemark(), StaticKeys.LOG_XTCZ);
    }

    /**
     * 检测数通设备任务具体处理线程
     *
     * @param h
     * @param date
     */
    public void taskThreadHandler(DceInfo h, Date date) {
        long resTimes = PingUtil.ping(h.getHostname(), 1, 3);
        h.setCreateTime(date);
        //设备下线后，不再更新时间 begin
        if (resTimes < 0 && h.getResTimes() < 0) {
            h.setCreateTime(null);
        }
        //设备下线后，不再更新时间 end
        h.setResTimes(Integer.valueOf(resTimes + ""));
        try {
            updateById(h);
        } catch (Exception e) {
            e.printStackTrace();
        }
        if (resTimes > 0) {
            DceState dceState = new DceState();
            dceState.setResTimes(h.getResTimes());
            dceState.setDceId(h.getId());
            dceState.setCreateTime(date);
            BatchData.DCE_STATE_LIST.add(dceState);
        }
        if (h.getResTimes() < 0) {
            //返回错误处理
            WarnMailUtil.sendDceInfo(h, true);
        } else {
            //返回成功处理
            if (null != WarnPools.MEM_WARN_MAP.get(h.getId())) {
                WarnMailUtil.sendDceInfo(h, false);
            }
        }
    }

    @Autowired
    private DceInfoMapper dceInfoMapper;
    @Resource
    private HostGroupService hostGroupService;
    @Autowired
    private LogInfoService logInfoService;


}