1

src/main/java/com/fdkankan/ucenter/controller/PlatformController.java

@@ -0,0 +1,21 @@
+package com.fdkankan.ucenter.controller;
+
+
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * <p>
+ *  前端控制器
+ * </p>
+ *
+ * @author 
+ * @since 2024-11-22
+ */
+@RestController
+@RequestMapping("/ucenter/platform")
+public class PlatformController {
+
+}
+

src/main/java/com/fdkankan/ucenter/entity/Platform.java

@@ -0,0 +1,48 @@
+package com.fdkankan.ucenter.entity;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableLogic;
+import com.baomidou.mybatisplus.annotation.TableName;
+import java.io.Serializable;
+import java.util.Date;
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ * <p>
+ * 
+ * </p>
+ *
+ * @author 
+ * @since 2024-11-22
+ */
+@Getter
+@Setter
+@TableName("t_platform")
+public class Platform implements Serializable {
+
+    private static final long serialVersionUID = 1L;
+
+    @TableId(value = "id", type = IdType.AUTO)
+    private Integer id;
+
+    @TableField("platform_name")
+    private String platformName;
+
+    @TableField("platform_key")
+    private String platformKey;
+
+    @TableField("rec_status")
+    @TableLogic(value = "A",delval = "I")
+    private String recStatus;
+
+    @TableField("create_time")
+    private Date createTime;
+
+    @TableField("update_time")
+    private Date updateTime;
+
+
+}

src/main/java/com/fdkankan/ucenter/generate/AutoGenerate.java

@@ -18,7 +18,7 @@ public class AutoGenerate {
         String path =System.getProperty("user.dir") ;
 
         generate(path,"ucenter", getTables(new String[]{
-                "t_user_platform",
+                "t_platform",
         }));
 
 //        generate(path,"goods", getTables(new String[]{

src/main/java/com/fdkankan/ucenter/mapper/IPlatformMapper.java

@@ -0,0 +1,18 @@
+package com.fdkankan.ucenter.mapper;
+
+import com.fdkankan.ucenter.entity.Platform;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * <p>
+ *  Mapper 接口
+ * </p>
+ *
+ * @author 
+ * @since 2024-11-22
+ */
+@Mapper
+public interface IPlatformMapper extends BaseMapper<Platform> {
+
+}

src/main/java/com/fdkankan/ucenter/service/IPlatformService.java

@@ -0,0 +1,16 @@
+package com.fdkankan.ucenter.service;
+
+import com.fdkankan.ucenter.entity.Platform;
+import com.baomidou.mybatisplus.extension.service.IService;
+
+/**
+ * <p>
+ *  服务类
+ * </p>
+ *
+ * @author 
+ * @since 2024-11-22
+ */
+public interface IPlatformService extends IService<Platform> {
+
+}

src/main/java/com/fdkankan/ucenter/service/impl/LoginService.java

@@ -20,10 +20,7 @@ import com.fdkankan.ucenter.common.RedisKeyUtil;
 import com.fdkankan.ucenter.common.constants.NacosProperty;
 import com.fdkankan.ucenter.constant.LoginConstant;
 import com.fdkankan.ucenter.constant.QrCodeFilePath;
-import com.fdkankan.ucenter.entity.Camera;
-import com.fdkankan.ucenter.entity.CameraDetail;
-import com.fdkankan.ucenter.entity.MailTemplate;
-import com.fdkankan.ucenter.entity.User;
+import com.fdkankan.ucenter.entity.*;
 import com.fdkankan.ucenter.service.*;
 import com.fdkankan.ucenter.vo.request.LoginParam;
 import com.fdkankan.ucenter.vo.request.RegisterParam;
@@ -31,9 +28,8 @@ import com.fdkankan.ucenter.vo.response.LoginVo;
 import com.fdkankan.ucenter.vo.response.UserVo;
 
 import java.io.File;
-import java.util.Date;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
+import java.util.stream.Collectors;
 
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.BeanUtils;
@@ -67,6 +63,10 @@ public class LoginService {
     private FYunFileServiceInterface fYunFileServiceInterface;
     @Autowired
     IMailTemplateService mailTemplateService;
+    @Autowired
+    IUserRoleService userRoleService;
+    @Autowired
+    IPlatformService platformService;
 
     public LoginVo login(LoginParam param) {
         if (StringUtils.isEmpty(param.getPassword()) || StringUtils.isEmpty(param.getPhoneNum())){
@@ -93,7 +93,14 @@ public class LoginService {
         LoginVo vo = new LoginVo();
         vo.setToken(token);
         vo.setUser(userVo);
-        vo.setPlatformKeys(userPlatformService.getByUserId(user.getId()));
+
+        Set<Long> byUser = userRoleService.getByUser(user);
+        if(byUser.contains(1L) || byUser.contains(5L)){
+            List<String> keyIds = platformService.list().stream().map(Platform::getPlatformKey).collect(Collectors.toList());
+            vo.setPlatformKeys(keyIds);
+        }else {
+            vo.setPlatformKeys(userPlatformService.getByUserId(user.getId()));
+        }
         return vo;
     }
 

src/main/java/com/fdkankan/ucenter/service/impl/PlatformServiceImpl.java

@@ -0,0 +1,20 @@
+package com.fdkankan.ucenter.service.impl;
+
+import com.fdkankan.ucenter.entity.Platform;
+import com.fdkankan.ucenter.mapper.IPlatformMapper;
+import com.fdkankan.ucenter.service.IPlatformService;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springframework.stereotype.Service;
+
+/**
+ * <p>
+ *  服务实现类
+ * </p>
+ *
+ * @author 
+ * @since 2024-11-22
+ */
+@Service
+public class PlatformServiceImpl extends ServiceImpl<IPlatformMapper, Platform> implements IPlatformService {
+
+}

src/main/resources/mapper/ucenter/PlatformMapper.xml

@@ -0,0 +1,5 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
+<mapper namespace="com.fdkankan.ucenter.mapper.IPlatformMapper">
+
+</mapper>