ms/blueprintsprocessor/modules/commons/grpc-lib/src/main/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/grpc/service/TokenAuthGrpcClientService.kt

   1 /*
   2  *  Copyright © 2019 IBM.
   3  *
   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
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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.
  15  */
  16
  17 package org.onap.ccsdk.cds.blueprintsprocessor.grpc.service
  18
  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.ManagedChannel
  25 import io.grpc.Metadata
  26 import io.grpc.MethodDescriptor
  27 import io.grpc.internal.DnsNameResolverProvider
  28 import io.grpc.internal.PickFirstLoadBalancerProvider
  29 import io.grpc.netty.NettyChannelBuilder
  30 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.TokenAuthGrpcClientProperties
  31 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.interceptor.GrpcClientLoggingInterceptor
  32
  33 class TokenAuthGrpcClientService(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties) :
  34     BluePrintGrpcClientService {
  35
  36     override suspend fun channel(): ManagedChannel {
  37         val managedChannel = NettyChannelBuilder
  38             .forAddress(tokenAuthGrpcClientProperties.host, tokenAuthGrpcClientProperties.port)
  39             .nameResolverFactory(DnsNameResolverProvider())
  40             .loadBalancerFactory(PickFirstLoadBalancerProvider())
  41             .intercept(GrpcClientLoggingInterceptor())
  42             .intercept(TokenAuthClientInterceptor(tokenAuthGrpcClientProperties)).usePlaintext().build()
  43         return managedChannel
  44     }
  45 }
  46
  47 class TokenAuthClientInterceptor(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties) : ClientInterceptor {
  48
  49     override fun <ReqT, RespT> interceptCall(
  50         method: MethodDescriptor<ReqT, RespT>,
  51         callOptions: CallOptions,
  52         channel: Channel
  53     ): ClientCall<ReqT, RespT> {
  54
  55         val authHeader = Metadata.Key.of("Authorization", Metadata.ASCII_STRING_MARSHALLER)
  56
  57         return object : ForwardingClientCall
  58         .SimpleForwardingClientCall<ReqT, RespT>(channel.newCall(method, callOptions)) {
  59
  60             override fun start(responseListener: Listener<RespT>, headers: Metadata) {
  61                 headers.put(authHeader, tokenAuthGrpcClientProperties.token)
  62                 super.start(responseListener, headers)
  63             }
  64         }
  65     }
  66 }