package com.jojubanking.boot.module.system.service.deptxq;

import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.Multimap;
import com.jojubanking.boot.framework.common.enums.CommonStatusEnum;
import com.jojubanking.boot.framework.common.exception.util.ServiceExceptionUtil;
import com.jojubanking.boot.framework.common.util.collection.CollectionUtils;
import com.jojubanking.boot.framework.tenant.core.aop.TenantIgnore;
import com.jojubanking.boot.module.system.controller.admin.dept.vo.dept.DeptListReqVO;
import com.jojubanking.boot.module.system.controller.app.xytjinfo.vo.DeptXqtjListReqVO;
import com.jojubanking.boot.module.system.convert.dept.DeptConvert;
import com.jojubanking.boot.module.system.dal.dataobject.dept.DeptDO;
import com.jojubanking.boot.module.system.enums.dept.DeptIdEnum;
import com.jojubanking.boot.module.system.mq.producer.dept.DeptProducer;
import com.jojubanking.boot.module.system.mq.producer.deptXq.DeptXqProducer;
import com.jojubanking.boot.module.system.service.dept.DeptService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Lazy;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;

import java.util.*;
import com.jojubanking.boot.module.system.controller.admin.deptxq.vo.*;
import com.jojubanking.boot.module.system.dal.dataobject.deptxq.DeptXqDO;
import com.jojubanking.boot.framework.common.pojo.PageResult;

import com.jojubanking.boot.module.system.convert.deptxq.DeptXqConvert;
import com.jojubanking.boot.module.system.dal.mysql.deptxq.DeptXqMapper;

import static com.jojubanking.boot.framework.common.exception.util.ServiceExceptionUtil.exception;
import static com.jojubanking.boot.module.system.enums.ErrorCodeConstants.*;

/**
 * 部门 Service 实现类
 *
 * @author admin
 */
@Service
@Validated
@Slf4j
public class DeptXqServiceImpl implements DeptXqService {

    /**
     * 定时执行 {@link #schedulePeriodicRefresh()} 的周期
     * 因为已经通过 Redis Pub/Sub 机制，所以频率不需要高
     */
    private static final long SCHEDULER_PERIOD = 5 * 60 * 1000L;
    /**
     * 部门缓存
     * key：部门编号 {@link DeptXqDO#getId()}
     *
     * 这里声明 volatile 修饰的原因是，每次刷新时，直接修改指向
     */
    @SuppressWarnings("FieldCanBeLocal")
    private volatile Map<Long, DeptXqDO> deptCache;

    /**
     * 父部门缓存
     * key：部门编号 {@link DeptXqDO#getParentId()}
     * value: 直接子部门列表
     *
     * 这里声明 volatile 修饰的原因是，每次刷新时，直接修改指向
     */
    private volatile Multimap<Long, DeptXqDO> parentDeptCache;
    /**
     * 缓存部门的最大更新时间，用于后续的增量轮询，判断是否有更新
     */
    private volatile Date maxUpdateTime;
    @Resource
    private DeptXqMapper deptXqMapper;
    @Resource
    private DeptXqProducer deptXqProducer;

    @Resource
    @Lazy // 注入自己，所以延迟加载
    private DeptXqService self;

    @Override
    @PostConstruct
    @TenantIgnore // 初始化缓存，无需租户过滤
    public synchronized void initLocalCache() {
        // 获取部门列表，如果有更新
        List<DeptXqDO> deptList = loadDeptIfUpdate(maxUpdateTime);
        if (CollUtil.isEmpty(deptList)) {
            return;
        }

        // 构建缓存
        ImmutableMap.Builder<Long, DeptXqDO> builder = ImmutableMap.builder();
        ImmutableMultimap.Builder<Long, DeptXqDO> parentBuilder = ImmutableMultimap.builder();
        deptList.forEach(sysRoleDO -> {
            builder.put(sysRoleDO.getId(), sysRoleDO);
            parentBuilder.put(sysRoleDO.getParentId(), sysRoleDO);
        });
        // 设置缓存
        deptCache = builder.build();
        parentDeptCache = parentBuilder.build();
        maxUpdateTime = CollectionUtils.getMaxValue(deptList, DeptXqDO::getUpdateTime);
        log.info("[initLocalCache][初始化 Dept 数量为 {}]", deptList.size());
    }

    @Scheduled(fixedDelay = SCHEDULER_PERIOD, initialDelay = SCHEDULER_PERIOD)
    public void schedulePeriodicRefresh() {
        self.initLocalCache();
    }

    /**
     * 如果部门发生变化，从数据库中获取最新的全量部门。
     * 如果未发生变化，则返回空
     *
     * @param maxUpdateTime 当前部门的最大更新时间
     * @return 部门列表
     */
    protected List<DeptXqDO> loadDeptIfUpdate(Date maxUpdateTime) {
        // 第一步，判断是否要更新。
        if (maxUpdateTime == null) { // 如果更新时间为空，说明 DB 一定有新数据
            log.info("[loadMenuIfUpdate][首次加载全量部门]");
        } else { // 判断数据库中是否有更新的部门
            if (deptXqMapper.selectCountByUpdateTimeGt(maxUpdateTime) == 0) {
                return null;
            }
            log.info("[loadMenuIfUpdate][增量加载全量部门]");
        }
        // 第二步，如果有更新，则从数据库加载所有部门
        return deptXqMapper.selectList();
    }


    @Override
    public Long createDeptXq(DeptXqCreateReqVO reqVO) {
//        // 插入
//        DeptXqDO deptXq = DeptXqConvert.INSTANCE.convert(createReqVO);
//        deptXqMapper.insert(deptXq);
//        // 返回
//        return deptXq.getId();

        // 校验正确性
        if (reqVO.getParentId() == null) {
            reqVO.setParentId(DeptIdEnum.ROOT.getId());
        }
        checkCreateOrUpdate(null, reqVO.getParentId(), reqVO.getName());
        // 插入部门
        DeptXqDO dept = DeptXqConvert.INSTANCE.convert(reqVO);
        deptXqMapper.insert(dept);
        // 发送刷新消息
        deptXqProducer.sendDeptRefreshMessage();
        return dept.getId();
    }
    private void checkCreateOrUpdate(Long id, Long parentId, String name) {
        // 校验自己存在
        checkDeptExists(id);
        // 校验父部门的有效性
        checkParentDeptEnable(id, parentId);
        // 校验部门名的唯一性
        checkDeptNameUnique(id, parentId, name);
    }
    private void checkParentDeptEnable(Long id, Long parentId) {
        if (parentId == null || DeptIdEnum.ROOT.getId().equals(parentId)) {
            return;
        }
        // 不能设置自己为父部门
        if (parentId.equals(id)) {
            throw ServiceExceptionUtil.exception(DEPT_PARENT_ERROR);
        }
        // 父岗位不存在
        DeptXqDO dept = deptXqMapper.selectById(parentId);
        if (dept == null) {
            throw ServiceExceptionUtil.exception(DEPT_PARENT_NOT_EXITS);
        }
        // 父部门被禁用
        if (!CommonStatusEnum.ENABLE.getStatus().equals(dept.getStatus())) {
            throw ServiceExceptionUtil.exception(DEPT_NOT_ENABLE);
        }
        // 父部门不能是原来的子部门
        List<DeptXqDO> children = this.getDeptsByParentIdFromCache(id, true);
        if (children.stream().anyMatch(dept1 -> dept1.getId().equals(parentId))) {
            throw ServiceExceptionUtil.exception(DEPT_PARENT_IS_CHILD);
        }
    }
    public List<DeptXqDO> getDeptsByParentIdFromCache(Long parentId, boolean recursive) {
        if (parentId == null) {
            return Collections.emptyList();
        }
        List<DeptXqDO> result = new ArrayList<>(); // TODO TW：待优化，新增缓存，避免每次遍历的计算
        // 递归，简单粗暴
        this.getDeptsByParentIdFromCache(result, parentId,
                recursive ? Integer.MAX_VALUE : 1, // 如果递归获取，则无限；否则，只递归 1 次
                parentDeptCache);
        return result;
    }
    private void getDeptsByParentIdFromCache(List<DeptXqDO> result, Long parentId, int recursiveCount,
                                             Multimap<Long, DeptXqDO> parentDeptMap) {
        // 递归次数为 0，结束！
        if (recursiveCount == 0) {
            return;
        }
        // 获得子部门
        Collection<DeptXqDO> depts = parentDeptMap.get(parentId);
        if (CollUtil.isEmpty(depts)) {
            return;
        }
        result.addAll(depts);
        // 继续递归
        depts.forEach(dept -> getDeptsByParentIdFromCache(result, dept.getId(),
                recursiveCount - 1, parentDeptMap));
    }

    private void checkDeptExists(Long id) {
        if (id == null) {
            return;
        }
        DeptXqDO dept = deptXqMapper.selectById(id);
        if (dept == null) {
            throw ServiceExceptionUtil.exception(DEPT_NOT_FOUND);
        }
    }

    private void checkDeptNameUnique(Long id, Long parentId, String name) {
        DeptXqDO menu = deptXqMapper.selectByParentIdAndName(parentId, name);
        if (menu == null) {
            return;
        }
        // 如果 id 为空，说明不用比较是否为相同 id 的岗位
        if (id == null) {
            throw ServiceExceptionUtil.exception(DEPT_NAME_DUPLICATE);
        }
        if (!menu.getId().equals(id)) {
            throw ServiceExceptionUtil.exception(DEPT_NAME_DUPLICATE);
        }
    }
    @Override
    public void updateDeptXq(DeptXqUpdateReqVO reqVO) {
//        // 校验存在
//        this.validateDeptXqExists(updateReqVO.getId());
//        // 更新
//        DeptXqDO updateObj = DeptXqConvert.INSTANCE.convert(updateReqVO);
//        deptXqMapper.updateById(updateObj);

        // 校验正确性
        if (reqVO.getParentId() == null) {
            reqVO.setParentId(DeptIdEnum.ROOT.getId());
        }
        checkCreateOrUpdate(reqVO.getId(), reqVO.getParentId(), reqVO.getName());
        // 更新部门
        DeptXqDO updateObj = DeptXqConvert.INSTANCE.convert(reqVO);
        deptXqMapper.updateById(updateObj);
        // 发送刷新消息
        deptXqProducer.sendDeptRefreshMessage();
    }

    @Override
    public void deleteDeptXq(Long id) {
//        // 校验存在
//        this.validateDeptXqExists(id);
//        // 删除
//        deptXqMapper.deleteById(id);

        // 校验是否存在
        checkDeptExists(id);
        // 校验是否有子部门
        if (deptXqMapper.selectCountByParentId(id) > 0) {
            throw ServiceExceptionUtil.exception(DEPT_EXITS_CHILDREN);
        }
        // 删除部门
        deptXqMapper.deleteById(id);
        // 发送刷新消息
        deptXqProducer.sendDeptRefreshMessage();
    }

    private void validateDeptXqExists(Long id) {
        if (deptXqMapper.selectById(id) == null) {
            throw exception(DEPT_XQ_NOT_EXISTS);
        }
    }

    @Override
    public DeptXqDO getDeptXq(Long id) {
        return deptXqMapper.selectById(id);
    }

    @Override
    public List<DeptXqDO> getDeptXqList(Collection<Long> ids) {
        return deptXqMapper.selectBatchIds(ids);
    }

    @Override
    public PageResult<DeptXqDO> getDeptXqPage(DeptXqPageReqVO pageReqVO) {
        return deptXqMapper.selectPage(pageReqVO);
    }

    @Override
    public List<DeptXqDO> getDeptXqList(DeptXqExportReqVO exportReqVO) {
        return deptXqMapper.selectList(exportReqVO);
    }

    @Override
    public List<DeptXqDO> getSimpleXqDepts(Collection<Long> ids) {
        return deptXqMapper.selectBatchIds(ids);
    }


    public List<DeptXqDO> getSimpleXqDepts(DeptXqListReqVO reqVO) {
        return deptXqMapper.selectXqList(reqVO);
    }
    public List<DeptXqDO> getSimpleXqtjDepts(DeptXqtjListReqVO reqVO) {
        return deptXqMapper.selectXqtjList(reqVO);
    }

}
