소스 검색

update VideoService

reghao 1 년 전
부모
커밋
93893553a2
1개의 변경된 파일5개의 추가작업 그리고 1개의 파일을 삭제
  1. 5 1
      content/content-service/src/main/java/cn/reghao/tnb/content/app/vod/service/VideoService.java

+ 5 - 1
content/content-service/src/main/java/cn/reghao/tnb/content/app/vod/service/VideoService.java

@@ -13,6 +13,8 @@ import lombok.extern.slf4j.Slf4j;
 import org.apache.dubbo.config.annotation.DubboReference;
 import org.springframework.stereotype.Service;
 
+import java.util.Set;
+
 /**
  * @author reghao
  * @date 2022-08-05 13:49:28
@@ -25,15 +27,17 @@ public class VideoService {
 
     private final VideoPostMapper videoPostMapper;
     private final VideoErrorMapper videoErrorMapper;
+    private final Set<Long> adminUsers;
 
     public VideoService(VideoPostMapper videoPostMapper, VideoErrorMapper videoErrorMapper) {
         this.videoPostMapper = videoPostMapper;
         this.videoErrorMapper = videoErrorMapper;
+        this.adminUsers = Set.of(10001L, 10002L);
     }
 
     public Result errorReport(VideoErrorReport videoErrorReport) {
         long loginUser = UserContext.getUser();
-        if (loginUser == -1 || loginUser > 10005) {
+        if (!adminUsers.contains(loginUser)) {
             return Result.fail("no permission");
         }