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 findUrlFromCache2(String url) { - - return Optional.ofNullable(repository.findByCodeAndType(url, "3")); - - } - public void clearCache() { cacheManager.getCache(Constants.CACHE_RESOURCE).clear();