如何在Spring中动态禁用特定的API?

我有一个标志DISABLE_flaG,我想用它来控制不同控制器中的多个特定API。

@RestController
public final class Controller1 {
    @RequestMapping(value = "/foo1",method = RequestMethod.POST)
    public String foo1()
}
@RestController
public final class Controller2 {
    @RequestMapping(value = "/foo2",method = RequestMethod.POST)
    public String foo2()
}

我可以使用拦截器来处理所有网址。有没有一种简单的方法可以完成类似注解的操作?

yuguanhai1 回答:如何在Spring中动态禁用特定的API?

我认为没有直接的方法可以禁用构造的请求映射,但是我们可以在一定条件下以多种方式禁用API。

以下是通过spring配置文件或JVM属性禁用的两种方法。

public class SampleController {
    @Autowired
    Environment env;

    @RequestMapping(value = "/foo",method = RequestMethod.POST)
    public String foo(HttpServletResponse response) {
        // Using profile
        if (env.acceptsProfiles("staging")) {
            response.setStatus(404);
            return "";
        }

        // Using JVM options
        if("true".equals(System.getProperty("DISABLE_FOO"))) {
            response.setStatus(404);
            return "";
        }

        return "";
    }
}

如果您正在考虑使用云配置的未来解决方案是最好的方法。 https://spring.io/guides/gs/centralized-configuration/

使用条件组件

这允许使用条件构建bean,如果条件在启动时失败,则将永远不会构建整个组件。将所有可选的请求映射分组到新控制器,并添加条件注释

@Conditional(ConditionalController.class)
public class SampleController {
    @Autowired
    Environment env;

    @RequestMapping(value = "/foo",method = RequestMethod.POST)
    public String foo(HttpServletResponse response) {
        return "";
    }

    public static class ConditionalController implements Condition {

        @Override
        public boolean matches(ConditionContext context,AnnotatedTypeMetadata metadata) {
            return context.getEnvironment().acceptsProfiles("staging"); // Or whatever condition
        }

    }
}
,

您可以使用AOP执行类似的操作。

创建自己的注释...

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Maybe { }

和相应的方面...

@Aspect
public class MaybeAspect {

  @Pointcut("@annotation(com.example.Maybe)")
  public void callMeMaybe() {}

  @Around("callMeMaybe()")
  public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
    // do your logic here..
    if(DISABLE_FOO) {
      // do nothing ? throw exception?
      // return null;
      throw new IllegalStateException();
    } else {
      // process the request normally
      return joinPoint.proceed();
    }
  }
}
,

您可以通过利用弹簧轮廓来解决此问题。您定义两个配置文件,一个用于启用标志,另一个用于禁用标志。您的示例如下所示:

@Profile("DISABLED_FLAG")
@RestController
public final class Controller1 {
    @RequestMapping(value = "/foo1",method = RequestMethod.POST)
    public String foo1()
}

@Profile("ENABLED_FLAG")
@RestController
public final class Controller2 {
    @RequestMapping(value = "/foo2",method = RequestMethod.POST)
    public String foo2()
}

以下是此功能的Spring框架文档的链接:https://docs.spring.io/spring-framework/docs/current/javadoc-api/org/springframework/context/annotation/Profile.html

,

我是这样做的:

@Retention(RUNTIME)
@Target(ElementType.METHOD)
public @interface DisableApiControl {
}

这个类是我的自定义语句。之后可以使用 AOP :

对于 AbstractBaseServiceImpl :

public abstract class AbstractBaseServiceImpl {

    private static boolean disableCheck = false;

    public void setDisableChecker(boolean checkParameter) {
        disableCheck = checkParameter;
    }

    public boolean getDisableChecker() {
        return disableCheck;
    }
}

注意:上面的类已经准备好提供一个动态结构。

@Aspect
@Component
public class DisableApiControlAspect extends AbstractBaseServiceImpl {

    @Autowired
    private HttpServletResponse httpServletResponse;

    @Pointcut(" @annotation(disableMe)")
    protected void disabledMethods(DisableApiControl disableMe) {

        // comment line 
        
    }

    @Around("disabledMethods(disableMe)")
    public Object dontRun(ProceedingJoinPoint joinPoint,DisableApiControl disableMe) throws Throwable {
        if (getDisableChecker()) {
            httpServletResponse.sendError(HttpStatus.NOT_FOUND.value(),"Not found");
            return null;
        } else {
            return joinPoint.proceed();
        }
    }
}

checker 参数此时已添加到全局。当需要时将值指定为 true / false 时,其余的会更容易。

@GetMapping("/map")
@DisableApiControl
public List<?> stateMachineFindMap() {
    return new ArrayList<>;
}
本文链接:https://www.f2er.com/3135171.html

大家都在问