From 7d87742aec63385f2cb0a72ed431a207a31a1100 Mon Sep 17 00:00:00 2001
From: lihongjie0209 <lihongjie0209@gmail.com>
Date: Sun, 10 Mar 2024 10:20:49 +0800
Subject: [PATCH] =?UTF-8?q?submitToken=20=E6=B3=A8=E8=A7=A3=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../coal/annotation/SubmitToken.java          | 13 +++++++++++++
 .../coal/resource/entity/ResourceEntity.java  |  4 ++++
 .../resource/service/ResourceService.java     | 19 ++++++++++++-------
 3 files changed, 29 insertions(+), 7 deletions(-)
 create mode 100644 src/main/java/cn/lihongjie/coal/annotation/SubmitToken.java

diff --git a/src/main/java/cn/lihongjie/coal/annotation/SubmitToken.java b/src/main/java/cn/lihongjie/coal/annotation/SubmitToken.java
new file mode 100644
index 00000000..c398238b
--- /dev/null
+++ b/src/main/java/cn/lihongjie/coal/annotation/SubmitToken.java
@@ -0,0 +1,13 @@
+package cn.lihongjie.coal.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ElementType.TYPE, ElementType.METHOD})
+public @interface SubmitToken {
+
+    boolean value() default true;
+}
diff --git a/src/main/java/cn/lihongjie/coal/resource/entity/ResourceEntity.java b/src/main/java/cn/lihongjie/coal/resource/entity/ResourceEntity.java
index d041aaf6..29d28042 100644
--- a/src/main/java/cn/lihongjie/coal/resource/entity/ResourceEntity.java
+++ b/src/main/java/cn/lihongjie/coal/resource/entity/ResourceEntity.java
@@ -81,6 +81,10 @@ public class ResourceEntity extends CommonEntity {
     private Boolean rateLimit;
 
 
+    @Comment("是否校验提交token")
+    private Boolean submitToken;
+
+
 
     private static String getParent(String path) {
         log.debug("getParent {}", path);
diff --git a/src/main/java/cn/lihongjie/coal/resource/service/ResourceService.java b/src/main/java/cn/lihongjie/coal/resource/service/ResourceService.java
index cdafd5ef..dd1e30bc 100644
--- a/src/main/java/cn/lihongjie/coal/resource/service/ResourceService.java
+++ b/src/main/java/cn/lihongjie/coal/resource/service/ResourceService.java
@@ -1,7 +1,9 @@
 package cn.lihongjie.coal.resource.service;
 
+import cn.lihongjie.coal.annotation.Anonymous;
 import cn.lihongjie.coal.annotation.RateLimit;
 import cn.lihongjie.coal.annotation.SignCheck;
+import cn.lihongjie.coal.annotation.SubmitToken;
 import cn.lihongjie.coal.base.dto.CommonQuery;
 import cn.lihongjie.coal.base.dto.IdRequest;
 import cn.lihongjie.coal.base.service.BaseService;
@@ -202,13 +204,22 @@ public class ResourceService extends BaseService<ResourceEntity, ResourceReposit
             entity.setType("3");
             entity.setName("");
             entity.setUrl(allUrl._1);
-            entity.setAnonymous(StringUtils.equalsAny(allUrl._1, "/login", "/logout", "/genCaptcha"));
+
+            Anonymous anonymous = allUrl._2.getMethodAnnotation(Anonymous.class);
+
+            entity.setAnonymous(
+                    StringUtils.equalsAny(allUrl._1, "/login", "/logout", "/genCaptcha")
+                            || (anonymous != null && anonymous.value()));
 
             RateLimit rateLimit = allUrl._2.getMethodAnnotation(RateLimit.class);
             entity.setRateLimit(rateLimit == null || rateLimit.value());
 
             SignCheck signCheck = allUrl._2.getMethodAnnotation(SignCheck.class);
             entity.setSignCheck(signCheck == null || signCheck.value());
+
+            SubmitToken submitToken = allUrl._2.getMethodAnnotation(SubmitToken.class);
+            entity.setSubmitToken(submitToken == null || submitToken.value());
+
             root.addChildren(entity);
         }
 
@@ -249,18 +260,12 @@ public class ResourceService extends BaseService<ResourceEntity, ResourceReposit
                 .findAny();
     }
 
-
     @Cacheable(cacheNames = Constants.CACHE_RESOURCE_BY_URL_2, key = "#url")
     public Optional<ResourceDto> findUrlFromCache2(String url) {
 
-
-
         return Optional.ofNullable(repository.findByCodeAndType(url, "3"));
-
-
     }
 
-
     public void clearCache() {
 
         cacheManager.getCache(Constants.CACHE_RESOURCE).clear();