com.eucalyptus.ws.IoMessage.java Source code

Introduction

Here is the source code for com.eucalyptus.ws.IoMessage.java

Source

/*************************************************************************
 * (c) Copyright 2016 Hewlett Packard Enterprise Development Company LP
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 3 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see http://www.gnu.org/licenses/.
 ************************************************************************/
package com.eucalyptus.ws;

import java.net.URI;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.soap.SOAPEnvelope;
import com.eucalyptus.util.Parameters;
import com.google.common.base.MoreObjects;
import edu.ucsb.eucalyptus.msgs.BaseMessage;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.FullHttpMessage;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpRequest;
import io.netty.handler.codec.http.HttpVersion;

/**
 *
 */
public class IoMessage<HT extends FullHttpMessage> {

    private String correlationId;
    private SOAPEnvelope soapEnvelope;
    private OMElement omMessage;
    private Object message;
    private HT httpMessage;

    private IoMessage() {
    }

    /**
     * Create a new message for outbound http use.
     *
     * @param endpoint The http endpoint
     * @param message The message to be sent
     * @return The new message
     */
    @Nonnull
    public static IoMessage<FullHttpRequest> httpRequest(@Nonnull final URI endpoint,
            @Nonnull final BaseMessage message) {
        Parameters.checkParamNotNull("endpoint", endpoint);
        Parameters.checkParamNotNull("message", message);
        final FullHttpRequest httpRequest = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.POST,
                endpoint.getPath());
        httpRequest.headers().add(HttpHeaders.Names.HOST, endpoint.getHost() + ":" + endpoint.getPort());
        final IoMessage<FullHttpRequest> ioMessage = new IoMessage<>();
        ioMessage.setHttpMessage(httpRequest);
        ioMessage.setMessage(message);
        ioMessage.setCorrelationId(message.getCorrelationId());
        return ioMessage;
    }

    /**
     * Create a new message from an existing http message.
     *
     * @param httpMessage The http message to use
     * @param <HT> The type of the http message
     * @return The new message
     */
    @Nonnull
    public static <HT extends FullHttpMessage> IoMessage<HT> http(@Nonnull HT httpMessage) {
        final IoMessage<HT> ioMessage = new IoMessage<>();
        ioMessage.setHttpMessage(httpMessage);
        return ioMessage;
    }

    @Nonnull
    public String getCorrelationId() {
        return correlationId;
    }

    public void setCorrelationId(@Nonnull final String correlationId) {
        this.correlationId = Parameters.checkParamNotNull("correlationId", correlationId);
    }

    @Nullable
    public Object getMessage() {
        return message;
    }

    public void setMessage(final Object message) {
        this.message = message;
    }

    @Nonnull
    public HT getHttpMessage() {
        return httpMessage;
    }

    public void setHttpMessage(@Nonnull final HT httpMessage) {
        this.httpMessage = Parameters.checkParamNotNull("httpMessage", httpMessage);
    }

    @Nullable
    public SOAPEnvelope getSoapEnvelope() {
        return soapEnvelope;
    }

    public void setSoapEnvelope(final SOAPEnvelope soapEnvelope) {
        this.soapEnvelope = soapEnvelope;
    }

    @Nullable
    public OMElement getOmMessage() {
        return omMessage;
    }

    public void setOmMessage(final OMElement omMessage) {
        this.omMessage = omMessage;
    }

    public boolean isRequest() {
        return getHttpMessage() instanceof HttpRequest;
    }

    @Override
    public String toString() {
        return MoreObjects.toStringHelper(this).add("correlationId", getCorrelationId()).toString();
    }
}