diff options
author | John St John <johnthesaintjohn@gmail.com> | 2017-09-26 17:03:32 -0700 |
---|---|---|
committer | John St John <johnthesaintjohn@gmail.com> | 2017-09-26 17:03:32 -0700 |
commit | c1628abaad79de60c7eafb27a5914dabd3fb75dc (patch) | |
tree | e38728e9a5423152b34e464a974c8b161a45334e /src/main/scala/xyz/driver/core/trace/LoggingTrace.scala | |
parent | e07838aab7b059a37bfaa8ff23335cc6e1c9867e (diff) | |
download | driver-core-c1628abaad79de60c7eafb27a5914dabd3fb75dc.tar.gz driver-core-c1628abaad79de60c7eafb27a5914dabd3fb75dc.tar.bz2 driver-core-c1628abaad79de60c7eafb27a5914dabd3fb75dc.zip |
only push traces every 10 requests by default
Diffstat (limited to 'src/main/scala/xyz/driver/core/trace/LoggingTrace.scala')
-rw-r--r-- | src/main/scala/xyz/driver/core/trace/LoggingTrace.scala | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala b/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala index e75de53..cd920f0 100644 --- a/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala +++ b/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala @@ -4,7 +4,8 @@ import akka.http.scaladsl.model.HttpRequest import com.google.cloud.trace.v1.consumer.TraceConsumer import com.typesafe.scalalogging.Logger -final class LoggingTrace(appName: String, appEnvironment: String, log: Logger) extends GoogleServiceTracer { +final class LoggingTrace(appName: String, appEnvironment: String, log: Logger, bufferSize: Int) + extends GoogleServiceTracer { private val traceConsumer: TraceConsumer = new LoggingTraceConsumer(log) private val googleServiceTracer = new GoogleStackdriverTraceWithConsumer( @@ -12,11 +13,14 @@ final class LoggingTrace(appName: String, appEnvironment: String, log: Logger) e appName, appEnvironment, traceConsumer, - log + log, + bufferSize ) override def startSpan(httpRequest: HttpRequest): GoogleStackdriverTraceSpan = googleServiceTracer.startSpan(httpRequest) override def endSpan(span: GoogleStackdriverTraceSpan): Unit = googleServiceTracer.endSpan(span) + + override def flush(): Unit = googleServiceTracer.flush() } |