com.twitter.finagle.ThriftMuxUtil.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of finagle-thriftmux_2.9.2 Show documentation

finagle-thriftmux

The newest version!

package com.twitter.finagle

import org.jboss.netty.buffer.{ChannelBuffer => CB}

private object ThriftMuxUtil {
  val role = Stack.Role("ProtocolRecorder")
  def bufferToArray(buf: CB): Array[Byte] =
    if (buf.hasArray && buf.arrayOffset == 0
        && buf.readableBytes == buf.array().length) {
      buf.array()
    } else {
      val arr = new Array[Byte](buf.readableBytes)
      buf.readBytes(arr)
      arr
    }

  def classForName(name: String) =
    try Class.forName(name) catch {
      case cause: ClassNotFoundException =>
        throw new IllegalArgumentException("Iface is not a valid thrift iface", cause)
    }

  val protocolRecorder = new Stack.Simple[ServiceFactory[CB, CB]] {
    val role = ThriftMuxUtil.role
    val description = "Record ThriftMux protocol usage"
    def make(next: ServiceFactory[CB, CB])(implicit params: Stack.Params) = {
      val param.Stats(stats) = params[param.Stats]
      stats.scope("protocol").counter("thriftmux").incr()
      next
    }
  }
}