4dkankan-center-manage/src/main/java/com/fdkankan/manage/service/impl/UserIncrementServiceImpl.java

package com.fdkankan.manage.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.manage.common.CacheUtil;
import com.fdkankan.manage.common.ResultCode;
import com.fdkankan.manage.entity.*;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.manage.common.PageInfo;
import com.fdkankan.common.util.DateUtil;
import com.fdkankan.manage.mapper.IUserIncrementMapper;
import com.fdkankan.manage.service.*;
import com.fdkankan.manage.util.Dateutils;
import com.fdkankan.manage.vo.request.UserIncrementParam;
import com.fdkankan.manage.vo.response.GroupByCount;
import com.fdkankan.manage.vo.response.UserIncrementVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
 * 用户增值权益表 服务实现类
 * </p>
 *
 * @author 
 * @since 2022-06-16
 */
@Service
public class UserIncrementServiceImpl extends ServiceImpl<IUserIncrementMapper, UserIncrement> implements IUserIncrementService {

    @Autowired
    private ICameraService cameraService;
    @Autowired
    IIncrementTypeService incrementTypeService;
    @Autowired
    IUserService userService;
    @Autowired
    IAgentNewLogService agentNewLogService;

    @Override
    public Long getValidCountByUserId(Long userId) {
        LambdaQueryWrapper<UserIncrement> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(UserIncrement::getUserId,userId);
        wrapper.gt(UserIncrement::getIncrementEndTime, DateUtil.date2String(new Date(),DateUtil.DEFAULT_DATE_FORMAT));
        return this.count(wrapper);
    }

    @Override
    public HashMap<Long, Long> getValidCountGroupByUserId(List<Long> userIdList) {
        HashMap<Long,Long> map = new HashMap<>();
        List<GroupByCount> result = this.getBaseMapper().getValidCountGroupByUserId(userIdList);
        result.forEach(entity ->map.put(entity.getId(),entity.getCount()));
        return map;
    }

    @Override
    public PageInfo pageList(Long userId, Integer pageNum, Integer pageSize) {
        if(userId == null){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS);
        }
        LambdaQueryWrapper<UserIncrement> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(UserIncrement::getUserId,userId);
        wrapper.and(i ->i.eq(UserIncrement::getIsExpired,0).or().isNotNull(UserIncrement::getCameraId));
        wrapper.orderByDesc(UserIncrement::getCreateTime);
        Page<UserIncrement> page = this.page(new Page<>(pageNum, pageSize), wrapper);

        List<Long> cameraIdList = page.getRecords().parallelStream()
                .filter(camera ->camera.getCameraId()!=null)
                .map(UserIncrement::getCameraId)
                .collect(Collectors.toList());
        HashMap<Long,Camera> cameraMap = new HashMap<>();
        if(cameraIdList.size() >0){
            List<Camera> details = cameraService.getListByCameraIdList(cameraIdList);
            for (Camera detail : details) {
                cameraMap.put(detail.getId(),detail);
            }
        }
        List<UserIncrementVo> voList = new ArrayList<>();
        for (UserIncrement record : page.getRecords()) {
            UserIncrementVo vo = new UserIncrementVo();
            BeanUtils.copyProperties(record,vo);
            if(record.getCameraId() != null){
                Camera camera = cameraMap.get(record.getCameraId());
                if(camera != null){
                    vo.setSnCode(camera.getSnCode());
                }
            }
            if(record.getIncrementTypeId()!= null){
                IncrementType incrementType = incrementTypeService.getById(record.getIncrementTypeId());
                vo.setValidTimeType(incrementType.getValidTimeType());
            }
            voList.add(vo);
        }
        Page<UserIncrementVo> voPage = new Page<>(pageNum, pageSize);
        voPage.setRecords(voList);
        voPage.setTotal(page.getTotal());
        return PageInfo.PageInfo(voPage);
    }

    @Override
    public void delayById(Long id, Integer year) {
        UserIncrement userIncrement = this.getById(id);
        if(userIncrement == null){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS);
        }
        IncrementType incrementType = incrementTypeService.getById(userIncrement.getIncrementTypeId());

        userIncrement.setIsExpired(0);
        Date date = DateUtil.string2Date(userIncrement.getIncrementEndTime(), DateUtil.DEFAULT_DATE_FORMAT);
        Date delay = null;
        switch (incrementType.getValidTimeType()){
            case 0 : delay = DateUtil.delay(date, 1, 1); break;
            case 1 : delay = DateUtil.delay(date, 1, 2); break;
            case 2 : delay = DateUtil.delay(date, 1, 5); break;
            default: throw new BusinessException(ResultCode.INCREMENT_TYPE_ERROR);
        }

        userIncrement.setIncrementEndTime(DateUtil.date2String(delay,DateUtil.DEFAULT_DATE_FORMAT));
        userIncrement.setUpdateTime(null);
        this.updateById(userIncrement);
        agentNewLogService.addByUserIncrement(userIncrement);
    }


    @Override
    public void add(UserIncrementParam param) {
        List<UserIncrement> userIncrementList = new ArrayList<>();
        IncrementType incrementType = incrementTypeService.getById(param.getIncrementTypeId());
        if(incrementType == null){
            throw new BusinessException(ResultCode.INCREMENT_TYPE_EMPTY);
        }
        for (int i = 0 ; i<param.getCount() ;i++) {
            UserIncrement userIncrement = new UserIncrement();
            String date = DateUtil.date2String(new Date(), DateUtil.DEFAULT_DATE_FORMAT);
            userIncrement.setUserId(param.getUserId());
            userIncrement.setKeyWord(UUID.randomUUID().toString().replace("-", ""));
            userIncrement.setIsExpired(0);
            userIncrement.setCreateTime(date);
            userIncrement.setUpdateTime(date);
            userIncrement.setIncrementStartTime(date);
            userIncrement.setIncrementEndTime(param.getIncrementEndTime());
            userIncrement.setIncrementTypeId(param.getIncrementTypeId());
            if(incrementType.getValidTimeType() == 0){
                userIncrement.setMemberLevels("PR");
            }
            if(incrementType.getValidTimeType() == 1){
                userIncrement.setMemberLevels("SE");
            }
            if(param.getMonthQy() != null){
                userIncrement.setMonthQy(param.getMonthQy());
            }
            userIncrementList.add(userIncrement);
        }

        if(userIncrementList.size() >0){
            this.saveBatch(userIncrementList);
            User user = userService.getById(param.getUserId());
            if(user == null){
                throw new BusinessException(ResultCode.USER_NOT_EXIST);
            }
            user.setDownloadNumTotal(user.getDownloadNumTotal() + param.getCount() * incrementType.getDownloadNum());
            userService.updateById(user);
        }
    }

    @Override
    public void unbindCamera(Long cameraId) {
        LambdaUpdateWrapper<UserIncrement> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(UserIncrement::getCameraId,cameraId);
        if("local".equals(CacheUtil.uploadType)){
            this.remove(wrapper);
        }else {
            wrapper.set(UserIncrement::getCameraId,null);
            this.update(wrapper);
        }

    }

    @Override
    public Long getValidCountByCameraId(Long cameraId) {
        LambdaQueryWrapper<UserIncrement> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(UserIncrement::getCameraId,cameraId);
        wrapper.gt(UserIncrement::getIncrementEndTime, DateUtil.date2String(new Date(),DateUtil.DEFAULT_DATE_FORMAT));
        return this.count(wrapper);
    }

    @Override
    public UserIncrement getByCameraId(Long cameraId) {
        LambdaQueryWrapper<UserIncrement> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(UserIncrement::getCameraId,cameraId);
        List<UserIncrement> list = this.list(wrapper);
        if(list !=null && list.size() >0){
            return list.get(0);
        }
        return null;
    }


    @Override
    public void delAgentId(Integer agentId) {
        LambdaUpdateWrapper<UserIncrement> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(UserIncrement::getAgentId,agentId);
        wrapper.set(UserIncrement::getAgentId,null);
        this.update(wrapper);
    }
}