com.streamreduce.core.service.OutboundStorageServiceImpl.java Source code

Introduction

Here is the source code for com.streamreduce.core.service.OutboundStorageServiceImpl.java

Source

/*
 * Copyright 2012 Nodeable Inc
 *
 *    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.streamreduce.core.service;

import com.streamreduce.ConnectionNotFoundException;
import com.streamreduce.OutboundStorageException;
import com.streamreduce.client.outbound.OutboundClientFactory;
import com.streamreduce.core.model.Connection;
import com.streamreduce.core.model.OutboundConfiguration;
import com.streamreduce.core.model.OutboundDataType;
import com.streamreduce.core.model.dto.OutboundConfigurationWithPayloadDTO;
import com.streamreduce.core.model.messages.MessageType;
import com.streamreduce.core.model.messages.SobaMessage;
import com.streamreduce.rest.dto.response.SobaMessageResponseDTO;

import java.io.IOException;

import net.sf.json.JSONObject;
import org.apache.camel.CamelContext;
import org.apache.camel.CamelContextAware;
import org.apache.camel.ProducerTemplate;
import org.apache.commons.collections.MapUtils;
import org.codehaus.jackson.map.ObjectMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

@Service
public class OutboundStorageServiceImpl extends AbstractService
        implements OutboundStorageService, CamelContextAware {

    @Autowired
    OutboundClientFactory outboundClientFactory;
    @Autowired
    ConnectionService connectionService;

    ProducerTemplate outboundStorageMessageProducer;

    @Override
    public int sendRawMessage(JSONObject jsonObject, Connection connection) throws OutboundStorageException {
        if (MapUtils.isEmpty(jsonObject) || CollectionUtils.isEmpty(connection.getOutboundConfigurations())) {
            return 0;
        }
        int numberOfMessagesRoutedOutbound = 0;
        for (OutboundConfiguration outboundConfiguration : connection.getOutboundConfigurations()) {
            if (outboundConfiguration.getDataTypes().contains(OutboundDataType.RAW)) {
                try {
                    routePayloadOutbound(outboundConfiguration, jsonObject.toString(), OutboundDataType.RAW);
                    numberOfMessagesRoutedOutbound++;
                } catch (Exception e) {
                    logger.error("Unable to route message outbound", e);
                }
            }
        }
        return numberOfMessagesRoutedOutbound;
    }

    @Override
    public int sendSobaMessage(SobaMessage sobaMessage) throws OutboundStorageException {
        try {
            Connection c = connectionService.getConnection(sobaMessage.getConnectionId());
            return sendSobaMessage(sobaMessage, c);
        } catch (ConnectionNotFoundException e) {
            throw new OutboundStorageException(
                    "Unable to retrieve Connection for sobaMessage with Id of " + sobaMessage.getId(), e);
        }
    }

    @Override
    public int sendSobaMessage(SobaMessage sobaMessage, Connection connection) throws OutboundStorageException {
        OutboundDataType dataType = MessageType.NODEBELLY.equals(sobaMessage.getType()) ? OutboundDataType.INSIGHT
                : OutboundDataType.PROCESSED;
        return sendSobaMessageOutbound(sobaMessage, connection, dataType);
    }

    int sendSobaMessageOutbound(SobaMessage sobaMessage, Connection c, OutboundDataType outboundDataType)
            throws OutboundStorageException {

        int numberOfMessagesRoutedOutbound = 0;
        for (OutboundConfiguration outboundConfiguration : c.getOutboundConfigurations()) {
            if (outboundConfiguration.getDataTypes().contains(outboundDataType)) {
                try {
                    String sobaMessageJSONString = SobaMessageResponseDTO.fromSobaMessage(sobaMessage, true)
                            .toString();
                    routePayloadOutbound(outboundConfiguration, sobaMessageJSONString, outboundDataType);
                    numberOfMessagesRoutedOutbound++;
                } catch (Exception e) {
                    logger.error("Unable to route message outbound", e);
                }
            }
        }
        return numberOfMessagesRoutedOutbound;
    }

    void routePayloadOutbound(OutboundConfiguration outboundConfiguration, String payload,
            OutboundDataType dataTypeForPayload) throws IOException {
        OutboundConfigurationWithPayloadDTO dto = new OutboundConfigurationWithPayloadDTO(outboundConfiguration,
                payload, dataTypeForPayload);
        String dtoAsJsonString = new ObjectMapper().writeValueAsString(dto);
        outboundStorageMessageProducer.sendBody(dtoAsJsonString);

    }

    @Override
    public void setCamelContext(CamelContext camelContext) {
        ProducerTemplate template = camelContext.createProducerTemplate();
        template.setDefaultEndpoint(camelContext.getEndpoint("direct:outbound-messages"));
        outboundStorageMessageProducer = template;
    }

    @Override
    @Deprecated
    /**
     * Do not call this method.  This CamelContextAware does not expose the CamelContext to other classes.  It is
     * CamelContextAware only to have the context injected in to it.
     */
    public CamelContext getCamelContext() {
        return null;
    }
}