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()
 }