com.github.xiaobingzhou.messageframe.repository.impl.HandlerRepositoryImpl Maven / Gradle / Ivy

Go to download
package com.github.xiaobingzhou.messageframe.repository.impl;

import com.github.xiaobingzhou.messageframe.annotation.CommandCode;
import com.github.xiaobingzhou.messageframe.repository.HandlerRepository;
import com.github.xiaobingzhou.messageframe.repository.Store;
import lombok.Getter;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanCreationException;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.LocalVariableTableParameterNameDiscoverer;
import org.springframework.core.ParameterNameDiscoverer;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 指令码处理器仓库
 * @author bell.zhouxiaobing
 * @since 1.3
 */
public class HandlerRepositoryImpl implements HandlerRepository, ApplicationContextAware, DisposableBean {

    @Getter
    private ApplicationContext applicationContext;

    /**
     * 保存指令码和spring中的HandlerStore的对应关系
     * command_code_match__handler_store_map
     */
    private static Map COMMAND_CODE_MATCH_HANDLER_STORE_MAP = new HashMap<>();

    @Override
    public void setHandler(Object handler, String beanName) {
        storeCommandCodeWithMap(handler, beanName);
    }

    protected void storeCommandCodeWithMap(Object handler, String beanName) {
        Method[] declaredMethods = handler.getClass().getDeclaredMethods();
        for (Method method : declaredMethods) {
            CommandCode annotation = method.getAnnotation(CommandCode.class);
            if (annotation != null) {
                String[] parameterNames = getParameterNames(method);
                String[] value = annotation.value();
                for (String commandCode : value) {
                    HandlerStore handlerStore = new HandlerStore();
                    handlerStore.setBeanName(beanName);
                    handlerStore.setMethod(method);
                    handlerStore.setParameterNames(parameterNames);
                    Store put = COMMAND_CODE_MATCH_HANDLER_STORE_MAP.put(commandCode, handlerStore);
                    if (put != null) {
                        String methodName = method.getDeclaringClass().getName()+"."+method.getName();
                        String OtherMethodName = put.getMethod().getDeclaringClass().getName()+"."+put.getMethod().getName();
                        throw new BeanCreationException(beanName, String.format("%s()和%s()指令码重复，指令码是：%s", methodName, OtherMethodName, commandCode));
                    }
                }
            }
        }
    }

    protected String[] getParameterNames(Method method) {
        ParameterNameDiscoverer parameterNameDiscoverer = new LocalVariableTableParameterNameDiscoverer();
        return parameterNameDiscoverer.getParameterNames(method);
    }

    @Override
    public Object getHandler(String commandCode) {
        Store store = getStoreByCommandCode(commandCode);
        if (store == null) return null;
        String beanName = store.getBeanName();
        return beanName == null ? null : applicationContext.getBean(beanName);
    }
    @Override
    public Method getHandlerMethod(String commandCode) {
        Store store = getStoreByCommandCode(commandCode);
        if (store == null) return null;
        return store.getMethod();
    }

    @Override
    public String[] getHandlerMethodParameterNames(String commandCode) {
        Store store = getStoreByCommandCode(commandCode);
        if (store == null) return null;
        return store.getParameterNames();
    }

    private Store getStoreByCommandCode(String commandCode) {
        if (isEmpty(commandCode)) {
            return null;
        }
        return getStore(commandCode);
    }

    @Override
    public Set getCommandCodes() {
        return COMMAND_CODE_MATCH_HANDLER_STORE_MAP.keySet();
    }

    /**
     * @param commandCode
     * @return
     */
    private Store getStore(String commandCode) {
        return COMMAND_CODE_MATCH_HANDLER_STORE_MAP.get(commandCode);
    }

    private boolean isEmpty(String commandCode) {
        return commandCode == null || "".equals(commandCode);
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    @Override
    public void destroy() throws Exception {
        this.applicationContext = null;
        COMMAND_CODE_MATCH_HANDLER_STORE_MAP.clear();
    }

}