/* =========================================================================================
* Copyright © 2013-2018 the kamon project <http://kamon.io/>
*
* 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 kamon
package context
import kamon.tag.TagSet
/**
* An immutable set of information that is tied to the processing of single operation in a service. A Context instance
* can contain tags and entries.
*
* Context tags are built on top of the TagSet abstraction that ships with Kamon and since Kamon knows exactly what
* types of values can be included in a TagSet it can automatically serialize and deserialize them when going over
* HTTP and/or Binary transports.
*
* Context entries can contain any arbitrary type specified by the user, but require additional configuration and
* implementation of entry readers and writers if you need them to go over HTTP and/or Binary transports.
*
* Context instances are meant to be constructed by using the builder functions on the Context companion object.
*
*/
class Context private (val entries: Map[String, Any], val tags: TagSet) {
/**
* Gets an entry from this Context. If the entry is present it's current value is returned, otherwise the empty value
* from the provided key will be returned.
*/
def get[T](key: Context.Key[T]): T =
entries.getOrElse(key.name, key.emptyValue).asInstanceOf[T]
/**
* Executes a lookup on the context tags. The actual return type depends on the provided lookup instance. Take a look
* at the built-in lookups available on the Lookups companion object.
*/
def getTag[T](lookup: TagSet.Lookup[T]): T =
tags.get(lookup)
/**
* Creates a new Context instance that includes the provided key and value. If the provided key was already
* associated with another value then the previous value will be discarded and overwritten with the provided one.
*/
def withKey[T](key: Context.Key[T], value: T): Context =
new Context(entries.updated(key.name, value), tags)
/**
* Creates a new Context instance that includes the provided tag key and value. If the provided tag key was already
* associated with another value then the previous tag value will be discarded and overwritten with the provided one.
*/
def withTag(key: String, value: String): Context =
new Context(entries, tags.withTag(key, value))
/**
* Creates a new Context instance that includes the provided tag key and value. If the provided tag key was already
* associated with another value then the previous tag value will be discarded and overwritten with the provided one.
*/
def withTag(key: String, value: Long): Context =
new Context(entries, tags.withTag(key, value))
/**
* Creates a new Context instance that includes the provided tag key and value. If the provided tag key was already
* associated with another value then the previous tag value will be discarded and overwritten with the provided one.
*/
def withTag(key: String, value: Boolean): Context =
new Context(entries, tags.withTag(key, value))
/**
* Creates a new Context instance that includes the provided tags. If any of the tags in this instance are associated
* to a key present on the provided tags then the previous values will be discarded and overwritten with the provided
* ones.
*/
def withTags(tags: TagSet): Context =
new Context(entries, this.tags.and(tags))
/**
* Returns whether this Context does not have any tags and does not have any entries.
*/
def isEmpty(): Boolean =
entries.isEmpty && tags.isEmpty
/**
* Returns whether this Context has any information, either as tags or entries.
*/
def nonEmpty(): Boolean =
!isEmpty()
}
object Context {
val Empty = new Context(Map.empty, TagSet.Empty)
/**
* Creates a new Context instance with the provided tags and no entries.
*/
def of(tags: TagSet): Context =
new Context(Map.empty, tags)
/**
* Creates a new Context instance with the provided key and no tags.
*/
def of[T](key: Context.Key[T], value: T): Context =
new Context(Map(key.name -> value), TagSet.Empty)
/**
* Creates a new Context instance with a single entry and the provided tags.
*/
def of[T](key: Context.Key[T], value: T, tags: TagSet): Context =
new Context(Map(key.name -> value), tags)
/**
* Creates a new Context instance with two entries and no tags.
*/
def of[T, U](keyOne: Context.Key[T], valueOne: T, keyTwo: Context.Key[U], valueTwo: U): Context =
new Context(Map(keyOne.name -> valueOne, keyTwo.name -> valueTwo), TagSet.Empty)
/**
* Creates a new Context instance with two entries and the provided tags.
*/
def of[T, U](keyOne: Context.Key[T], valueOne: T, keyTwo: Context.Key[U], valueTwo: U, tags: TagSet): Context =
new Context(Map(keyOne.name -> valueOne, keyTwo.name -> valueTwo), tags)
/**
* Creates a new Context.Key instance that can be used to insert and retrieve values from the context entries.
* Context keys must have a unique name since they will be looked up in transports by their name and the context
* entries are internally stored using their key name as index.
*/
def key[T](name: String, emptyValue: T): Context.Key[T] =
new Context.Key(name, emptyValue)
/**
* Encapsulates the type, name and empty value for a context entry. All reads and writes from a context instance
* must be done using a context key, which will ensure the right type is used on both operations. The key's name
* is used when configuring mappings and incoming/outgoing/returning codecs for context propagation across channels.
*
* If you try to read an entry from a context and such entry is not present, the empty value for the key is returned
* instead.
*
* @param name Key name. Must be unique.
* @param emptyValue Value to be returned when reading from a context that doesn't have an entry with this key.
* @tparam ValueType Type of the value to be held on the context with this key.
*/
final class Key[ValueType](val name: String, val emptyValue: ValueType) {
override def hashCode(): Int =
name.hashCode
override def equals(that: Any): Boolean =
that.isInstanceOf[Context.Key[_]] && that.asInstanceOf[Context.Key[_]].name == this.name
}
}