com.dimajix.flowman.kernel.grpc.ServiceHandler.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of flowman-kernel-server Show documentation
Flowman Kernel server
The newest version!
/*
 * Copyright (C) 2018 The Flowman Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.dimajix.flowman.kernel.grpc

import scala.util.control.NonFatal

import io.grpc.Status
import io.grpc.StatusException
import io.grpc.StatusRuntimeException
import io.grpc.stub.StreamObserver
import org.slf4j.Logger

import com.dimajix.common.ExceptionUtils.reasons
import com.dimajix.flowman.grpc.ExceptionUtils
import com.dimajix.flowman.grpc.GrpcService


trait ServiceHandler extends GrpcService {
    protected val logger:Logger

    def respondTo[T](methodName:String, responseObserver:StreamObserver[T])(callable: => T) {
        val response = try {
            callable;
        }
        catch {
            case e@(_: StatusException | _: StatusRuntimeException) =>
                logger.error(s"Exception during execution of RPC call ${getClass.getSimpleName}.${methodName}:\n  ${reasons(e)}")
                responseObserver.onError(e)
                return;
            case NonFatal(t) =>
                logger.error(s"Exception during execution of RPC call ${getClass.getSimpleName}.${methodName}:\n  ${reasons(t)}")
                val e = ExceptionUtils.asStatusException(Status.INTERNAL, t, true)
                responseObserver.onError(e)
                return;
        }

        responseObserver.onNext(response);
        responseObserver.onCompleted();
    }


    def streamRequests[S,T](methodName:String, responseObserver: StreamObserver[T])(next: S => Unit)(result: => T) : StreamObserver[S] = {
        new StreamObserver[S]() {
            @Override
            def onNext(req: S): Unit = {
                try {
                    next(req)
                }
                catch {
                    case e@(_: StatusException | _: StatusRuntimeException) =>
                        logger.error(s"Exception during execution of RPC call ${getClass.getSimpleName}.${methodName}:\n  ${reasons(e)}")
                        responseObserver.onError(e)
                    case NonFatal(t) =>
                        logger.error(s"Exception during execution of RPC call ${getClass.getSimpleName}.${methodName}:\n  ${reasons(t)}")
                        val e = ExceptionUtils.asStatusException(Status.INTERNAL, t, true)
                        responseObserver.onError(e)
                }
            }

            @Override
            def onError(t: Throwable): Unit = {
            }

            @Override
            def onCompleted(): Unit = {
                respondTo(methodName, responseObserver)(result)
            }
        };
    }
}