package cn.fw.morax.server.task;

import cn.fw.common.cache.locker.DistributedLocker;
import cn.fw.morax.common.constant.TimeTaskConstant;
import cn.fw.morax.common.utils.PublicUtil;
import cn.fw.morax.domain.db.eval.EvalGroup;
import cn.fw.morax.domain.db.eval.EvalGroupUser;
import cn.fw.morax.service.biz.eval.EvalGroupBizService;
import cn.fw.morax.service.biz.eval.EvalGroupUserBizService;
import cn.fw.morax.service.data.eval.EvalGroupUserService;
import cn.hutool.core.date.StopWatch;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.google.common.collect.Lists;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RLock;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDate;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.stream.Collectors;

/**
 * @author : jiangchao
 * @className : KpiGroupStatusTask
 * @description : 考评组人员定时器
 * @date : 2022-04-07 15:29
 */
@Component
@Slf4j
@RequiredArgsConstructor
@ConditionalOnProperty(prefix = "task", name = "switch", havingValue = "on")
public class EvalGroupUserTask {

    private final EvalGroupUserBizService evalGroupUserBizService;
    private final EvalGroupUserService evalGroupUserService;
    private final DistributedLocker distributedLocker;

    @Value("${spring.cache.custom.global-prefix}:eval:group:user")
    @Getter
    private String evalGroupUserDistKey;

    /**
     * 每天凌晨30分执行（人事角色、岗位调整在5点执行）
     * 1. 将待生效数据改为生效中
     * 2. 将之前的配置设置为失效
     * 3. 处理重复数据，合并、拆分门店的考评组配置需要把其他的重复配置失效
     */
    @Scheduled(cron = TimeTaskConstant.EVAL_GROUP_USER)
    @Transactional(rollbackFor = Exception.class)
    public void processEvalUser() {
        Lock lock = distributedLocker.lock(getEvalGroupUserDistKey());
        if (((RLock) lock).isLocked()) {
            try {
                log.info("定时任务【考评组人员更新】开始执行");
                StopWatch stopWatch = new StopWatch();
                stopWatch.start("考评组人员更新,查询人事系统人员状态");
                //查询员工
                LocalDate yesterday = LocalDate.now().minusDays(1);
                List<EvalGroupUser> evalGroupUsersForSave = this.queryKpiStaff(yesterday);
                stopWatch.stop();
                if (PublicUtil.isEmpty(evalGroupUsersForSave)) {
                    log.info(stopWatch.prettyPrint());
                    return;
                }
                stopWatch.start("考评组人员保存，保存条数："+ evalGroupUsersForSave.size());
                //逻辑删除今天已保存人员，保存人员
                Set<Long> evalGroupIds = evalGroupUsersForSave.stream().map(EvalGroupUser::getEvalGroupId).collect(Collectors.toSet());
                evalGroupUserService.update(Wrappers.<EvalGroupUser>lambdaUpdate()
                        .in(EvalGroupUser::getEvalGroupId, evalGroupIds)
                        .eq(EvalGroupUser::getDataDate, yesterday)
                        .eq(EvalGroupUser::getYn, Boolean.TRUE)
                        .set(EvalGroupUser::getYn, Boolean.FALSE)
                );
                evalGroupUserService.saveBatch(evalGroupUsersForSave);
                stopWatch.stop();
                log.info(stopWatch.prettyPrint(TimeUnit.SECONDS));
            } catch (Exception e){
                log.error(e.getMessage(), e);
            } finally {
                lock.unlock();
            }
        }
    }

    /**
     * 查询考评员工
     * @return
     */
    private List<EvalGroupUser> queryKpiStaff(LocalDate yesterday) {
        List<EvalGroup> evalGroups = evalGroupUserBizService.postEffectEvalGroups(yesterday);
        if (! evalGroups.isEmpty()) {
            log.info("需要生成考评组人员的考评组：{}", evalGroups);
        }
        List<EvalGroupUser> evalGroupUsersForSave = Lists.newArrayListWithCapacity(300);
        List<EvalGroupUser> evalGroupUsers = null;
        for (EvalGroup evalGroup : evalGroups) {
            for (Long postId : evalGroup.getPostIds()) {
                evalGroupUsers = evalGroupUserBizService.fetchBuildKpiUser(postId, evalGroup, yesterday);
                evalGroupUsersForSave.addAll(evalGroupUsers);
            }
        }
        return evalGroupUsersForSave;
    }

}