pipIrr-platform/pipIrr-web/pipIrr-web-base/src/main/java/com/dy/pipIrrBase/user/UserSv.java
@@ -3,7 +3,9 @@
import com.dy.common.webUtil.QueryResultVo;
import com.dy.pipIrrGlobal.daoBa.BaUserMapper;
import com.dy.pipIrrGlobal.daoBa.BaUserRoleMapper;
import com.dy.pipIrrGlobal.pojoBa.BaUser;
import com.dy.pipIrrGlobal.pojoBa.BaUserRole;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
@@ -19,25 +21,31 @@
public class UserSv {
    private BaUserMapper dao;
    private BaUserRoleMapper urDao;
    @Autowired
    private void setDao(BaUserMapper dao){
        this.dao = dao;
    }
    @Autowired
    private void setDao(BaUserRoleMapper dao){
        this.urDao = dao;
    }
    /**
     * 得到一个片区
     * @param id 片区ID
     * @return 片区实体
     * 得到一个用户
     * @param id 用户ID
     * @return 用户实体
     */
    public BaUser selectById(Long id){
        return this.dao.selectById(id) ;
    }
    /**
     * 得到一个片区
     * 得到一个用户
     * @param vo 查询条件值对象
     * @return 片区实体
     * @return 用户实体
     */
    @SuppressWarnings("unchecked")
    public QueryResultVo<List<BaUser>> selectSome(QueryVo vo){
@@ -84,6 +92,23 @@
    }
    /**
     * 设置用户角色
     * @param userId 用户id
     * @param roleIds 选择的角色id集合
     * @return 插入用户与角色关联记录数量
     */
    public int setRoles(Long userId, Long[] roleIds){
        this.urDao.deleteByUserId(userId) ;
        int count = 0 ;
        if(roleIds != null && roleIds.length > 0){
            for(Long roleId : roleIds){
                count += this.urDao.insertSelective(new BaUserRole(userId,roleId)) ;
            }
        }
        return count ;
    }
    /**
     * 保存修改实体
     * @param id 实体ID
     * @return 影响记录数量