ThriftSecureEventTransmissionServiceImpl.java

/*
*  Copyright (c) 2005-2010, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
*
*  WSO2 Inc. licenses this file to you 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 org.wso2.carbon.databridge.receiver.thrift.service;

import org.apache.thrift.TException;
import org.wso2.carbon.databridge.commons.exception.AuthenticationException;
import org.wso2.carbon.databridge.commons.exception.DifferentStreamDefinitionAlreadyDefinedException;
import org.wso2.carbon.databridge.commons.exception.MalformedStreamDefinitionException;
import org.wso2.carbon.databridge.commons.exception.SessionTimeoutException;
import org.wso2.carbon.databridge.commons.exception.UndefinedEventTypeException;
import org.wso2.carbon.databridge.commons.thrift.data.ThriftEventBundle;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftAuthenticationException;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftDifferentStreamDefinitionAlreadyDefinedException;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftMalformedStreamDefinitionException;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftNoStreamDefinitionExistException;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftSessionExpiredException;
import org.wso2.carbon.databridge.commons.thrift.exception.ThriftUndefinedEventTypeException;
import org.wso2.carbon.databridge.commons.thrift.service.secure.ThriftSecureEventTransmissionService;
import org.wso2.carbon.databridge.core.DataBridgeReceiverService;
import org.wso2.carbon.databridge.core.EventConverter;
import org.wso2.carbon.databridge.receiver.thrift.converter.ThriftEventConverter;

/**
 * The client implementation for ThriftSecureEventTransmissionService
 */
public class ThriftSecureEventTransmissionServiceImpl
        implements ThriftSecureEventTransmissionService.Iface {

    private DataBridgeReceiverService dataBridgeReceiverService;
    private EventConverter eventConverter = new ThriftEventConverter();

    public ThriftSecureEventTransmissionServiceImpl(
            DataBridgeReceiverService dataBridgeReceiverService) {
        this.dataBridgeReceiverService = dataBridgeReceiverService;
    }

    public String connect(String username, String password) throws ThriftAuthenticationException {
        try {
            return dataBridgeReceiverService.login(username, password);
        } catch (AuthenticationException e) {
            throw new ThriftAuthenticationException(e.getErrorMessage());
        }
    }

    public void disconnect(String sessionId) throws TException {
        try {
            dataBridgeReceiverService.logout(sessionId);
        } catch (Exception e) {
            throw new TException(e.getMessage());
        }
    }

    @Override
    public String defineStream(String sessionId, String streamDefinition)
            throws TException, ThriftSessionExpiredException,
                   ThriftDifferentStreamDefinitionAlreadyDefinedException,
                   ThriftMalformedStreamDefinitionException {
        try {
            return dataBridgeReceiverService.defineStream(sessionId, streamDefinition);
        } catch (DifferentStreamDefinitionAlreadyDefinedException e) {
            throw new ThriftDifferentStreamDefinitionAlreadyDefinedException(e.getErrorMessage());
        } catch (MalformedStreamDefinitionException e) {
            throw new ThriftMalformedStreamDefinitionException(e.getErrorMessage());
        } catch (SessionTimeoutException e) {
            throw new ThriftSessionExpiredException(e.getErrorMessage());
        }
    }

    @Override
    public String findStreamId(String sessionId, String streamName, String streamVersion)
            throws ThriftNoStreamDefinitionExistException, ThriftSessionExpiredException,
                   TException {
        try {
            String streamDefinition = dataBridgeReceiverService.findStreamId(sessionId, streamName, streamVersion);
            if (streamDefinition == null) {
                //this is used as Thrift cannot send null values
                throw new ThriftNoStreamDefinitionExistException("Stream definition not exist for " + streamName + " " + streamVersion);
            }
            return streamDefinition;
        } catch (SessionTimeoutException e) {
            throw new ThriftSessionExpiredException(e.getErrorMessage());
        }
    }

    @Override
    public boolean deleteStreamById(String sessionId, String streamId)
            throws ThriftSessionExpiredException,
                   TException {
        try {
            return dataBridgeReceiverService.deleteStream(sessionId, sessionId);
        } catch (SessionTimeoutException e) {
            throw new ThriftSessionExpiredException(e.getErrorMessage());
        }
    }

    @Override
    public boolean deleteStreamByNameVersion(String sessionId, String streamName,
                                             String streamVersion)
            throws ThriftSessionExpiredException,
                   TException {
        try {
            return dataBridgeReceiverService.deleteStream(sessionId, streamName, streamVersion);
        } catch (SessionTimeoutException e) {
            throw new ThriftSessionExpiredException(e.getErrorMessage());
        }
    }

    public void publish(ThriftEventBundle eventBundle)
            throws ThriftUndefinedEventTypeException, ThriftSessionExpiredException, TException {
        try {
            dataBridgeReceiverService.publish(eventBundle, eventBundle.getSessionId(), eventConverter);
        } catch (UndefinedEventTypeException e) {
            throw new ThriftUndefinedEventTypeException(e.getErrorMessage());
        } catch (SessionTimeoutException e) {
            throw new ThriftSessionExpiredException(e.getErrorMessage());
        }
    }
}