first

src/main/java/com/spring/modules/sys/dao/SysUserDao.java

@@ -3,8 +3,10 @@
 package com.spring.modules.sys.dao;
 
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.spring.modules.base.entity.AccessSiteData;
 import com.spring.modules.sys.entity.SysUserEntity;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
 
 import java.util.List;
 
@@ -32,4 +34,27 @@ public interface SysUserDao extends BaseMapper<SysUserEntity> {
 	 */
 	SysUserEntity queryByUserName(String username);
 
+	/**
+	 * @Description 检查权限
+	 * @Title checkAccessSite
+	 * @param site
+	 * @param username
+	 * @author rq
+	 * @date 2022/11/1 16:53
+	 * @return List<AccessSiteData>
+	 * @throw
+	 */
+	List<AccessSiteData> checkAccessSite(@Param("site") String site, @Param("username") String username);
+
+	/**
+	 * @Description 添加工厂权限
+	 * @Title addAccessSite
+	 * @param inData
+	 * @author rq
+	 * @date 2022/11/1 17:45
+	 * @return void
+	 * @throw
+	 */
+	void addAccessSite(AccessSiteData inData);
+
 }

src/main/java/com/spring/modules/sys/service/impl/SysUserServiceImpl.java

@@ -87,7 +87,14 @@ public class SysUserServiceImpl extends ServiceImpl<SysUserDao, SysUserEntity> i
 		
 		//检查角色是否越权
 		checkRole(user);
-		
+		//添加工厂权限
+		List<AccessSiteData> checkList=baseMapper.checkAccessSite( user.getSite(),  user.getUsername());
+		if(checkList.size()==0){
+			AccessSiteData newData=new AccessSiteData();
+			newData.setSite(user.getSite());
+			newData.setUserID(user.getUsername());
+			baseMapper.addAccessSite(newData);
+		}
 		//保存用户与角色关系
 		sysUserRoleService.saveOrUpdate(user.getUserId(), user.getRoleIdList());
 	}

src/main/resources/mapper/sys/SysUserDao.xml

@@ -21,4 +21,12 @@
 		select * from sys_user where username = #{username}
 	</select>
 
+	    <select id="checkAccessSite" resultType="com.spring.modules.base.entity.AccessSiteData">
+        select Site,userID from AccessSite where site=#{site} and userID=#{username}
+    </select>
+
+     <insert id="addAccessSite">
+        insert into AccessSite(UserID,Site) values (#{userID},#{site})
+    </insert>
+
 </mapper>