com.gnss.mqutil.converter.ProtobufMessageConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gnss-mqutil Show documentation

There is a newer version: 1.0.10

package com.gnss.mqutil.converter;

import com.baidu.bjf.remoting.protobuf.Codec;
import com.baidu.bjf.remoting.protobuf.annotation.ProtobufClass;
import lombok.extern.slf4j.Slf4j;
import org.reflections.Reflections;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.support.converter.MessageConversionException;
import org.springframework.amqp.support.converter.MessageConverter;

import java.util.HashMap;
import java.util.Map;

/**
 * Description: Protobuf转换器
 * Company: www.gps-pro.cn
 *
 * @author huangguangbin
 * @version 1.0.1
 * @date 2018/4/13
 */
@Slf4j
public class ProtobufMessageConverter implements MessageConverter {
    private static Map codecMap = new HashMap<>();

    static {
        Reflections reflections = new Reflections("com.gnss.common.proto");
        reflections.getTypesAnnotatedWith(ProtobufClass.class).stream()
                .forEach(clazz -> {
                    try {
                        Codec codec = (Codec) Class.forName(clazz.getName() + "$$JProtoBufClass").newInstance();
                        codecMap.put(clazz.getSimpleName(), codec);
                        log.info("加载Protobuf转换器{}", clazz.getName());
                    } catch (Exception e) {
                        log.error("加载Protobuf转换器{}失败", clazz.getName(), e);
                    }
                });
    }

    @Override
    public Message toMessage(Object obj, MessageProperties messageProperties) throws MessageConversionException {
        String messageType = obj.getClass().getSimpleName();
        Codec