001/**
002 * Copyright (c) 2015-2022, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.apache.org/licenses/LICENSE-2.0
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package io.jboot.components.limiter.interceptor;
017
018import io.jboot.aop.InterceptorBuilder;
019import io.jboot.aop.Interceptors;
020import io.jboot.aop.annotation.AutoLoad;
021import io.jboot.components.limiter.LimiterManager;
022import io.jboot.components.limiter.annotation.EnableLimit;
023import io.jboot.core.weight.Weight;
024
025import java.lang.reflect.Method;
026
027/**
028 * @author michael yang (fuhai999@gmail.com)
029 */
030@AutoLoad
031@Weight(-1)
032public class LimiterInterceptorBuilder implements InterceptorBuilder {
033
034    private LimiterManager manager = LimiterManager.me();
035
036    @Override
037    public void build(Class<?> targetClass, Method method, Interceptors interceptors) {
038
039        if (manager.isEnable() && !manager.getLimitConfigBeans().isEmpty()) {
040            interceptors.add(LimiterGlobalInterceptor.class);
041            return;
042        }
043
044        if (Util.hasAnnotation(method, EnableLimit.class)) {
045            interceptors.add(LimiterInterceptor.class);
046        }
047    }
048
049
050}