All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.twitter.finagle.thrift.thriftscala.RequestContext.scala Maven / Gradle / Ivy

There is a newer version: 24.2.0
Show newest version
/**
 * Generated by Scrooge
 *   version: 20.4.1
 *   rev: 74d6bed636c90dc4b67e342ecd80033bad43e94d
 *   built at: 20200425-214521
 */
package com.twitter.finagle.thrift.thriftscala

import com.twitter.io.Buf
import com.twitter.scrooge.{
  InvalidFieldsException,
  LazyTProtocol,
  StructBuilder,
  StructBuilderFactory,
  TFieldBlob,
  ThriftStruct,
  ThriftStructCodec3,
  ThriftStructField,
  ThriftStructFieldInfo,
  ThriftStructMetaData,
  ValidatingThriftStruct,
  ValidatingThriftStructCodec3
}
import org.apache.thrift.protocol._
import org.apache.thrift.transport.TMemoryBuffer
import scala.collection.immutable.{Map => immutable$Map}
import scala.collection.mutable.Builder
import scala.reflect.{ClassTag, classTag}

/**
 * This struct serializes com.twitter.finagle.Context
 */
object RequestContext extends ValidatingThriftStructCodec3[RequestContext] with StructBuilderFactory[RequestContext] {
  val NoPassthroughFields: immutable$Map[Short, TFieldBlob] = immutable$Map.empty[Short, TFieldBlob]
  val Struct: TStruct = new TStruct("RequestContext")
  val KeyField: TField = new TField("key", TType.STRING, 1)
  val KeyFieldManifest: Manifest[_root_.java.nio.ByteBuffer] = manifest[_root_.java.nio.ByteBuffer]
  val ValueField: TField = new TField("value", TType.STRING, 2)
  val ValueFieldManifest: Manifest[_root_.java.nio.ByteBuffer] = manifest[_root_.java.nio.ByteBuffer]

  /**
   * Field information in declaration order.
   */
  lazy val fieldInfos: scala.List[ThriftStructFieldInfo] = scala.List[ThriftStructFieldInfo](
    new ThriftStructFieldInfo(
      KeyField,
      false,
      false,
      KeyFieldManifest,
      _root_.scala.None,
      _root_.scala.None,
      immutable$Map.empty[String, String],
      immutable$Map.empty[String, String],
      None
    ),
    new ThriftStructFieldInfo(
      ValueField,
      false,
      false,
      ValueFieldManifest,
      _root_.scala.None,
      _root_.scala.None,
      immutable$Map.empty[String, String],
      immutable$Map.empty[String, String],
      None
    )
  )


  val structAnnotations: immutable$Map[String, String] =
    immutable$Map.empty[String, String]

  private val fieldTypes: IndexedSeq[ClassTag[_]] = IndexedSeq[ClassTag[_]](
    classTag[_root_.java.nio.ByteBuffer].asInstanceOf[ClassTag[_]],
    classTag[_root_.java.nio.ByteBuffer].asInstanceOf[ClassTag[_]]
  )

  private[this] val structFields: Seq[ThriftStructField[RequestContext]] = Seq[ThriftStructField[RequestContext]](
    new ThriftStructField[RequestContext](
      KeyField,
      _root_.scala.Some(KeyFieldManifest),
      classOf[RequestContext]) {
        def getValue[R](struct: RequestContext): R = struct.key.asInstanceOf[R]
    },
    new ThriftStructField[RequestContext](
      ValueField,
      _root_.scala.Some(ValueFieldManifest),
      classOf[RequestContext]) {
        def getValue[R](struct: RequestContext): R = struct.value.asInstanceOf[R]
    }
  )

  override lazy val metaData: ThriftStructMetaData[RequestContext] =
    new ThriftStructMetaData(this, structFields, fieldInfos, Nil, structAnnotations)

  /**
   * Checks that all required fields are non-null.
   */
  def validate(_item: RequestContext): Unit = {
  }

  /**
   * Checks that the struct is a valid as a new instance. If there are any missing required or
   * construction required fields, return a non-empty list.
   */
  def validateNewInstance(item: RequestContext): scala.Seq[com.twitter.scrooge.validation.Issue] = {
    val buf = scala.collection.mutable.ListBuffer.empty[com.twitter.scrooge.validation.Issue]

    buf ++= validateField(item.key)
    buf ++= validateField(item.value)
    buf.toList
  }

  def withoutPassthroughFields(original: RequestContext): RequestContext =
    new Immutable(
      key = original.key,
      value = original.value
    )

  def newBuilder(): StructBuilder[RequestContext] = new RequestContextStructBuilder(_root_.scala.None, fieldTypes)

  override def encode(_item: RequestContext, _oproto: TProtocol): Unit = {
    _item.write(_oproto)
  }


  private[this] def lazyDecode(_iprot: LazyTProtocol): RequestContext = {

    var key: _root_.java.nio.ByteBuffer = null
    var value: _root_.java.nio.ByteBuffer = null

    var _passthroughFields: Builder[(Short, TFieldBlob), immutable$Map[Short, TFieldBlob]] = null
    var _done = false
    val _start_offset = _iprot.offset

    _iprot.readStructBegin()
    do {
      val _field = _iprot.readFieldBegin()
      val _fieldType = _field.`type`
      if (_fieldType == TType.STOP) {
        _done = true
      } else {
        _field.id match {
          case 1 =>
            if (_fieldType == TType.STRING) {
              key = readKeyValue(_iprot)
            } else {
              _root_.com.twitter.scrooge.internal.ApplicationExceptions.throwWrongFieldTypeException(
                "Received wrong type for field 'key' (expected=%s, actual=%s).",
                TType.STRING,
                _fieldType
              )
            }
          case 2 =>
            if (_fieldType == TType.STRING) {
              value = readValueValue(_iprot)
            } else {
              _root_.com.twitter.scrooge.internal.ApplicationExceptions.throwWrongFieldTypeException(
                "Received wrong type for field 'value' (expected=%s, actual=%s).",
                TType.STRING,
                _fieldType
              )
            }
          case _ =>
            if (_passthroughFields eq null)
              _passthroughFields = immutable$Map.newBuilder[Short, TFieldBlob]
            _passthroughFields += _root_.scala.Tuple2(_field.id, TFieldBlob.read(_field, _iprot))
        }
        _iprot.readFieldEnd()
      }
    } while (!_done)
    _iprot.readStructEnd()

    new LazyImmutable(
      _iprot,
      _iprot.buffer,
      _start_offset,
      _iprot.offset,
      key,
      value,
      if (_passthroughFields eq null)
        NoPassthroughFields
      else
        _passthroughFields.result()
    )
  }

  override def decode(_iprot: TProtocol): RequestContext = {
    if (_iprot.isInstanceOf[LazyTProtocol]) {
      lazyDecode(_iprot.asInstanceOf[LazyTProtocol])
    } else {
      eagerDecode(_iprot)
    }
  }

  private[thriftscala] def eagerDecode(_iprot: TProtocol): RequestContext = {
    var key: _root_.java.nio.ByteBuffer = null
    var value: _root_.java.nio.ByteBuffer = null
    var _passthroughFields: Builder[(Short, TFieldBlob), immutable$Map[Short, TFieldBlob]] = null
    var _done = false

    _iprot.readStructBegin()
    do {
      val _field = _iprot.readFieldBegin()
      val _fieldType = _field.`type`
      if (_fieldType == TType.STOP) {
        _done = true
      } else {
        _field.id match {
          case 1 =>
            if (_fieldType == TType.STRING) {
              key = readKeyValue(_iprot)
            } else {
              _root_.com.twitter.scrooge.internal.ApplicationExceptions.throwWrongFieldTypeException(
                "Received wrong type for field 'key' (expected=%s, actual=%s).",
                TType.STRING,
                _fieldType
              )
            }
          case 2 =>
            if (_fieldType == TType.STRING) {
              value = readValueValue(_iprot)
            } else {
              _root_.com.twitter.scrooge.internal.ApplicationExceptions.throwWrongFieldTypeException(
                "Received wrong type for field 'value' (expected=%s, actual=%s).",
                TType.STRING,
                _fieldType
              )
            }
          case _ =>
            if (_passthroughFields eq null)
              _passthroughFields = immutable$Map.newBuilder[Short, TFieldBlob]
            _passthroughFields += _root_.scala.Tuple2(_field.id, TFieldBlob.read(_field, _iprot))
        }
        _iprot.readFieldEnd()
      }
    } while (!_done)
    _iprot.readStructEnd()

    new Immutable(
      key,
      value,
      if (_passthroughFields eq null)
        NoPassthroughFields
      else
        _passthroughFields.result()
    )
  }

  def apply(
    key: _root_.java.nio.ByteBuffer,
    value: _root_.java.nio.ByteBuffer
  ): RequestContext =
    new Immutable(
      key,
      value
    )

  def unapply(_item: RequestContext): _root_.scala.Option[_root_.scala.Tuple2[_root_.java.nio.ByteBuffer, _root_.java.nio.ByteBuffer]] = _root_.scala.Some(_item.toTuple)


  @inline private[thriftscala] def readKeyValue(_iprot: TProtocol): _root_.java.nio.ByteBuffer = {
    _iprot.readBinary()
  }

  @inline private def writeKeyField(key_item: _root_.java.nio.ByteBuffer, _oprot: TProtocol): Unit = {
    _oprot.writeFieldBegin(KeyField)
    writeKeyValue(key_item, _oprot)
    _oprot.writeFieldEnd()
  }

  @inline private def writeKeyValue(key_item: _root_.java.nio.ByteBuffer, _oprot: TProtocol): Unit = {
    _oprot.writeBinary(key_item)
  }

  @inline private[thriftscala] def readValueValue(_iprot: TProtocol): _root_.java.nio.ByteBuffer = {
    _iprot.readBinary()
  }

  @inline private def writeValueField(value_item: _root_.java.nio.ByteBuffer, _oprot: TProtocol): Unit = {
    _oprot.writeFieldBegin(ValueField)
    writeValueValue(value_item, _oprot)
    _oprot.writeFieldEnd()
  }

  @inline private def writeValueValue(value_item: _root_.java.nio.ByteBuffer, _oprot: TProtocol): Unit = {
    _oprot.writeBinary(value_item)
  }


  object Immutable extends ThriftStructCodec3[RequestContext] {
    override def encode(_item: RequestContext, _oproto: TProtocol): Unit = { _item.write(_oproto) }
    override def decode(_iprot: TProtocol): RequestContext = RequestContext.decode(_iprot)
    override lazy val metaData: ThriftStructMetaData[RequestContext] = RequestContext.metaData
  }

  /**
   * The default read-only implementation of RequestContext.  You typically should not need to
   * directly reference this class; instead, use the RequestContext.apply method to construct
   * new instances.
   */
  class Immutable(
      val key: _root_.java.nio.ByteBuffer,
      val value: _root_.java.nio.ByteBuffer,
      override val _passthroughFields: immutable$Map[Short, TFieldBlob])
    extends RequestContext {
    def this(
      key: _root_.java.nio.ByteBuffer,
      value: _root_.java.nio.ByteBuffer
    ) = this(
      key,
      value,
      immutable$Map.empty[Short, TFieldBlob]
    )
  }

  /**
   * This is another Immutable, this however keeps strings as lazy values that are lazily decoded from the backing
   * array byte on read.
   */
  private[this] class LazyImmutable(
      _proto: LazyTProtocol,
      _buf: Array[Byte],
      _start_offset: Int,
      _end_offset: Int,
      val key: _root_.java.nio.ByteBuffer,
      val value: _root_.java.nio.ByteBuffer,
      override val _passthroughFields: immutable$Map[Short, TFieldBlob])
    extends RequestContext {

    override def write(_oprot: TProtocol): Unit = {
      if (_oprot.isInstanceOf[LazyTProtocol]) {
        _oprot.asInstanceOf[LazyTProtocol].writeRaw(_buf, _start_offset, _end_offset - _start_offset)
      } else {
        super.write(_oprot)
      }
    }


    /**
     * Override the super hash code to make it a lazy val rather than def.
     *
     * Calculating the hash code can be expensive, caching it where possible
     * can provide significant performance wins. (Key in a hash map for instance)
     * Usually not safe since the normal constructor will accept a mutable map or
     * set as an arg
     * Here however we control how the class is generated from serialized data.
     * With the class private and the contract that we throw away our mutable references
     * having the hash code lazy here is safe.
     */
    override lazy val hashCode: Int = super.hashCode
  }

  /**
   * This Proxy trait allows you to extend the RequestContext trait with additional state or
   * behavior and implement the read-only methods from RequestContext using an underlying
   * instance.
   */
  trait Proxy extends RequestContext {
    protected def _underlying_RequestContext: RequestContext
    override def key: _root_.java.nio.ByteBuffer = _underlying_RequestContext.key
    override def value: _root_.java.nio.ByteBuffer = _underlying_RequestContext.value
    override def _passthroughFields: immutable$Map[Short, TFieldBlob] = _underlying_RequestContext._passthroughFields
  }
}

/**
 * Prefer the companion object's [[com.twitter.finagle.thrift.thriftscala.RequestContext.apply]]
 * for construction if you don't need to specify passthrough fields.
 */
trait RequestContext
  extends ThriftStruct
  with _root_.scala.Product2[_root_.java.nio.ByteBuffer, _root_.java.nio.ByteBuffer]
  with ValidatingThriftStruct[RequestContext]
  with java.io.Serializable
{
  import RequestContext._

  def key: _root_.java.nio.ByteBuffer
  def value: _root_.java.nio.ByteBuffer

  def _passthroughFields: immutable$Map[Short, TFieldBlob] = immutable$Map.empty

  def _1: _root_.java.nio.ByteBuffer = key
  def _2: _root_.java.nio.ByteBuffer = value

  def toTuple: _root_.scala.Tuple2[_root_.java.nio.ByteBuffer, _root_.java.nio.ByteBuffer] =
    _root_.scala.Tuple2[_root_.java.nio.ByteBuffer, _root_.java.nio.ByteBuffer](
      key,
      value
    )


  /**
   * Gets a field value encoded as a binary blob using TCompactProtocol.  If the specified field
   * is present in the passthrough map, that value is returned.  Otherwise, if the specified field
   * is known and not optional and set to None, then the field is serialized and returned.
   */
  def getFieldBlob(_fieldId: Short): _root_.scala.Option[TFieldBlob] = {
    val passedthroughValue = _passthroughFields.get(_fieldId)
    if (passedthroughValue.isDefined) {
      passedthroughValue
    } else {
      val _buff = new TMemoryBuffer(32)
      val _oprot = new TCompactProtocol(_buff)

      val _fieldOpt: _root_.scala.Option[TField] = _fieldId match {
        case 1 =>
          if (key ne null) {
            writeKeyValue(key, _oprot)
            _root_.scala.Some(RequestContext.KeyField)
          } else {
            _root_.scala.None
          }
        case 2 =>
          if (value ne null) {
            writeValueValue(value, _oprot)
            _root_.scala.Some(RequestContext.ValueField)
          } else {
            _root_.scala.None
          }
        case _ => _root_.scala.None
      }
      if (_fieldOpt.isDefined) {
        _root_.scala.Some(TFieldBlob(_fieldOpt.get, Buf.ByteArray.Owned(_buff.getArray)))
      } else {
        _root_.scala.None
      }
    }
  }


  /**
   * Collects TCompactProtocol-encoded field values according to `getFieldBlob` into a map.
   */
  def getFieldBlobs(ids: TraversableOnce[Short]): immutable$Map[Short, TFieldBlob] =
    (ids.flatMap { id => getFieldBlob(id).map { fieldBlob => (id, fieldBlob) } }).toMap

  /**
   * Sets a field using a TCompactProtocol-encoded binary blob.  If the field is a known
   * field, the blob is decoded and the field is set to the decoded value.  If the field
   * is unknown and passthrough fields are enabled, then the blob will be stored in
   * _passthroughFields.
   */
  def setField(_blob: TFieldBlob): RequestContext = {
    var key: _root_.java.nio.ByteBuffer = this.key
    var value: _root_.java.nio.ByteBuffer = this.value
    var _passthroughFields = this._passthroughFields
    _blob.id match {
      case 1 =>
        key = readKeyValue(_blob.read)
      case 2 =>
        value = readValueValue(_blob.read)
      case _ => _passthroughFields += _root_.scala.Tuple2(_blob.id, _blob)
    }
    new Immutable(
      key,
      value,
      _passthroughFields
    )
  }

  /**
   * If the specified field is optional, it is set to None.  Otherwise, if the field is
   * known, it is reverted to its default value; if the field is unknown, it is removed
   * from the passthroughFields map, if present.
   */
  def unsetField(_fieldId: Short): RequestContext = {
    var key: _root_.java.nio.ByteBuffer = this.key
    var value: _root_.java.nio.ByteBuffer = this.value

    _fieldId match {
      case 1 =>
        key = null
      case 2 =>
        value = null
      case _ =>
    }
    new Immutable(
      key,
      value,
      _passthroughFields - _fieldId
    )
  }

  /**
   * If the specified field is optional, it is set to None.  Otherwise, if the field is
   * known, it is reverted to its default value; if the field is unknown, it is removed
   * from the passthroughFields map, if present.
   */
  def unsetKey: RequestContext = unsetField(1)

  def unsetValue: RequestContext = unsetField(2)


  override def write(_oprot: TProtocol): Unit = {
    RequestContext.validate(this)
    _oprot.writeStructBegin(Struct)
    if (key ne null) writeKeyField(key, _oprot)
    if (value ne null) writeValueField(value, _oprot)
    if (_passthroughFields.nonEmpty) {
      _passthroughFields.values.foreach { _.write(_oprot) }
    }
    _oprot.writeFieldStop()
    _oprot.writeStructEnd()
  }

  def copy(
    key: _root_.java.nio.ByteBuffer = this.key,
    value: _root_.java.nio.ByteBuffer = this.value,
    _passthroughFields: immutable$Map[Short, TFieldBlob] = this._passthroughFields
  ): RequestContext =
    new Immutable(
      key,
      value,
      _passthroughFields
    )

  override def canEqual(other: Any): Boolean = other.isInstanceOf[RequestContext]

  private[this] def _equals(other: RequestContext): Boolean =
      this.productArity == other.productArity &&
      this.productIterator.sameElements(other.productIterator) &&
      this._passthroughFields == other._passthroughFields

  override def equals(other: Any): Boolean =
    canEqual(other) && _equals(other.asInstanceOf[RequestContext])

  override def hashCode: Int = {
    _root_.scala.runtime.ScalaRunTime._hashCode(this)
  }

  override def toString: String = _root_.scala.runtime.ScalaRunTime._toString(this)

  override def productPrefix: String = "RequestContext"

  def _codec: ValidatingThriftStructCodec3[RequestContext] = RequestContext

  def newBuilder(): StructBuilder[RequestContext] = new RequestContextStructBuilder(_root_.scala.Some(this), fieldTypes)
}

private[thriftscala] class RequestContextStructBuilder(instance: _root_.scala.Option[RequestContext], fieldTypes: IndexedSeq[ClassTag[_]])
    extends StructBuilder[RequestContext](fieldTypes) {

  def build(): RequestContext = {
    val _fieldArray = fieldArray // shadow variable
    if (instance.isDefined) {
      val instanceValue = instance.get
      RequestContext(
        if (_fieldArray(0) == null) instanceValue.key else _fieldArray(0).asInstanceOf[_root_.java.nio.ByteBuffer],
        if (_fieldArray(1) == null) instanceValue.value else _fieldArray(1).asInstanceOf[_root_.java.nio.ByteBuffer]
      )
    } else {
      if (genericArrayOps(_fieldArray).contains(null)) throw new InvalidFieldsException(structBuildError("RequestContext"))
      RequestContext(
        _fieldArray(0).asInstanceOf[_root_.java.nio.ByteBuffer],
        _fieldArray(1).asInstanceOf[_root_.java.nio.ByteBuffer]
      )
    }
  }
}





© 2015 - 2024 Weber Informatics LLC | Privacy Policy