com.tvd12.ezymq.common.EzyMQProxyBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ezymq-common Show documentation
The newest version!
package com.tvd12.ezymq.common;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.tvd12.ezyfox.bean.EzyBeanContext;
import com.tvd12.ezyfox.bean.EzyBeanContextBuilder;
import com.tvd12.ezyfox.binding.EzyBindingContext;
import com.tvd12.ezyfox.binding.EzyBindingContextBuilder;
import com.tvd12.ezyfox.binding.EzyMarshaller;
import com.tvd12.ezyfox.binding.EzyUnmarshaller;
import com.tvd12.ezyfox.binding.codec.EzyBindingEntityCodec;
import com.tvd12.ezyfox.binding.impl.EzySimpleBindingContext;
import com.tvd12.ezyfox.builder.EzyBuilder;
import com.tvd12.ezyfox.codec.*;
import com.tvd12.ezyfox.collect.Sets;
import com.tvd12.ezyfox.entity.EzyData;
import com.tvd12.ezyfox.message.annotation.EzyMessage;
import com.tvd12.ezyfox.message.annotation.Message;
import com.tvd12.ezyfox.reflect.EzyReflection;
import com.tvd12.ezyfox.reflect.EzyReflectionProxy;
import com.tvd12.ezyfox.reflect.EzyTypes;
import com.tvd12.ezyfox.util.EzyPropertiesKeeper;
import com.tvd12.ezymq.common.setting.EzyMQSettings;

import java.util.*;

@SuppressWarnings({"rawtypes", "unchecked"})
public abstract class EzyMQProxyBuilder<
    S extends EzyMQSettings,
    DC,
    P extends EzyMQProxy,
    PB extends EzyMQProxyBuilder
    >
    extends EzyPropertiesKeeper
    implements EzyBuilder {

    protected S settings;
    protected DC dataCodec;
    protected boolean scanAndAddAllBeans;
    protected EzyMarshaller marshaller;
    protected EzyUnmarshaller unmarshaller;
    protected EzyEntityCodec entityCodec;
    protected EzyBeanContext beanContext;
    protected EzyBindingContext bindingContext;
    protected EzyMessageSerializer messageSerializer;
    protected EzyMessageDeserializer messageDeserializer;
    protected EzyMessageDeserializer textMessageDeserializer;
    protected EzyMQSettings.Builder settingsBuilder;
    protected final Set packagesToScan;
    protected final EzyBeanContextBuilder beanContextBuilder;

    public EzyMQProxyBuilder() {
        this.packagesToScan = new HashSet<>();
        this.beanContextBuilder = EzyBeanContext.builder();
    }

    public PB entryClass(Class entryClass) {
        return scan(entryClass.getPackage().getName());
    }

    public PB scan(String packageName) {
        this.packagesToScan.add(packageName);
        return (PB) this;
    }

    public PB scan(String... packageNames) {
        return scan(Arrays.asList(packageNames));
    }

    public PB scan(Iterable packageNames) {
        for (String packageName : packageNames) {
            scan(packageName);
        }
        return (PB) this;
    }

    public EzyMQSettings.Builder settingsBuilder() {
        if (settingsBuilder == null) {
            settingsBuilder = newSettingBuilder();
        }
        return settingsBuilder;
    }

    protected abstract EzyMQSettings.Builder newSettingBuilder();

    public PB settings(S settings) {
        this.settings = settings;
        return (PB) this;
    }

    public PB marshaller(EzyMarshaller marshaller) {
        this.marshaller = marshaller;
        return (PB) this;
    }

    public PB unmarshaller(EzyUnmarshaller unmarshaller) {
        this.unmarshaller = unmarshaller;
        return (PB) this;
    }

    public PB dataCodec(DC dataCodec) {
        this.dataCodec = dataCodec;
        return (PB) this;
    }

    public PB entityCodec(EzyEntityCodec entityCodec) {
        this.entityCodec = entityCodec;
        return (PB) this;
    }

    public PB addSingleton(Object singleton) {
        this.beanContextBuilder.addSingleton(singleton);
        return (PB) this;
    }

    public PB addSingleton(String name, Object singleton) {
        this.beanContextBuilder.addSingleton(name, singleton);
        return (PB) this;
    }

    public PB addSingletons(List singletons) {
        singletons.forEach(beanContextBuilder::addSingleton);
        return (PB) this;
    }

    public PB addSingletons(Map singletons) {
        beanContextBuilder.addSingletons(singletons);
        return (PB) this;
    }

    public PB beanContext(EzyBeanContext beanContext) {
        this.beanContextBuilder.addSingletonsByKey(beanContext.getSingletonMapByKey());
        return (PB) this;
    }

    public PB bindingContext(EzyBindingContext bindingContext) {
        this.bindingContext = bindingContext;
        return (PB) this;
    }

    public PB scanAndAddAllBeans(boolean scanAndAddAllBeans) {
        this.scanAndAddAllBeans = scanAndAddAllBeans;
        return (PB) this;
    }

    public PB messageSerializer(EzyMessageSerializer messageSerializer) {
        this.messageSerializer = messageSerializer;
        return (PB) this;
    }

    public PB messageDeserializer(EzyMessageDeserializer messageDeserializer) {
        this.messageDeserializer = messageDeserializer;
        return (PB) this;
    }

    public PB textMessageDeserializer(EzyMessageDeserializer textMessageDeserializer) {
        this.textMessageDeserializer = textMessageDeserializer;
        return (PB) this;
    }

    @Override
    public P build() {
        beanContext = newBeanContext();
        if (settings == null) {
            if (settingsBuilder == null) {
                settingsBuilder = newSettingBuilder();
            }
            settingsBuilder
                .properties(properties)
                .properties(beanContext.getProperties());
            decorateSettingBuilder(settingsBuilder);
            settings = (S) settingsBuilder.build();
        }
        if (bindingContext == null) {
            bindingContext = newBindingContext();
        }
        marshaller = bindingContext.newMarshaller();
        unmarshaller = bindingContext.newUnmarshaller();

        if (messageSerializer == null) {
            messageSerializer = newMessageSerializer();
        }
        if (messageDeserializer == null) {
            messageDeserializer = newMessageDeserializer();
        }
        if (textMessageDeserializer == null) {
            textMessageDeserializer = newTextMessageDeserializer();
        }
        if (dataCodec == null) {
            dataCodec = newDataCodec();
        }
        if (entityCodec == null) {
            entityCodec = newEntityCodec();
        }
        preNewProxy();
        return newProxy();
    }

    protected abstract void decorateSettingBuilder(
        EzyMQSettings.Builder settingsBuilder
    );

    protected abstract DC newDataCodec();

    protected void preNewProxy() {}

    protected abstract P newProxy();

    protected EzyEntityCodec newEntityCodec() {
        return EzyBindingEntityCodec.builder()
            .marshaller(marshaller)
            .unmarshaller(unmarshaller)
            .messageSerializer(messageSerializer)
            .messageDeserializer(messageDeserializer)
            .build();
    }

    protected EzyMessageSerializer newMessageSerializer() {
        return new MsgPackSimpleSerializer();
    }

    protected EzyMessageDeserializer newMessageDeserializer() {
        return new MsgPackSimpleDeserializer();
    }

    protected EzyMessageDeserializer newTextMessageDeserializer() {
        try {
            return newJacksonSimpleDeserializer();
        } catch (Throwable e) {
            return null;
        }
    }

    protected EzyMessageDeserializer newJacksonSimpleDeserializer() {
        return new JacksonSimpleDeserializer(new ObjectMapper());
    }

    private EzyBeanContext newBeanContext() {
        if (packagesToScan.size() > 0) {
            EzyReflection reflection = new EzyReflectionProxy(packagesToScan);
            for (Class beanAnnotationClass : getBeanAnnotationClasses()) {
                beanContextBuilder.addSingletonClasses(
                    reflection.getAnnotatedClasses(beanAnnotationClass)
                );
            }
            if (scanAndAddAllBeans) {
                beanContextBuilder.scan(packagesToScan);
            }
        }
        return beanContextBuilder.build();
    }

    protected Set> getBeanAnnotationClasses() {
        return Sets.newHashSet(
            getRequestInterceptorAnnotationClass(),
            getRequestHandlerAnnotationClass()
        );
    }

    public abstract Class getRequestInterceptorAnnotationClass();

    public abstract Class getRequestHandlerAnnotationClass();

    private EzyBindingContext newBindingContext() {
        EzyBindingContextBuilder builder = EzySimpleBindingContext.builder();
        for (Class messageType : settings.getMessageTypes()) {
            if (!EzyTypes.ALL_TYPES.contains(messageType)
                && !EzyData.class.isAssignableFrom(messageType)
            ) {
                builder.addClasses(messageType);
            }
        }
        if (packagesToScan.size() > 0) {
            EzyReflection reflection = new EzyReflectionProxy(packagesToScan);
            builder
                .addAllClasses(reflection)
                .addClasses((Set) reflection.getAnnotatedClasses(Message.class))
                .addClasses((Set) reflection.getAnnotatedClasses(EzyMessage.class));
        }
        return builder.build();
    }
}