diff options
Diffstat (limited to 'src/main/scala/xyz/driver/core/trace/LoggingTrace.scala')
-rw-r--r-- | src/main/scala/xyz/driver/core/trace/LoggingTrace.scala | 9 |
1 files changed, 7 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 9db85b7..cd920f0 100644 --- a/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala +++ b/src/main/scala/xyz/driver/core/trace/LoggingTrace.scala @@ -4,18 +4,23 @@ 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( "logging-tracer", appName, appEnvironment, - traceConsumer + traceConsumer, + 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() } |