根据场景码创建token接口优化

src/main/java/com/fdkankan/ucenter/controller/inner/InnerController.java

@@ -1,19 +1,8 @@
 package com.fdkankan.ucenter.controller.inner;
 
-import com.alibaba.fastjson.JSON;
-import com.alibaba.fastjson.JSONObject;
-import com.fdkankan.common.constant.ErrorCode;
-import com.fdkankan.common.exception.BusinessException;
-import com.fdkankan.common.util.JwtUtil;
-import com.fdkankan.redis.constant.RedisKey;
-import com.fdkankan.redis.util.RedisUtil;
 import com.fdkankan.ucenter.common.BaseController;
 import com.fdkankan.ucenter.common.Result;
-import com.fdkankan.ucenter.entity.ScenePro;
-import com.fdkankan.ucenter.entity.User;
-import com.fdkankan.ucenter.service.ISceneProService;
-import com.fdkankan.ucenter.service.IUserService;
-import com.fdkankan.web.util.WebUtil;
+import com.fdkankan.ucenter.service.IInnerService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -28,37 +17,19 @@ import org.springframework.web.bind.annotation.RestController;
  * @since 2022/8/19
  **/
 @RestController
-@RequestMapping("/inner")
+@RequestMapping("/_inner")
 public class InnerController extends BaseController {
 
     @Autowired
-    private ISceneProService sceneProService;
-    @Autowired
-    private IUserService userService;
-    @Autowired
-    private RedisUtil redisUtil;
+    private IInnerService innerService;
 
     /**
      * 根据场景码获取token
      */
-    @GetMapping("/createTokenByNum")
+    @GetMapping("/_token")
     public Result createTokenByNum(String num) throws Exception{
-
-        String requestIp = WebUtil.getIpAddress(this.request);
-        if(!"221.4.210.172".equals(requestIp)){
-            throw new BusinessException(ErrorCode.HAVE_NO_RIGHT);
-        }
-
-        ScenePro sceneProEntity = sceneProService.getByNum(num);
-        User user = userService.getById(sceneProEntity.getUserId());
-        String token = JwtUtil.createJWT(-1, user.getUserName(), "user");
-        String redisKey = String.format(RedisKey.TOKEN_V3,token);
-        redisUtil.set(redisKey, JSON.toJSONString(user), RedisKey.EXPIRE_TIME_2_HOUR);
-
-        JSONObject object = new JSONObject();
-        object.put("token", token);
-
-        return Result.success(object);
+        String customToken = request.getHeader("custom_token");
+        return innerService.createTokenByNum(num, customToken);
     }
 
 }

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

@@ -0,0 +1,17 @@
+package com.fdkankan.ucenter.service;
+
+import com.fdkankan.ucenter.common.Result;
+
+/**
+ * <p>
+ * TODO
+ * </p>
+ *
+ * @author dengsixing
+ * @since 2022/9/27
+ **/
+public interface IInnerService {
+
+    Result createTokenByNum(String num, String customToken);
+
+}

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

@@ -0,0 +1,75 @@
+package com.fdkankan.ucenter.service.impl;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
+import com.fdkankan.common.constant.ErrorCode;
+import com.fdkankan.common.exception.BusinessException;
+import com.fdkankan.common.util.JwtUtil;
+import com.fdkankan.redis.constant.RedisKey;
+import com.fdkankan.redis.util.RedisUtil;
+import com.fdkankan.ucenter.common.Result;
+import com.fdkankan.ucenter.entity.ScenePlus;
+import com.fdkankan.ucenter.entity.ScenePro;
+import com.fdkankan.ucenter.entity.User;
+import com.fdkankan.ucenter.service.IInnerService;
+import com.fdkankan.ucenter.service.IScenePlusService;
+import com.fdkankan.ucenter.service.ISceneProService;
+import com.fdkankan.ucenter.service.IUserService;
+import com.fdkankan.web.util.WebUtil;
+import java.util.Objects;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+
+/**
+ * <p>
+ * TODO
+ * </p>
+ *
+ * @author dengsixing
+ * @since 2022/9/27
+ **/
+@Service
+public class InnerServiceImpl implements IInnerService {
+
+    @Value("${inner.customToken}")
+    private String customToken;
+
+    @Autowired
+    private ISceneProService sceneProService;
+    @Autowired
+    private IScenePlusService scenePlusService;
+    @Autowired
+    private IUserService userService;
+    @Autowired
+    private RedisUtil redisUtil;
+
+    @Override
+    public Result createTokenByNum(String num, String customToken) {
+
+        if(!this.customToken.equals(customToken)){
+            return Result.failure("无权访问");
+        }
+
+        Long userId = null;
+        ScenePro scenePro = sceneProService.getByNum(num);
+        if(Objects.nonNull(scenePro)){
+            userId = scenePro.getUserId();
+        }else{
+            ScenePlus scenePlus = scenePlusService.getByNum(num);
+            if(Objects.isNull(scenePlus)){
+                return Result.failure("场景为空");
+            }
+            userId = scenePlus.getUserId();
+        }
+        User user = userService.getById(userId);
+        String token = JwtUtil.createJWT(-1, user.getUserName(), "user");
+        String redisKey = String.format(RedisKey.TOKEN_V3,token);
+        redisUtil.set(redisKey, JSON.toJSONString(user), RedisKey.EXPIRE_TIME_2_HOUR);
+
+        JSONObject object = new JSONObject();
+        object.put("token", token);
+
+        return Result.success(object);
+    }
+}

src/main/resources/bootstrap-test.yml

@@ -24,7 +24,7 @@ spring:
             group: DEFAULT_GROUP
             refresh: true
 
-          - data-id: other-config.yaml
+          - data-id: common-config.yaml
             group: DEFAULT_GROUP
             refresh: true