diff options
author | John St John <johnthesaintjohn@gmail.com> | 2017-09-26 15:51:38 -0700 |
---|---|---|
committer | John St John <johnthesaintjohn@gmail.com> | 2017-09-26 15:51:38 -0700 |
commit | e07838aab7b059a37bfaa8ff23335cc6e1c9867e (patch) | |
tree | e7e7502163e5965bbfe9688255e0a03f254de7be /src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala | |
parent | 26a5eb401c48c08812f1292664848595e7f74806 (diff) | |
download | driver-core-e07838aab7b059a37bfaa8ff23335cc6e1c9867e.tar.gz driver-core-e07838aab7b059a37bfaa8ff23335cc6e1c9867e.tar.bz2 driver-core-e07838aab7b059a37bfaa8ff23335cc6e1c9867e.zip |
Catch and log google tracing exceptions rather than crashing the entire app
Diffstat (limited to 'src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala')
-rw-r--r-- | src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala b/src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala new file mode 100644 index 0000000..534df8c --- /dev/null +++ b/src/main/scala/xyz/driver/core/trace/ExceptionLoggingSizedBufferingTraceConsumer.scala @@ -0,0 +1,26 @@ +package xyz.driver.core.trace + +import com.google.cloud.trace.v1.consumer.{FlushableTraceConsumer} +import com.google.devtools.cloudtrace.v1.Traces +import com.typesafe.scalalogging.Logger +import scala.util.Try + +/** + * ExceptionLoggingFlushableTraceConsumer simply wraps a flushable trace consumer and catches/logs any exceptions + * @param traceConsumer the flusable trace consumer to wrap + * @param log where to log any exceptions + */ +class ExceptionLoggingFlushableTraceConsumer(traceConsumer: FlushableTraceConsumer, log: Logger) + extends FlushableTraceConsumer { + + private val flushableTraceConsumer = traceConsumer + + private def exceptionLogger(exception: Throwable): Unit = + log.trace(s"Encountered exception logging to google $exception") + + override def receive(trace: Traces): Unit = + Try(flushableTraceConsumer.receive(trace)).fold(exceptionLogger, identity) + + override def flush(): Unit = + Try(flushableTraceConsumer.flush()).fold(exceptionLogger, identity) +} |