gerrit.onap Code Review - ccsdk/cds.git/blob

   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.splitCommaAsList
  25
  26 open class TokenAuthNatsService(private val natsConnectionProperties: TokenAuthNatsConnectionProperties) :
  27     BluePrintNatsService {
  28
  29     lateinit var streamingConnection: StreamingConnection
  30
  31     override suspend fun connection(): StreamingConnection {
  32         if (!::streamingConnection.isInitialized) {
  33             val serverList = natsConnectionProperties.host.splitCommaAsList()
  34
  35             val options = Options.Builder()
  36                 .connectionName(natsConnectionProperties.clientId)
  37                 .servers(serverList.toTypedArray())
  38                 .token(natsConnectionProperties.token.toCharArray())
  39                 .build()
  40             val natsConnection = Nats.connect(options)
  41             val streamingOptions = io.nats.streaming.Options.Builder().natsConn(natsConnection).build()
  42             streamingConnection = NatsStreaming.connect(
  43                 natsConnectionProperties.clusterId,
  44                 natsConnectionProperties.clientId, streamingOptions
  45             )
  46         }
  47         return streamingConnection
  48     }
  49 }