io.zulia.server.connection.client.handler.InternalRequestHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zulia-server Show documentation

$project.description

There is a newer version: 1.6.4

package io.zulia.server.connection.client.handler;

import io.grpc.Metadata;
import io.grpc.StatusRuntimeException;
import io.zulia.cache.MetaKeys;
import io.zulia.message.ZuliaBase.Node;
import io.zulia.server.connection.client.InternalClient;
import io.zulia.server.connection.client.InternalRpcConnection;

public abstract class InternalRequestHandler {

	private InternalClient internalClient;

	public InternalRequestHandler(InternalClient internalClient) {
		this.internalClient = internalClient;
	}

	public S handleRequest(Node node, Q q) throws Exception {
		InternalRpcConnection rpcConnection = null;

		boolean valid = true;
		try {
			rpcConnection = internalClient.getInternalRpcConnection(node);
			return getResponse(q, rpcConnection);
		}
		catch (StatusRuntimeException e) {
			Metadata trailers = e.getTrailers();
			if (trailers.containsKey(MetaKeys.ERROR_KEY)) {
				throw new Exception(trailers.get(MetaKeys.ERROR_KEY));
			}
			else {
				throw e;
			}
		}
		catch (Exception e) {
			valid = false;
			throw e;
		}
		finally {
			internalClient.returnInternalBlockingConnection(node, rpcConnection, valid);
		}
	}

	protected abstract S getResponse(Q q, InternalRpcConnection rpcConnection);

}