com.allogy.amazonaws.elasticbeanstalk.worker.simulator.application.WorkerApplication.java Source code

Introduction

Here is the source code for com.allogy.amazonaws.elasticbeanstalk.worker.simulator.application.WorkerApplication.java

Source

/*
 * Copyright (c) 2015 Allogy Interactive.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package com.allogy.amazonaws.elasticbeanstalk.worker.simulator.application;

import com.allogy.amazonaws.elasticbeanstalk.worker.simulator.model.MessageWrapper;
import com.amazonaws.services.sqs.model.Message;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.http.*;
import org.springframework.stereotype.Service;
import org.springframework.web.client.HttpStatusCodeException;
import org.springframework.web.client.RestClientException;
import org.springframework.web.client.RestTemplate;

/**
 * This service provides the interaction of forwarding a message to the
 * application web service. Thus, it makes the HTTP call that the daemon
 * would make.
 *
 * @author David Venable
 */
@Service
public class WorkerApplication {
    private final String targetUrl;
    private final RestTemplate restTemplate;
    private final Logger logger;

    @Autowired
    public WorkerApplication(@Qualifier("targetUrl") String targetUrl, RestTemplate restTemplate) {
        this.targetUrl = targetUrl;
        this.restTemplate = restTemplate;
        logger = LoggerFactory.getLogger(this.getClass());
    }

    public HttpStatus forward(MessageWrapper messageWrapper) {
        Message message = messageWrapper.getMessage();
        HttpEntity<String> messageEntity = createHttpEntity(messageWrapper, message);

        logger.info("Forwarding message: messageId={}", messageWrapper.getMessage().getMessageId());

        try {
            ResponseEntity<String> responseEntity = restTemplate.exchange(targetUrl, HttpMethod.POST, messageEntity,
                    String.class);
            return responseEntity.getStatusCode();
        } catch (HttpStatusCodeException ex) {
            HttpStatus statusCode = ex.getStatusCode();
            logger.error("The HTTP Worker Application returned a non successful error code. statusCode={}",
                    statusCode);
            return statusCode;
        } catch (RestClientException ex) {
            logger.error("Unable to connect to the HTTP Worker Application.");
            return HttpStatus.NOT_FOUND;
        }
    }

    private HttpEntity<String> createHttpEntity(MessageWrapper messageWrapper, Message message) {
        String messageId = message.getMessageId();

        HttpHeaders headers = new HttpHeaders();
        headers.set("User-Agent", "aws-sqsd/1.1 (simulated bridge)");
        headers.set("X-Aws-Sqsd-Msgid", messageId);
        headers.set("X-Aws-Sqsd-Queue", messageWrapper.getQueueName());
        headers.set("X-Aws-Sqsd-Receive-Count", Integer.toString(messageWrapper.getMessageCount()));
        headers.setContentType(MediaType.APPLICATION_JSON);

        return new HttpEntity<>(message.getBody(), headers);
    }
}