com.github.yizzuide.milkomeda.particle.ParticleConfig Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2021 yizzuide All rights Reserved.
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package com.github.yizzuide.milkomeda.particle;

import com.github.yizzuide.milkomeda.universe.context.SpringContext;
import com.github.yizzuide.milkomeda.util.IOUtils;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.OrderComparator;
import org.springframework.core.Ordered;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.lang.NonNull;
import org.springframework.util.CollectionUtils;
import org.springframework.web.filter.DelegatingFilterProxy;

import java.util.*;
import java.util.stream.Collectors;

/**
 * ParticleConfig
 *
 * @author yizzuide
 * @since 1.14.0
 * @version 3.12.10
 * 

 * Create at 2019/11/11 11:26
 */
@Slf4j
@Configuration
@ConditionalOnClass(RedisTemplate.class)
@AutoConfigureAfter(RedisAutoConfiguration.class)
@EnableConfigurationProperties(ParticleProperties.class)
public class ParticleConfig implements ApplicationContextAware {

    @Autowired
    private ParticleProperties particleProperties;

    /**
     * 缓存创建的LimitHandler
     */
    private static final Map cacheHandlerBeans = new HashMap<>();

    @Bean
    public ParticleAspect particleAspect() {
        return new ParticleAspect();
    }

    @Bean
    public IdempotentLimiter idempotentLimiter() {
        IdempotentLimiter idempotentLimiter = new IdempotentLimiter();
        idempotentLimiter.setExpire(60L);
        return idempotentLimiter;
    }

    @Bean
    public ParticleFilter particleFilter() {
        return new ParticleFilter();
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    @Bean
    @ConditionalOnProperty(prefix = "milkomeda.particle", name = "enable-filter", havingValue = "true")
    public FilterRegistrationBean particleFilterRegistrationBean() {
        FilterRegistrationBean particleFilterRegistrationBean = new FilterRegistrationBean();
        particleFilterRegistrationBean.setFilter(new DelegatingFilterProxy("particleFilter"));
        particleFilterRegistrationBean.setName("particleFilter");
        particleFilterRegistrationBean.setUrlPatterns(Collections.singleton("/*"));
        particleFilterRegistrationBean.setOrder(Ordered.HIGHEST_PRECEDENCE + 20);
        return particleFilterRegistrationBean;
    }

    @SneakyThrows
    @Override
    public void setApplicationContext(@NonNull ApplicationContext applicationContext) throws BeansException {
        List limiters = particleProperties.getLimiters();
        if (CollectionUtils.isEmpty(limiters)) {
            return;
        }
        List barrierLimiters = new ArrayList<>();
        for (ParticleProperties.Limiter limiter : limiters) {
            String limiterName = limiter.getName();
            // 使用枚举类型填充处理器clazz
            if (limiter.getHandlerClazz() == null && limiter.getType() != null) {
                switch (limiter.getType()) {
                    case IDEMPOTENT:
                        limiter.setHandlerClazz(IdempotentLimiter.class);
                        break;
                    case TIMES:
                        limiter.setHandlerClazz(TimesLimiter.class);
                        break;
                    case BARRIER:
                        limiter.setHandlerClazz(BarrierLimiter.class);
                        break;
                    case BLOOM:
                        limiter.setHandlerClazz(BloomLimiter.class);
                        break;
                }
            }
            LimitHandler limitHandler = SpringContext.registerBean((ConfigurableApplicationContext) applicationContext, limiterName, limiter.getHandlerClazz(), limiter.getProps());
            limitHandler.setExpire(limiter.getKeyExpire().getSeconds());
            limiter.setLimitHandler(limitHandler);
            if (limiter.getHandlerClazz() == BarrierLimiter.class) {
                barrierLimiters.add(limiter);
            }
            // 缓存并设置属性
            cacheHandlerBeans.put(limiterName, limitHandler);
        }

        // 创建barrierLimiter类型限制器链
        for (ParticleProperties.Limiter limiter : barrierLimiters) {
            BarrierLimiter barrierLimiter = (BarrierLimiter) limiter.getLimitHandler();
            List chain = barrierLimiter.getChain();
            if (CollectionUtils.isEmpty(chain)) {
                log.warn("Particle add barrier limiter find chain is empty, make true add at least one limiter.");
                continue;
            }
            List handlers = new ArrayList<>();
            for (String name : chain) {
                handlers.add(cacheHandlerBeans.get(name));
            }
            barrierLimiter.addLimitHandlerList(handlers);
        }

        // 根据order排序
        List orderLimiters = limiters.stream()
                .sorted(OrderComparator.INSTANCE.withSourceProvider(limiter -> limiter)).collect(Collectors.toList());
        particleProperties.setLimiters(orderLimiters);

        // 读取lua脚本
        String luaScript = IOUtils.loadLua(IOUtils.LUA_PATH, "particle_times_limiter.lua");
        TimesLimiter.setLuaScript(luaScript);
    }

    static Map getCacheHandlerBeans() {
        return cacheHandlerBeans;
    }
}