com.arangodb.internal.velocystream.VstCommunicationDestructor.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of persistence Show documentation

There is a newer version: 0.7.9

/*
 * Copyright 2020 ABSA Group Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.arangodb.internal.velocystream

import com.arangodb.internal.net.{AccessType, HostDescription, HostHandler, HostImpl}
import com.arangodb.internal.velocystream.internal.VstConnection
import za.co.absa.commons.reflect.ReflectionUtils.extractFieldValue

import javax.net.ssl.SSLContext

object VstCommunicationDestructor {

  private object Fields {
    final val HostHandler = "hostHandler"
    final val SslContext = "sslContext"
  }

  case class ConnectionParams(
    hostDescription: HostDescription,
    user: String,
    password: String,
    sslContext: Option[SSLContext]
  )

  def unapply(comm: VstCommunication[_, _]): Option[ConnectionParams] = {
    val hostHandler = extractFieldValue[VstCommunication[_, _], HostHandler](comm, Fields.HostHandler)
    val host = hostHandler.get(null, AccessType.WRITE).asInstanceOf[HostImpl]
    val sslContext = extractFieldValue[VstConnection[_], SSLContext](host.connection, Fields.SslContext)

    val connParams = ConnectionParams(
      host.getDescription,
      comm.user,
      comm.password,
      Option(sslContext)
    )

    Some(connParams)
  }
}