2 * Copyright © 2018-2019 AT&T Intellectual Property.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.onap.ccsdk.cds.blueprintsprocessor.grpc.interceptor
19 import io.grpc.CallOptions
20 import io.grpc.Channel
21 import io.grpc.ClientCall
22 import io.grpc.ClientInterceptor
23 import io.grpc.ForwardingClientCall
24 import io.grpc.ForwardingClientCallListener
25 import io.grpc.Metadata
26 import io.grpc.MethodDescriptor
27 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.service.GrpcLoggerService
28 import org.onap.ccsdk.cds.controllerblueprints.core.logger
30 class GrpcClientLoggingInterceptor : ClientInterceptor {
31 val log = logger(GrpcClientLoggingInterceptor::class)
33 val loggingService = GrpcLoggerService()
35 override fun <ReqT, RespT> interceptCall(
36 method: MethodDescriptor<ReqT, RespT>,
37 callOptions: CallOptions,
39 ): ClientCall<ReqT, RespT> {
41 return object : ForwardingClientCall
42 .SimpleForwardingClientCall<ReqT, RespT>(channel.newCall(method, callOptions)) {
44 override fun start(responseListener: Listener<RespT>, headers: Metadata) {
45 val listener = object : ForwardingClientCallListener.SimpleForwardingClientCallListener<RespT>(responseListener) {
46 override fun onMessage(message: RespT) {
47 loggingService.grpcInvoking(headers)
48 super.onMessage(message)
51 super.start(listener, headers)