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

   1 /*
   2  * Copyright © 2018-2019 AT&T Intellectual Property.
   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.nats.service
  18
  19 import io.nats.client.Nats
  20 import io.nats.client.Options
  21 import io.nats.streaming.NatsStreaming
  22 import io.nats.streaming.StreamingConnection
  23 import org.onap.ccsdk.cds.blueprintsprocessor.nats.TokenAuthNatsConnectionProperties
  24 import org.onap.ccsdk.cds.controllerblueprints.core.logger
  25 import org.onap.ccsdk.cds.controllerblueprints.core.splitCommaAsList
  26
  27 open class TokenAuthNatsService(private val natsConnectionProperties: TokenAuthNatsConnectionProperties) :
  28     BluePrintNatsService {
  29
  30     private val log = logger(TokenAuthNatsService::class)
  31
  32     lateinit var streamingConnection: StreamingConnection
  33
  34     override suspend fun connection(): StreamingConnection {
  35         if (!::streamingConnection.isInitialized) {
  36             log.info(
  37                 "NATS connection requesting for cluster(${natsConnectionProperties.clusterId}) with" +
  38                     "clientId(${natsConnectionProperties.clientId})"
  39             )
  40
  41             val serverList = natsConnectionProperties.host.splitCommaAsList()
  42
  43             val options = Options.Builder()
  44                 .connectionName(natsConnectionProperties.clientId)
  45                 .servers(serverList.toTypedArray())
  46                 .token(natsConnectionProperties.token.toCharArray())
  47                 .build()
  48             val natsConnection = Nats.connect(options)
  49             val streamingOptions = io.nats.streaming.Options.Builder().natsConn(natsConnection).build()
  50             streamingConnection = NatsStreaming.connect(
  51                 natsConnectionProperties.clusterId,
  52                 natsConnectionProperties.clientId, streamingOptions
  53             )
  54         }
  55         return streamingConnection
  56     }
  57 }