org.khasanof.collector.loader.DefaultSpringHandlerLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-starter-fluent Show documentation
Fluent - Easy Telegram Bots with Spring
The newest version!
package org.khasanof.collector.loader;

import org.khasanof.annotation.ExceptionController;
import org.khasanof.annotation.StateController;
import org.khasanof.annotation.UpdateController;
import org.khasanof.config.Config;
import org.khasanof.config.FluentProperties;
import org.khasanof.enums.ClassLevelTypes;
import org.khasanof.enums.ProcessType;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import java.lang.annotation.Annotation;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Author: Nurislom
 * 

 * Date: 18.06.2023
 * 

 * Time: 11:32
 * 

 * Package: org.khasanof.core.collector
 */
@Component
public class DefaultSpringHandlerLoader implements Config, HandlerLoader {

    private final FluentProperties.Bot bot = new FluentProperties.Bot();
    private final ApplicationContext applicationContext;
    private final Map beanMap = new ConcurrentHashMap<>();
    private final Set> classLevelAnnotations = new HashSet<>();

    public DefaultSpringHandlerLoader(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }

    @Override
    public Map getHandlers() {
        return this.beanMap;
    }

    @Override
    public  Map getHandlersOfType(Class type) {
        return applicationContext.getBeansOfType(type);
    }

    @Override
    public Map getHandlers(Class annotation) {
        return applicationContext.getBeansWithAnnotation(annotation);
    }

    @Override
    public void runnable() {
        ProcessType processType = bot.getProcessType();
        addAnnotationWithProcessType(processType);
        fillBeanMap();
    }

    private void addAnnotationWithProcessType(ProcessType processType) {
        if (processType.equals(ProcessType.BOTH)) {
            this.classLevelAnnotations.addAll(ClassLevelTypes.getAllAnnotations());
        } else {
            internalAddAnnotationWithProcessType(processType);
        }
    }

    private void internalAddAnnotationWithProcessType(ProcessType processType) {
        if (processType.equals(ProcessType.STATE)) {
            this.classLevelAnnotations.add(StateController.class);
        }
        this.classLevelAnnotations.add(UpdateController.class);
        this.classLevelAnnotations.add(ExceptionController.class);
    }

    private void fillBeanMap() {
        for (var classLevelAnnotation : classLevelAnnotations) {
            var validBeansTake = applicationContext.getBeansWithAnnotation(classLevelAnnotation);
            if (!validBeansTake.isEmpty()) {
                this.beanMap.putAll(validBeansTake);
            }
        }
    }

    @Override
    public ProcessType processType() {
        return ProcessType.BOTH;
    }
}