org.reactivecommons.async.impl.RabbitDirectAsyncGateway Maven / Gradle / Ivy

Go to download
package org.reactivecommons.async.impl;

import org.reactivecommons.api.domain.Command;
import org.reactivecommons.async.api.AsyncQuery;
import org.reactivecommons.async.api.DirectAsyncGateway;
import org.reactivecommons.async.impl.communications.ReactiveMessageSender;
import org.reactivecommons.async.impl.config.BrokerConfig;
import org.reactivecommons.async.impl.converters.MessageConverter;
import org.reactivecommons.async.impl.reply.ReactiveReplyRouter;
import reactor.core.publisher.Mono;

import java.time.Duration;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

import static org.reactivecommons.async.impl.Headers.*;
import static reactor.core.publisher.Mono.fromCallable;

public class RabbitDirectAsyncGateway implements DirectAsyncGateway {

    private final BrokerConfig config;
    private final ReactiveReplyRouter router;
    private final ReactiveMessageSender sender;
    private final String exchange;
    private final MessageConverter converter;


    public RabbitDirectAsyncGateway(BrokerConfig config, ReactiveReplyRouter router, ReactiveMessageSender sender, String exchange, MessageConverter converter) {
        this.config = config;
        this.router = router;
        this.sender = sender;
        this.exchange = exchange;
        this.converter = converter;
    }

    @Override
    public  Mono sendCommand(Command command, String targetName) {
        return sender.sendWithConfirm(command, exchange, targetName, Collections.emptyMap());
    }

    @Override
    public  Mono requestReply(AsyncQuery query, String targetName, Class type) {
        final String correlationID = UUID.randomUUID().toString().replaceAll("-", "");

        final Mono replyHolder = router.register(correlationID)
            .timeout(Duration.ofSeconds(15))
            .flatMap(s -> fromCallable(() -> (R)converter.readValue(s, type)));

        Map headers = new HashMap<>();
        headers.put(REPLY_ID, config.getRoutingKey());
        headers.put(SERVED_QUERY_ID, query.getResource());
        headers.put(CORRELATION_ID, correlationID);

        return sender.sendWithConfirm(query, exchange, targetName + ".query", headers).then(replyHolder);
    }

}