data-resource-management-service/drms-graph-impl/drms-api/src/main/java/org/apache/airavata/drms/api/interceptors/InterceptorPipelineExecutor.java [14:66]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class InterceptorPipelineExecutor implements ServerInterceptor {
    private final Logger LOGGER = LoggerFactory.getLogger(InterceptorPipelineExecutor.class);

    private Stack<ServiceInterceptor> interceptorSet;

    public InterceptorPipelineExecutor(Stack<ServiceInterceptor> integrationServiceInterceptors) {
        this.interceptorSet = integrationServiceInterceptors;
    }

    @Override
    public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(ServerCall<ReqT, RespT> serverCall,
                                                                 Metadata metadata, ServerCallHandler<ReqT, RespT> serverCallHandler) {
        String fullMethod = serverCall.getMethodDescriptor().getFullMethodName();
        String methodName = fullMethod.split("/")[1];
        String serviceName = fullMethod.split("/")[0];

        LOGGER.debug("Calling method : " + serverCall.getMethodDescriptor().getFullMethodName());
        metadata.put(Metadata.Key.of("service-name", Metadata.ASCII_STRING_MARSHALLER), serviceName);

        return new ForwardingServerCallListener.SimpleForwardingServerCallListener<ReqT>(serverCallHandler.startCall(serverCall, metadata)) {

            ReqT resp = null;

            @Override
            public void onMessage(ReqT message) {
                try {
                    Iterator it = interceptorSet.iterator();
                    while (it.hasNext()) {
                        ServiceInterceptor interceptor = (ServiceInterceptor) it.next();
                        resp = interceptor.intercept(methodName, metadata, (resp == null) ? message : resp);
                    }
                    super.onMessage(resp);
                } catch (Exception ex) {
                    String msg = "Error while validating method " + methodName + ", " + ex.getMessage();
                    LOGGER.error(msg, ex);
                    serverCall.close(Status.INVALID_ARGUMENT.withDescription(msg), new Metadata());
                }
            }

            @Override
            public void onHalfClose() {
                try {
                    super.onHalfClose();
                } catch (IllegalStateException e) {
                    LOGGER.debug(e.getMessage());
                } catch (Exception e) {
                    String msg = "Error while validating method " + methodName + ", " + e.getMessage();
                    LOGGER.error(msg);
                    serverCall.close(Status.INVALID_ARGUMENT.withDescription(msg), metadata);
                }
            }

        };
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



data-resource-management-service/drms-rdbms-impl/drms-server/src/main/java/org/apache/airavata/drms/api/interceptors/InterceptorPipelineExecutor.java [14:66]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class InterceptorPipelineExecutor implements ServerInterceptor {
    private final Logger LOGGER = LoggerFactory.getLogger(InterceptorPipelineExecutor.class);

    private Stack<ServiceInterceptor> interceptorSet;

    public InterceptorPipelineExecutor(Stack<ServiceInterceptor> integrationServiceInterceptors) {
        this.interceptorSet = integrationServiceInterceptors;
    }

    @Override
    public <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(ServerCall<ReqT, RespT> serverCall,
                                                                 Metadata metadata, ServerCallHandler<ReqT, RespT> serverCallHandler) {
        String fullMethod = serverCall.getMethodDescriptor().getFullMethodName();
        String methodName = fullMethod.split("/")[1];
        String serviceName = fullMethod.split("/")[0];

        LOGGER.debug("Calling method : " + serverCall.getMethodDescriptor().getFullMethodName());
        metadata.put(Metadata.Key.of("service-name", Metadata.ASCII_STRING_MARSHALLER), serviceName);

        return new ForwardingServerCallListener.SimpleForwardingServerCallListener<ReqT>(serverCallHandler.startCall(serverCall, metadata)) {

            ReqT resp = null;

            @Override
            public void onMessage(ReqT message) {
                try {
                    Iterator it = interceptorSet.iterator();
                    while (it.hasNext()) {
                        ServiceInterceptor interceptor = (ServiceInterceptor) it.next();
                        resp = interceptor.intercept(methodName, metadata, (resp == null) ? message : resp);
                    }
                    super.onMessage(resp);
                } catch (Exception ex) {
                    String msg = "Error while validating method " + methodName + ", " + ex.getMessage();
                    LOGGER.error(msg, ex);
                    serverCall.close(Status.INVALID_ARGUMENT.withDescription(msg), new Metadata());
                }
            }

            @Override
            public void onHalfClose() {
                try {
                    super.onHalfClose();
                } catch (IllegalStateException e) {
                    LOGGER.debug(e.getMessage());
                } catch (Exception e) {
                    String msg = "Error while validating method " + methodName + ", " + e.getMessage();
                    LOGGER.error(msg);
                    serverCall.close(Status.INVALID_ARGUMENT.withDescription(msg), metadata);
                }
            }

        };
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -