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.netty.NettyChannelBuilder
  29 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.TokenAuthGrpcClientProperties
  30 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.interceptor.GrpcClientLoggingInterceptor
  31
  32 class TokenAuthGrpcClientService(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties) :
  33     BlueprintGrpcClientService {
  34
  35     override suspend fun channel(): ManagedChannel {
  36         val target =
  37             if (tokenAuthGrpcClientProperties.port == -1) tokenAuthGrpcClientProperties.host
  38             else "${tokenAuthGrpcClientProperties.host}:${tokenAuthGrpcClientProperties.port}"
  39
  40         val managedChannel = NettyChannelBuilder
  41             .forTarget(target)
  42             .nameResolverFactory(DnsNameResolverProvider())
  43             .intercept(GrpcClientLoggingInterceptor())
  44             .intercept(TokenAuthClientInterceptor(tokenAuthGrpcClientProperties)).usePlaintext().build()
  45         return managedChannel
  46     }
  47 }
  48
  49 class TokenAuthClientInterceptor(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties) :
  50     ClientInterceptor {
  51
  52     override fun <ReqT, RespT> interceptCall(
  53         method: MethodDescriptor<ReqT, RespT>,
  54         callOptions: CallOptions,
  55         channel: Channel
  56     ): ClientCall<ReqT, RespT> {
  57
  58         val authHeader = Metadata.Key.of("Authorization", Metadata.ASCII_STRING_MARSHALLER)
  59
  60         return object : ForwardingClientCall
  61         .SimpleForwardingClientCall<ReqT, RespT>(channel.newCall(method, callOptions)) {
  62
  63             override fun start(responseListener: Listener<RespT>, headers: Metadata) {
  64                 headers.put(authHeader, tokenAuthGrpcClientProperties.token)
  65                 super.start(responseListener, headers)
  66             }
  67         }
  68     }
  69 }