com.farao_community.farao.rao_runner.starter.RaoRunnerClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gridcapa-rao-runner-spring-boot-starter Show documentation
Starter to be used in Spring application to use RAO runner service
The newest version!
/*
 * Copyright (c) 2021, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package com.farao_community.farao.rao_runner.starter;

import com.farao_community.farao.rao_runner.api.JsonApiConverter;
import com.farao_community.farao.rao_runner.api.resource.RaoFailureResponse;
import com.farao_community.farao.rao_runner.api.resource.RaoRequest;
import com.farao_community.farao.rao_runner.api.resource.AbstractRaoResponse;
import org.springframework.amqp.core.AmqpTemplate;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageBuilder;
import org.springframework.amqp.core.MessageDeliveryMode;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.core.MessagePropertiesBuilder;

/**
 * @author Mohamed BenRejeb {@literal }
 */
public class RaoRunnerClient {

    private static final String CONTENT_ENCODING = "UTF-8";
    private static final String CONTENT_TYPE = "application/vnd.api+json";
    private static final int DEFAULT_PRIORITY = 1;

    private final AmqpTemplate amqpTemplate;
    private final RaoRunnerClientProperties raoRunnerClientProperties;
    private final JsonApiConverter jsonConverter;

    public RaoRunnerClient(AmqpTemplate amqpTemplate, RaoRunnerClientProperties raoRunnerClientProperties) {
        this.amqpTemplate = amqpTemplate;
        this.raoRunnerClientProperties = raoRunnerClientProperties;
        this.jsonConverter = new JsonApiConverter();
    }

    public AbstractRaoResponse runRao(final RaoRequest raoRequest, final int priority) {
        final Message responseMessage = amqpTemplate.sendAndReceive(raoRunnerClientProperties.getAmqp().getQueueName(), buildMessage(raoRequest, priority));
        if (responseMessage != null) {
            return RaoResponseConversionHelper.convertRaoResponse(responseMessage, jsonConverter);
        } else {
            return new RaoFailureResponse.Builder()
                    .withId(raoRequest.getId())
                    .withErrorMessage("Rao Runner server did not respond")
                    .build();
        }
    }

    public AbstractRaoResponse runRao(final RaoRequest raoRequest) {
        return runRao(raoRequest, DEFAULT_PRIORITY);
    }

    private Message buildMessage(final RaoRequest raoRequest, final int priority) {
        return MessageBuilder.withBody(jsonConverter.toJsonMessage(raoRequest))
                .andProperties(buildMessageProperties(priority))
                .build();
    }

    private MessageProperties buildMessageProperties(final int priority) {
        return MessagePropertiesBuilder.newInstance()
                .setAppId(raoRunnerClientProperties.getAmqp().getClientAppId())
                .setContentEncoding(CONTENT_ENCODING)
                .setContentType(CONTENT_TYPE)
                .setDeliveryMode(MessageDeliveryMode.NON_PERSISTENT)
                .setExpiration(raoRunnerClientProperties.getAmqp().getExpiration())
                .setPriority(priority)
                .build();
    }
}