io.buoyant.namer.k8s.K8sInitializer.scala Maven / Gradle / Ivy

Go to download
package io.buoyant.namer.k8s

import com.fasterxml.jackson.annotation.JsonIgnore
import com.twitter.finagle._
import io.buoyant.config.types.Port
import io.buoyant.k8s._
import io.buoyant.k8s.v1.Api
import io.buoyant.namer.{NamerConfig, NamerInitializer}

/**
 * Supports namer configurations in the form:
 *
 *  * namers:
 * - kind: io.l5d.k8s
 *   exerimental: true
 *   host: localhost
 *   port: 8001
 * 
 */
class K8sInitializer extends NamerInitializer {
  val configClass = classOf[K8sConfig]
  override def configId = "io.l5d.k8s"
}

object K8sInitializer extends K8sInitializer

case class K8sConfig(
  host: Option[String],
  port: Option[Port],
  labelSelector: Option[String]
) extends NamerConfig with ClientConfig {

  @JsonIgnore
  override def defaultPrefix: Path = Path.read("/io.l5d.k8s")

  @JsonIgnore
  def portNum = port.map(_.port)

  /**
   * Construct a namer.
   */
  @JsonIgnore
  override def newNamer(params: Stack.Params): Namer = {
    val client = mkClient(params)
    def mkNs(ns: String) = {
      val label = param.Label(s"namer${prefix.show}/$ns")
      Api(client.configured(label).newService(dst)).withNamespace(ns)
    }
    new EndpointsNamer(prefix, labelSelector, mkNs)
  }
}