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

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

import com.github.xiaobingzhou.messageframe.codec.BodyCodec;
import com.github.xiaobingzhou.messageframe.mapper.MapperField;
import com.github.xiaobingzhou.messageframe.repository.BodyCodecRepository;
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.context.ApplicationContextException;

import java.util.*;

/**
 * bodyCodec解码器的仓库
 * @author bell.zhouxiaobing
 * @since 1.5.4
 */
public class BodyCodecRepositoryImpl implements BodyCodecRepository, ApplicationContextAware, DisposableBean {

    @Getter
    private ApplicationContext applicationContext;

    /**
     * 保存指令码和bodyCodec解码器的对应关系
     * body_codec_map
     */
    private static Map BODY_CODEC_MAP = new HashMap<>();

    @Override
    public BodyCodec getBodyCodec(String commandCode) {
        String beanName = BODY_CODEC_MAP.get(commandCode);
        return beanName != null ? applicationContext.getBean(beanName, BodyCodec.class) : null;
    }

    @Override
    public void setBodyCodec(BodyCodec bodyCodec, String beanName) {
        // 检查解码器
        check(bodyCodec, beanName);
        List commandCodes = bodyCodec.getCommandCodes();
        for (String commandCode : commandCodes) {
            String exist = BODY_CODEC_MAP.put(commandCode, beanName);
            if (exist != null) {
                throw new BeanCreationException(String.format("指令码:[%s] 匹配到两个BodyCodec解码器:[%s] 和 [%s]",
                                commandCode, beanName, exist));
            }
        }
    }

    protected void check(BodyCodec bodyCodec, String beanName) {
        List mapperFields = bodyCodec.getMapperFields();
        List existNames = new ArrayList<>();
        for (MapperField mapperField : mapperFields) {
            String name = mapperField.getName();
            // 检查length
            if (mapperField.getLength() % 2 != 0) {
                throw new BeanCreationException(
                        String.format("BodyCodec解码器:[%s], 字段:[%s], 长度:[%s] 是奇数",
                                beanName, name, mapperField.getLength()));
            }
            // 检查name是否相同
            if (existNames.contains(name)) {
                throw new BeanCreationException(
                        String.format("BodyCodec解码器:[%s], 字段:[%s] 有重复",
                                beanName, name));
            }
            existNames.add(name);
        }
    }

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

    @Override
    public void destroy() throws Exception {
        BODY_CODEC_MAP.clear();
    }

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