acl_tousu/src/main/java/com/nokia/service/user/UserServiceImpl.java

package com.nokia.service.user;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.nokia.dao.CityDao;
import com.nokia.dao.RoleDao;
import com.nokia.dao.UserDao;
import com.nokia.dao.UserRoleDao;
import com.nokia.pojo.CityEntity;
import com.nokia.pojo.UserEntity;
import com.nokia.pojo.UserRoleEntity;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class UserServiceImpl extends ServiceImpl<UserDao, UserEntity> implements UserService {

    private final UserRoleDao userRoleDao;

    private final CityDao cityDao;

    @Autowired
    public UserServiceImpl(RoleDao roleDao, UserRoleDao userRoleDao, CityDao cityDao) {
        this.userRoleDao = userRoleDao;
        this.cityDao = cityDao;
    }

    @Transactional
    public String add(UserEntity table2User) {
        UserEntity query = new UserEntity();
        query.setLoginName(table2User.getLoginName());
        UserEntity one = baseMapper.selectOne(Wrappers.query(query));
        if (one != null) {
            return "loginName: " + table2User.getLoginName() + " 已存在";
        } else {
            table2User.setCityId(table2User.getCityId());
            baseMapper.insert(table2User);
            List<Integer> roleEntityList1 = table2User.getRoleIdList();
            UserRoleEntity userRoleEntity = new UserRoleEntity();
            for (Integer integer : roleEntityList1) {
                userRoleEntity.setUserId(table2User.getUserId());
                userRoleEntity.setRoleId(integer);
                userRoleDao.insert(userRoleEntity);
            }
        }
        return "ok";
    }

    @Override
    @Transactional
    public String delete(Integer userId) {
        // UserEntity userEntity = baseMapper.selectById(userId);
        /* cityDao.deleteById(userEntity.getCityId()); */
        QueryWrapper<UserRoleEntity> objectQueryWrapper = new QueryWrapper<>();
        objectQueryWrapper.eq("user_id", userId);
        List<UserRoleEntity> userRoleEntities = userRoleDao.selectList(objectQueryWrapper);
        for (UserRoleEntity userRoleEntity : userRoleEntities) {
            Integer id = userRoleEntity.getId();
            userRoleDao.deleteById(id);
        }
        baseMapper.deleteById(userId);
        return "ok";
    }

    @Override
    @Transactional
    public String update(UserEntity user) {
        baseMapper.updateById(user);
        Integer userId = user.getUserId();
        QueryWrapper<UserRoleEntity> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("user_id", userId);
        userRoleDao.delete(queryWrapper);
        for (Integer rolid : user.getRoleIdList()) {
            UserRoleEntity userRoleEntity = new UserRoleEntity();
            userRoleEntity.setUserId(user.getUserId());
            userRoleEntity.setRoleId(rolid);
            userRoleDao.insert(userRoleEntity);
        }
        return "ok";
    }

    @Override
    public Map<String, Object> select(Integer userId) {
        Map<String, Object> hashMap = new HashMap<>();
        if (userId == null) {
            hashMap.put("userId", "为空");
            return hashMap;
        }
        UserEntity userEntity = baseMapper.selectById(userId);
        QueryWrapper<UserEntity> userEntityQueryWrapper = new QueryWrapper<>();
        userEntityQueryWrapper.eq("user_id", userId);
        UserEntity one = baseMapper.selectOne(userEntityQueryWrapper);
        if (one == null) {
            hashMap.put("userId", "不正确");
            return hashMap;
        }
        Integer cityId = userEntity.getCityId();
        CityEntity cityEntity = cityDao.selectById(cityId);
        // 他有两个角色 对 我知道 用for遍历来查 不需要for遍历查
        QueryWrapper<UserRoleEntity> objectQueryWrapper = new QueryWrapper<>();
        objectQueryWrapper.eq("user_id", userId);
        List<UserRoleEntity> userRoleEntities = userRoleDao.selectList(objectQueryWrapper);
        hashMap.put("userEntity", userEntity);
        hashMap.put("cityEntity", cityEntity);
        hashMap.put("userRoleEntities", userRoleEntities);
        return hashMap;
    }

}