4dkankan-after-sale/src/main/java/com/fdkankan/sale/service/impl/SysUserServiceImpl.java

package com.fdkankan.sale.service.impl;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fdkankan.common.util.Base64Converter;
import com.fdkankan.common.util.SecurityUtil;
import com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.sale.common.PageInfo;
import com.fdkankan.sale.common.ResultCode;
import com.fdkankan.sale.entity.SysRole;
import com.fdkankan.sale.entity.SysUser;
import com.fdkankan.sale.entity.User;
import com.fdkankan.sale.exception.BusinessException;
import com.fdkankan.sale.mapper.ISysUserMapper;
import com.fdkankan.sale.service.ISysMenuService;
import com.fdkankan.sale.service.ISysRoleService;
import com.fdkankan.sale.service.ISysUserService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.sale.vo.request.SysUserParam;
import com.fdkankan.sale.vo.response.ManageLoginResponse;
import com.fdkankan.sale.vo.response.SysRoleVo;
import com.fdkankan.sale.vo.response.SysUserVo;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.List;
import java.util.Set;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 
 * @since 2022-12-09
 */
@Service
public class SysUserServiceImpl extends ServiceImpl<ISysUserMapper, SysUser> implements ISysUserService {

    @Autowired
    private ISysRoleService roleService;
    @Autowired
    private ISysMenuService sysMenuService;
    @Autowired
    private RedisUtil redisUtil;


    @Override
    public SysUser getByUserName(String userName) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysUser::getUserName,userName);
        List<SysUser> list = this.list(wrapper);
        if(list == null || list.size() <=0){
            return null;
        }
        return list.get(0);
    }
    @DS("db2")
    public SysUser getByUserNameDb2(String userName) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysUser::getUserName,userName);
        List<SysUser> list = this.list(wrapper);
        if(list == null || list.size() <=0){
            return null;
        }
        return list.get(0);
    }


    @Override
    public ManageLoginResponse getUserByUserNameAndPassword(String userName, String password) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysUser::getUserName,userName)
                .eq(SysUser::getPassword,password);
        List<SysUser> list = this.list(wrapper);
        if(list == null || list.size() <=0){
            return null;
        }
        SysUser sysUser = list.get(0);
        SysRole role = roleService.getById(sysUser.getRoleId());
        ManageLoginResponse loginVO = new ManageLoginResponse();
        BeanUtils.copyProperties(sysUser,loginVO);
        loginVO.setRoleId(role.getId());
        loginVO.setRoleName(role.getRoleName());

        sysMenuService.setRoleAndMenuCache(role,sysUser);

        return loginVO;
    }

    @Override
    public void addOrUpdate(SysUserParam param) {
        SysUser user = this.getByUserNameDb2(param.getUserName());
        if(user ==null){
            throw new BusinessException(ResultCode.USER_NOT_EXIST);
        }
        SysUser user2 = this.getByUserName(param.getUserName());
        if(user2 !=null){
            throw new BusinessException(ResultCode.USER_EXIST);
        }
        user.setId(null);
        user.setManageId(user.getId());
        user.setRoleId(param.getRoleId());
        user.setCreateUserId(param.getCreateUserId());
        user.setUpdateTime(null);
        this.save(user);
    }

    @Override
    public PageInfo pageList(SysUserParam param) {
        Page<SysUserVo> page = this.getBaseMapper().pageList(new Page<>(param.getPageNum(),param.getPageSize()),param);
        return PageInfo.PageInfo(page);
    }

    @Override
    public HashMap<Long, Long> groupByRoleId() {
        HashMap<Long, Long>  map = new HashMap<>();
        List<SysRoleVo> roleVos = this.getBaseMapper().groupByRoleId();
        for (SysRoleVo roleVo : roleVos) {
            map.put(roleVo.getId(),roleVo.getAdminCount());
        }
        return map;
    }

    @Override
    public HashMap<Long, SysUser> getByIds(Set<Long> sysUserIds) {
        HashMap<Long, SysUser> map = new HashMap<>();
        if(sysUserIds.size() >0){
            List<SysUser> sysUsers = this.listByIds(sysUserIds);
            sysUsers.forEach(entity-> map.put(entity.getId(),entity));
        }
        return map;
    }

    @Override
    public void saveByRedisKey(String redisKey) {
        String value = redisUtil.get(redisKey);
        if(StringUtils.isBlank(value)){
            return;
        }

        ManageLoginResponse manageLoginResponse = JSONObject.parseObject(value, ManageLoginResponse.class);

        SysUser sysUser = this.getByManageId(manageLoginResponse.getId());
        if(sysUser == null){
            sysUser= new SysUser();
            BeanUtils.copyProperties(manageLoginResponse,sysUser);
            sysUser.setId(null);
            sysUser.setRoleId(null);
            sysUser.setManageId(manageLoginResponse.getId());
            this.save(sysUser);
        }

    }

    @Override
    public SysUser getByManageId(Long userId) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysUser::getManageId,userId);
        return this.getOne(wrapper);
    }

    @Override
    public HashMap<Long, SysUser> getMapByIds(Set<Long> userIds) {
        HashMap<Long,SysUser> map = new HashMap<>();
        if(userIds == null || userIds.size() <= 0){
            return map;
        }
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(SysUser::getId,userIds);
        List<SysUser> list = this.list(wrapper);
        list.forEach(entity -> map.put(entity.getId(),entity));
        return map;
    }

    @Override
    @DS("db2")
    public Object allList(String userName) {
        LambdaQueryWrapper<SysUser> wrapper = new LambdaQueryWrapper<>();
        if(StringUtils.isNotBlank(userName)){
            wrapper.like(SysUser::getUserName,userName);
        }
        return this.list(wrapper);
    }
}