io.buoyant.linkerd.Client.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of linkerd-core_2.11 Show documentation
linkerd-core
The newest version!
package io.buoyant.linkerd

import com.fasterxml.jackson.annotation.{JsonIgnore, JsonSubTypes, JsonTypeInfo}
import com.twitter.finagle.{Dentry, Path, Stack}
import com.twitter.finagle.buoyant.PathMatcher
import io.buoyant.config.PolymorphicConfig
import io.buoyant.router.StackRouter.Client.{ClientParams, PerClientParams}

/**
 * Client is the polymorphic type the jackson will use to deserialize the
 * `config` section of a linkerd config.
 */
@JsonTypeInfo(
  use = JsonTypeInfo.Id.NAME,
  include = JsonTypeInfo.As.EXISTING_PROPERTY,
  property = "kind",
  visible = true,
  defaultImpl = classOf[DefaultClientImpl]
)
@JsonSubTypes(Array(
  new JsonSubTypes.Type(value = classOf[DefaultClientImpl], name = "io.l5d.global"),
  new JsonSubTypes.Type(value = classOf[StaticClientImpl], name = "io.l5d.static")
))
abstract class Client extends PolymorphicConfig {
  @JsonIgnore
  def clientParams: PerClientParams
}

/**
 * DefaultClient mixes in ClientConfig so that client configuration properties
 * can be specified directly on the `client` object in the linkerd config.  This
 * is a trait so that it can be mixed in to protocol specific versions.
 */
trait DefaultClient extends Client with ClientConfig {
  kind = "io.l5d.global"

  @JsonIgnore
  private[this] val matchAll = PathMatcher("/")

  @JsonIgnore
  private[this] val mk: Map[String, String] => Stack.Params = { vars =>
    params(vars)
  }

  @JsonIgnore
  def clientParams = PerClientParams(Seq(ClientParams(matchAll, mk)))
}

class DefaultClientImpl extends Client with DefaultClient

/**
 * StaticClient consists of a list of PrefixConfigs.  This is a trait so that it
 * can be mixed in to protocol specific versions.
 */
trait StaticClient {
  val configs: Seq[PrefixConfig]

  @JsonIgnore
  def clientParams = PerClientParams(configs.map { config =>
    ClientParams(config.prefix, config.params)
  })
}

class StaticClientImpl(val configs: Seq[PrefixConfig]) extends Client with StaticClient

class PrefixConfig(val prefix: PathMatcher) extends ClientConfig