com.payneteasy.grpc.longpolling.client.http.HttpClientUploading Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grpc-long-polling-client Show documentation
There is a newer version: 1.0-9
Show newest version
package com.payneteasy.grpc.longpolling.client.http;

import com.payneteasy.grpc.longpolling.client.util.ServerEndPoint;
import com.payneteasy.grpc.longpolling.common.MethodDirection;
import com.payneteasy.grpc.longpolling.common.Streams;
import com.payneteasy.tlv.HexUtil;
import io.grpc.Metadata;
import io.grpc.Status;
import io.grpc.internal.ClientStreamListener;
import io.grpc.internal.IoUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.net.ConnectException;
import java.net.HttpURLConnection;
import java.net.URL;

public class HttpClientUploading implements IHttpClient {

    private static final Logger LOG = LoggerFactory.getLogger(HttpClientUploading.class);

    private final ClientStreamListener listener;
    private final URL                  sendUrl;
    private final Streams              streams = new Streams(LOG);

    public HttpClientUploading(ServerEndPoint aEndpoint, ClientStreamListener aListener) {
        sendUrl  = aEndpoint.createUrl(MethodDirection.UP);
        listener = aListener;
    }

    @Override
    public void sendMessage(InputStream aInputStream) {
        try {
            LOG.debug("Sending to {} ...", sendUrl);

            HttpURLConnection connection = createBidirectionalConnection();
            streams.sendMessage(aInputStream, connection);
            if (hasError(connection)) {
                return;
            }
            readOutputFromServer(connection);

        } catch (FileNotFoundException e) {
            fireError(Status.NOT_FOUND, e, "Not found");
        } catch (ConnectException e) {
            fireError(Status.UNAVAILABLE, e, "Cannot connect");
        } catch (IOException e) {
            fireError(Status.DATA_LOSS, e, "IO error");
        }
    }

    private boolean hasError(HttpURLConnection connection) throws IOException {
        int status = connection.getResponseCode();
        if(status != 200) {
            fireError(Status.ABORTED, new IOException(connection.getResponseMessage()), "Invalid status code " + status);
            return true;
        }
        return false;
    }

    private HttpURLConnection createBidirectionalConnection() throws IOException {
        HttpURLConnection connection = (HttpURLConnection) sendUrl.openConnection();
        connection.setDoInput(true);
        connection.setDoOutput(true);
        return connection;
    }

    private void readOutputFromServer(HttpURLConnection connection) throws IOException {
        try(InputStream in = connection.getInputStream()) {
            byte[] bytes = IoUtils.toByteArray(in);
            LOG.debug("INPUT: {}", HexUtil.toFormattedHexString(bytes));
        }
    }

    private void fireError(Status aStatus, Exception aException, String aReason) {
        LOG.error(aReason + " " + sendUrl, aException);
        if(listener != null) {
            listener.closed(aStatus, new Metadata());
        }
    }

    @Override
    public void cancelStream(Status aReason) {
        // we can't cancel HttpURLConnection execution
    }

}