com.github.likavn.eventbus.config.EventBusAutoConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eventbus-spring-boot-starter Show documentation
There is a newer version: 2.5.0-RC3
Show newest version
/**
 * Copyright 2023-2033, likavn ([email protected]).
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.github.likavn.eventbus.config;

import com.github.likavn.eventbus.BootConnectionWatchdog;
import com.github.likavn.eventbus.core.ConnectionWatchdog;
import com.github.likavn.eventbus.core.DeliveryBus;
import com.github.likavn.eventbus.core.ListenerRegistry;
import com.github.likavn.eventbus.core.annotation.EventbusListener;
import com.github.likavn.eventbus.core.api.MsgSender;
import com.github.likavn.eventbus.core.api.interceptor.*;
import com.github.likavn.eventbus.core.base.Lifecycle;
import com.github.likavn.eventbus.core.base.NodeTestConnect;
import com.github.likavn.eventbus.core.constant.BusConstant;
import com.github.likavn.eventbus.core.metadata.BusConfig;
import com.github.likavn.eventbus.core.metadata.BusType;
import com.github.likavn.eventbus.core.base.InterceptorContainer;
import com.github.likavn.eventbus.prop.BusProperties;
import com.github.likavn.eventbus.provider.rabbit.config.BusBootRabbitConfiguration;
import com.github.likavn.eventbus.provider.redis.config.BusBootRedisConfiguration;
import com.github.likavn.eventbus.provider.rocket.config.BusBootRocketConfiguration;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.util.StringUtils;

import java.util.List;
import java.util.Map;

/**
 * boot 启动配置
 *
 * @author likavn
 * @date 2024/01/01
 **/
@Slf4j
@Configuration
@EnableConfigurationProperties(BusProperties.class)
@ConditionalOnProperty(prefix = BusConstant.CONFIG_PREFIX, name = "enable", havingValue = "true", matchIfMissing = true)
@Import({RequestIdGeneratorConfiguration.class, BusBootRabbitConfiguration.class, BusBootRedisConfiguration.class, BusBootRocketConfiguration.class})
public class EventBusAutoConfiguration {
    private final BusProperties config;

    public EventBusAutoConfiguration(BusProperties config) {
        this.config = config;
        initializing();
    }

    /**
     * 事件总线配置
     */
    private void initializing() {
        log.info("Eventbus Initializing... {}", config.getType());
        BusType.isValid(config.getType());
        if (!config.getType().equals(config.getOldType())) {
            log.info("Eventbus compatibility type by {}", config.getOldType());
        }
        // 自动获取服务名
        String serviceId = config.getServiceId();
        if (!StringUtils.hasLength(serviceId)) {
            serviceId = System.getProperties().getProperty("sun.java.command");
        }
        config.setServiceId(serviceId);
    }

    /**
     * 事件总线订阅者注册
     */
    @Bean
    @ConditionalOnMissingBean(ListenerRegistry.class)
    public ListenerRegistry listenerRegistry(ApplicationContext context, BusProperties busConfig) {
        Map beanMap = context.getBeansWithAnnotation(EventbusListener.class);
        ListenerRegistry registry = new ListenerRegistry(busConfig);
        registry.register(beanMap.values());
        return registry;
    }

    /**
     * 事件总线拦截器配置
     */
    @Bean
    @ConditionalOnMissingBean(InterceptorContainer.class)
    public InterceptorContainer interceptorConfig(
            @Autowired(required = false) List sendBeforeInterceptors,
            @Autowired(required = false) List sendAfterInterceptors,
            @Autowired(required = false) List deliverBeforeInterceptors,
            @Autowired(required = false) List deliverAfterInterceptors,
            @Autowired(required = false) List deliverThrowableLastInterceptors) {
        return new InterceptorContainer(sendBeforeInterceptors, sendAfterInterceptors,deliverBeforeInterceptors, deliverAfterInterceptors, deliverThrowableLastInterceptors);
    }

    /**
     * 事件总线分发器
     */
    @Bean
    @ConditionalOnBean(MsgSender.class)
    @ConditionalOnMissingBean(DeliveryBus.class)
    public DeliveryBus deliveryBus(InterceptorContainer interceptorContainer, BusConfig busConfig, MsgSender msgSender) {
        return new DeliveryBus(interceptorContainer, busConfig, msgSender);
    }

    /**
     * 连接监控
     */
    @Bean
    @ConditionalOnBean(NodeTestConnect.class)
    public ConnectionWatchdog connectionWatchdog(NodeTestConnect nodeTestConnect, BusConfig busConfig, List listeners) {
        return new BootConnectionWatchdog(nodeTestConnect, busConfig.getTestConnect(), listeners);
    }
}