com.tvd12.ezymq.activemq.endpoint.EzyActiveRpcServer Maven / Gradle / Ivy

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

import com.tvd12.ezyfox.util.EzyStartable;
import com.tvd12.ezymq.activemq.handler.EzyActiveRpcCallHandler;
import com.tvd12.ezymq.activemq.util.EzyActiveProperties;
import lombok.Setter;

import javax.jms.*;
import java.util.concurrent.atomic.AtomicBoolean;

public class EzyActiveRpcServer
    extends EzyActiveRpcEndpoint
    implements EzyStartable {

    @Setter
    protected EzyActiveRpcCallHandler callHandler;
    protected final AtomicBoolean started = new AtomicBoolean();

    public EzyActiveRpcServer(
        Session session,
        Destination requestQueue,
        Destination replyQueue,
        int threadPoolSize
    ) throws Exception {
        super(session, requestQueue, replyQueue, threadPoolSize);
    }

    public static Builder builder() {
        return new Builder();
    }

    @Override
    protected MessageProducer createProducer() throws Exception {
        return session.createProducer(replyQueue);
    }

    @Override
    protected MessageConsumer createConsumer() throws Exception {
        return session.createConsumer(requestQueue);
    }

    @Override
    public void start() {
        if (started.compareAndSet(false, true)) {
            this.active = true;
            this.executorService.execute();
        } else {
            throw new java.lang.IllegalStateException("server's already started");
        }
    }

    @Override
    protected void handleLoopOne() {
        BytesMessage message = null;
        try {
            message = (BytesMessage) consumer.receive();
            if (message != null) {
                processRequest(message);
            }
        } catch (Exception e) {
            logger.warn("handle message: {} error", message, e);
        }
    }

    public void processRequest(BytesMessage message) throws Exception {
        EzyActiveProperties requestProperties = getMessageProperties(message);
        byte[] requestBody = getMessageBody(message);
        String correlationId = requestProperties.getCorrelationId();
        if (correlationId != null) {
            EzyActiveProperties.Builder replyPropertiesBuilder = new EzyActiveProperties.Builder();
            byte[] replyBody = handleCall(requestProperties, requestBody, replyPropertiesBuilder);
            replyPropertiesBuilder.correlationId(correlationId);
            EzyActiveProperties replyProperties = replyPropertiesBuilder.build();
            publish(replyProperties, replyBody);
        } else {
            handleFire(requestProperties, requestBody);
        }
    }

    protected void handleFire(
        EzyActiveProperties requestProperties,
        byte[] requestBody
    ) {
        callHandler.handleFire(requestProperties, requestBody);
    }

    protected byte[] handleCall(
        EzyActiveProperties requestProperties,
        byte[] requestBody,
        EzyActiveProperties.Builder replyPropertiesBuilder
    ) {
        return callHandler.handleCall(
            requestProperties,
            requestBody,
            replyPropertiesBuilder
        );
    }

    @Override
    protected String getThreadName() {
        return "rpc-server";
    }

    public static class Builder extends EzyActiveRpcEndpoint.Builder {

        @Override
        public EzyActiveRpcServer build() {
            return (EzyActiveRpcServer) super.build();
        }

        @Override
        protected EzyActiveRpcEndpoint newProduct() throws Exception {
            return new EzyActiveRpcServer(
                session,
                requestQueue,
                replyQueue,
                threadPoolSize
            );
        }
    }
}