2 * Copyright © 2019 IBM.
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.service
20 import io.grpc.internal.DnsNameResolverProvider
21 import io.grpc.internal.PickFirstLoadBalancerProvider
22 import io.grpc.netty.NettyChannelBuilder
23 import org.onap.ccsdk.cds.blueprintsprocessor.grpc.TokenAuthGrpcClientProperties
25 class TokenAuthGrpcClientService(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties)
26 : BluePrintGrpcClientService {
28 override suspend fun channel(): ManagedChannel {
29 val managedChannel = NettyChannelBuilder
30 .forAddress(tokenAuthGrpcClientProperties.host, tokenAuthGrpcClientProperties.port)
31 .nameResolverFactory(DnsNameResolverProvider())
32 .loadBalancerFactory(PickFirstLoadBalancerProvider())
33 .intercept(TokenAuthClientInterceptor(tokenAuthGrpcClientProperties)).usePlaintext().build()
38 class TokenAuthClientInterceptor(private val tokenAuthGrpcClientProperties: TokenAuthGrpcClientProperties) : ClientInterceptor {
40 override fun <ReqT, RespT> interceptCall(method: MethodDescriptor<ReqT, RespT>,
41 callOptions: CallOptions, channel: Channel): ClientCall<ReqT, RespT> {
43 val authHeader = Metadata.Key.of("Authorization", Metadata.ASCII_STRING_MARSHALLER)
45 return object : ForwardingClientCall
46 .SimpleForwardingClientCall<ReqT, RespT>(channel.newCall(method, callOptions)) {
48 override fun start(responseListener: Listener<RespT>, headers: Metadata) {
49 headers.put(authHeader, tokenAuthGrpcClientProperties.token)
50 super.start(responseListener, headers)