fxkh-txl-service/src/main/java/com/whdc/service/impl/UserServiceImpl.java

package com.whdc.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.whdc.exception.MyException;
import com.whdc.mapper.AddressBookMapper;
import com.whdc.mapper.UserMapper;
import com.whdc.model.dto.LoginDto;
import com.whdc.model.dto.UserDto;
import com.whdc.model.entity.AddressBook;
import com.whdc.model.entity.User;
import com.whdc.model.vo.LoginVo;
import com.whdc.service.IUserService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.whdc.utils.ResultJson;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;

import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.UUID;


/**
 * <p>
 * 服务实现类
 * </p>
 * @author xusan
 * @date 2024-05-11
 */
@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements IUserService {

    public static final String DEFAULT_PASSWORD = "Fxkh@123456";

    @Autowired
    private AddressBookMapper addressBookMapper;

    @Override
    public IPage<User> page(User dto) {
        return baseMapper.page(new Page<>(), dto);
    }


    @Override
    public List<User> find(User dto) {
        return baseMapper.find(dto);
    }



    @Override
    public LoginVo login(LoginDto dto) {

        User sysUser = findByLoginName(dto.getLoginName());

        if (sysUser == null) {
            throw new MyException("用户名或密码错误");
        }

        if (getPassword(sysUser.getSalt() ,dto.getPassword()).equals(sysUser.getPassword())) {
            LoginVo out = new LoginVo();
            BeanUtils.copyProperties(sysUser, out);

            AddressBook book = addressBookMapper.selectById(sysUser.getAbId());
            if (Objects.nonNull(book)){
                out.setName(book.getName());
            }
            return out;

        }

        throw new MyException("用户名或密码错误");
    }

    @Override
    public boolean register(User entity) {
        // 生成盐值
        String salt = UUID.randomUUID().toString().replaceAll("-", "");
        // 生成密码
        entity.setSalt(salt)
                .setPassword(getPassword(salt, DEFAULT_PASSWORD));
        return this.save(entity);
    }

    @Override
    public String getRole(Integer id) {

        User byId = this.getById(id);
        if (byId == null){
            throw new MyException("该用户不存在");
        }

        return byId.getRole();
    }

    @Override
    public boolean register(String phone, Integer abId, String role) {
        User user = new User();
        user.setAbId(abId)
                .setRole(role)
                .setUsername(phone);
        return register(user);
    }

    @Override
    public boolean updateLastDate(Integer id) {
        return this.lambdaUpdate()
                .set(User::getLastLoginDate,new Date())
                .eq(User::getId,id)
                .update();
    }

    @Override
    public ResultJson updatePwd(UserDto dto) {


        User sysUser = this.getById(dto.getId());
        if (Objects.isNull(sysUser)) {

            return ResultJson.error("当前数据不存在");

        }


        if (getPassword(sysUser.getSalt() ,dto.getOldPassword()).equals(sysUser.getPassword())) {
            // 更新密码
            sysUser.setPassword(getPassword(sysUser.getSalt() ,dto.getNewPassword()));
            return ResultJson.ok(sysUser.updateById());
        }

        return ResultJson.error("密码错误");
    }

    private User findByLoginName(String loginName) {
        if (loginName == null) {
            throw new MyException("登录名不能为空");
        }

        return this.lambdaQuery().eq(User::getUsername, loginName).one();

    }

    private String getPassword(String salt,String password) {
        return DigestUtils.md5DigestAsHex((salt + password).getBytes());
    }

    public static void main(String[] args) {
        String salt = UUID.randomUUID().toString().replaceAll("-", "");
        System.out.println(salt);
        System.out.println(DigestUtils.md5DigestAsHex((salt + DEFAULT_PASSWORD).getBytes()));
    }
}