消费返现相关

cash-common/cash-common-sa-token/src/main/java/com/czg/annotation/SaCheckMainShop.java

@@ -0,0 +1,25 @@
+package com.czg.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * 权限认证(User版)：必须具有指定权限才能进入该方法
+ * <p> 可标注在函数、类上（效果等同于标注在此类的所有方法上）
+ * @author click33
+ *
+ */
+//@SaCheckPermission(type = "user")
+@Retention(RetentionPolicy.RUNTIME)
+@Target({ ElementType.METHOD, ElementType.TYPE})
+public @interface SaCheckMainShop {
+
+    /**
+     * 需要校验的权限码
+     * @return 需要校验的权限码
+     */
+    String  value() default "";
+
+}

cash-common/cash-common-sa-token/src/main/java/com/czg/handler/SaCheckMainShopHandler.java

@@ -0,0 +1,34 @@
+package com.czg.handler;
+
+import cn.dev33.satoken.annotation.handler.SaAnnotationHandlerInterface;
+import cn.hutool.core.util.StrUtil;
+import com.czg.annotation.SaCheckMainShop;
+import com.czg.annotation.SaStaffCheckPermission;
+import com.czg.exception.ApiNotPrintException;
+import com.czg.sa.StpKit;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+
+import java.lang.reflect.Method;
+
+/**
+ * 员工权限校验
+ * @author Administrator
+ */
+@Component
+@Slf4j
+public class SaCheckMainShopHandler implements SaAnnotationHandlerInterface<SaCheckMainShop> {
+
+    @Override
+    public Class<SaCheckMainShop> getHandlerAnnotationClass() {
+        return SaCheckMainShop.class;
+    }
+
+    @Override
+    public void checkMethod(SaCheckMainShop at, Method method) {
+        boolean mainShop = StpKit.USER.isMainShop();
+        if (!mainShop) {
+            throw new ApiNotPrintException("此接口仅主店可操作");
+        }
+    }
+}