com.payneteasy.grpc.longpolling.server.servlet.unary.UnaryHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of grpc-long-polling-server Show documentation
The newest version!
package com.payneteasy.grpc.longpolling.server.servlet.unary;

import com.payneteasy.grpc.longpolling.server.LongPollingServerTransport;
import com.payneteasy.grpc.longpolling.server.servlet.MethodCall;
import com.payneteasy.grpc.longpolling.server.servlet.ServletOptions;
import io.grpc.Attributes;
import io.grpc.Metadata;
import io.grpc.internal.ServerListener;
import io.grpc.internal.ServerTransportListener;
import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.concurrent.TimeUnit;

public class UnaryHandler {

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

    private final ServerListener             serverListener;
    private final LongPollingServerTransport serverTransport;
    private final ServletOptions             options;

    public UnaryHandler(ServerListener aServerListener, LongPollingServerTransport aTransport, ServletOptions aOptions) {
        serverListener  = aServerListener;
        serverTransport = aTransport;
        options         = aOptions;
    }

    public void handle(HttpServletRequest aRequest, MethodCall aMethod, HttpServletResponse aResponse) throws IOException {
        LOG.debug("Creating unary transport and stream: {}", aMethod.getStreamId());
        ServerTransportListener listener = serverListener.transportCreated(serverTransport);

        byte[] buffer = IOUtils.toByteArray(aRequest.getInputStream());
        UnaryServerStream stream = new UnaryServerStream(buffer, aResponse, aMethod.getStreamId());
        listener.streamCreated(stream, aMethod.getMethod(), new Metadata());
        listener.transportReady(Attributes.EMPTY);

        try {
            LOG.debug("Waiting {} ms while sending response to servlet output ...", options.getReadTimeout());
            if(!stream.waitDone(options.getReadTimeout(), TimeUnit.MILLISECONDS)) {
                LOG.error("No response from gRPC service");
                aResponse.sendError(HttpServletResponse.SC_BAD_GATEWAY);
            }
        } catch (InterruptedException e) {
            LOG.error("Cannot wait 1 min", e);
            Thread.currentThread().interrupt();
        } finally {
            listener.transportTerminated();
            LOG.debug("Transport {} terminated", aMethod.getStreamId().getTransportId());
        }
    }
}