scalagen.finagleClientFunction.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scrooge-generator Show documentation

The newest version!

private[this] object {{__stats_name}} {
  val RequestsCounter = scopedStats.scope("{{clientFuncNameForWire}}").counter("requests")
  val SuccessCounter = scopedStats.scope("{{clientFuncNameForWire}}").counter("success")
  val FailuresCounter = scopedStats.scope("{{clientFuncNameForWire}}").counter("failures")
  val FailuresScope = scopedStats.scope("{{clientFuncNameForWire}}").scope("failures")
}

{{#headerInfo}}{{>header}}{{/headerInfo}} = {
  {{__stats_name}}.RequestsCounter.incr()
  this.service(encodeRequest("{{clientFuncNameForWire}}", {{ArgsStruct}}({{argNames}}))) flatMap { response =>
    val result = decodeResponse(response, {{ResultStruct}})
    val exception =
{{#hasThrows}}
      ({{#throws}}result.{{throwName}}{{/throws| orElse }}).map(Future.exception)
{{/hasThrows}}
{{^hasThrows}}
      None
{{/hasThrows}}
{{#void}}
    exception.getOrElse(Future.Done)
{{/void}}
{{^void}}
    exception.orElse(result.success.map(Future.value)).getOrElse(Future.exception(missingResult("{{clientFuncNameForWire}}")))
{{/void}}
  } rescue {
    case ex: SourcedException => {
      if (this.serviceName != "") { ex.serviceName = this.serviceName }
      Future.exception(ex)
    }
  } onSuccess { _ =>
    {{__stats_name}}.SuccessCounter.incr()
  } onFailure { ex =>
    {{__stats_name}}.FailuresCounter.incr()
    {{__stats_name}}.FailuresScope.counter(ex.getClass.getName).incr()
  }
}