From 52271527979cbd4c3e51a647fd6121186ec7ec50 Mon Sep 17 00:00:00 2001
From: Eliza Weisman <eliza@buoyant.io>
Date: Fri, 21 Sep 2018 14:43:27 -0700
Subject: [PATCH] split out sloggish formatter for future use in other examples

---
 tokio-trace/examples/sloggish.rs            |  12 +-
 tokio-trace/examples/sloggish_subscriber.rs | 128 ++++++++++++++++++++
 2 files changed, 137 insertions(+), 3 deletions(-)
 create mode 100644 tokio-trace/examples/sloggish_subscriber.rs

diff --git a/tokio-trace/examples/sloggish.rs b/tokio-trace/examples/sloggish.rs
index 4f5a542a..44e5ef31 100644
--- a/tokio-trace/examples/sloggish.rs
+++ b/tokio-trace/examples/sloggish.rs
@@ -60,7 +60,13 @@ impl SloggishSubscriber {
     {
         let mut kvs = kvs.into_iter();
         if let Some((k, v)) = kvs.next() {
-            write!(writer, "{}{}: {:?}", leading, Style::new().bold().paint(k), v)?;
+            write!(
+                writer,
+                "{}{}: {:?}",
+                leading,
+                Style::new().bold().paint(k),
+                v
+            )?;
         }
         for (k, v) in kvs {
             write!(writer, ", {}: {:?}", Style::new().bold().paint(k), v)?;
@@ -147,12 +153,12 @@ fn main() {
             peer1.enter(|| {
                 event!(Level::Warn, { algo = "xor" }, "weak encryption requested");
                 event!(Level::Debug, { length = 8 }, "response sent");
-                event!(Level::Debug, { }, "disconnected");
+                event!(Level::Debug, {}, "disconnected");
             });
             peer2.enter(|| {
                 event!(Level::Debug, { length = 5 }, "message received");
                 event!(Level::Debug, { length = 8 }, "response sent");
-                event!(Level::Debug, { }, "disconnected");
+                event!(Level::Debug, {}, "disconnected");
             });
             event!(Level::Error, {}, "internal error");
             event!(Level::Info, {}, "exit");
diff --git a/tokio-trace/examples/sloggish_subscriber.rs b/tokio-trace/examples/sloggish_subscriber.rs
new file mode 100644
index 00000000..d2d0d51c
--- /dev/null
+++ b/tokio-trace/examples/sloggish_subscriber.rs
@@ -0,0 +1,128 @@
+//! A simple example demonstrating how one might implement a custom
+//! subscriber.
+//!
+//! This subscriber implements a tree-structured logger similar to
+//! the "compact" formatter in [`slog-term`]. The demo mimicks the
+//! example output in the screenshot in the [`slog` README].
+//!
+//! Note that this logger isn't ready for actual production use.
+//! Several corners were cut to make the example simple.
+//!
+//! [`slog-term`]: https://docs.rs/slog-term/2.4.0/slog_term/
+//! [`slog` README]: https://github.com/slog-rs/slog#terminal-output-example
+use ansi_term::{Color, Style};
+use tokio_trace::Level;
+
+use std::{
+    fmt,
+    io::{self, Write},
+    sync::atomic::{AtomicUsize, Ordering},
+    time::SystemTime,
+};
+
+pub struct SloggishSubscriber {
+    indent: AtomicUsize,
+    indent_amount: usize,
+    stderr: io::Stderr,
+}
+
+struct ColorLevel(Level);
+
+impl fmt::Display for ColorLevel {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match self.0 {
+            Level::Trace => Color::Purple.paint("TRACE"),
+            Level::Debug => Color::Blue.paint("DEBUG"),
+            Level::Info => Color::Green.paint("INFO"),
+            Level::Warn => Color::Yellow.paint("WARN "),
+            Level::Error => Color::Red.paint("ERROR"),
+        }.fmt(f)
+    }
+}
+
+impl SloggishSubscriber {
+    pub fn new(indent_amount: usize) -> Self {
+        Self {
+            indent: AtomicUsize::new(0),
+            indent_amount,
+            stderr: io::stderr(),
+        }
+    }
+
+    fn print_kvs<'a, I>(&self, writer: &mut impl Write, kvs: I, leading: &str) -> io::Result<()>
+    where
+        I: IntoIterator<Item = (&'static str, &'a dyn tokio_trace::Value)>,
+    {
+        let mut kvs = kvs.into_iter();
+        if let Some((k, v)) = kvs.next() {
+            write!(
+                writer,
+                "{}{}: {:?}",
+                leading,
+                Style::new().bold().paint(k),
+                v
+            )?;
+        }
+        for (k, v) in kvs {
+            write!(writer, ", {}: {:?}", Style::new().bold().paint(k), v)?;
+        }
+        Ok(())
+    }
+
+    fn print_meta(
+        &self,
+        writer: &mut impl Write,
+        meta: &tokio_trace::StaticMeta,
+    ) -> io::Result<()> {
+        write!(
+            writer,
+            "{level} {target} ",
+            level = ColorLevel(meta.level),
+            target = meta.target.unwrap_or(meta.module_path),
+        )
+    }
+
+    fn print_indent(&self, writer: &mut impl Write) -> io::Result<usize> {
+        let indent = self.indent.load(Ordering::Acquire);
+        for _ in 0..indent {
+            write!(writer, " ")?;
+        }
+        Ok(indent)
+    }
+}
+
+impl tokio_trace::Subscriber for SloggishSubscriber {
+    #[inline]
+    fn observe_event<'event>(&self, event: &'event tokio_trace::Event<'event>) {
+        let mut stderr = self.stderr.lock();
+        self.print_indent(&mut stderr).unwrap();
+        write!(
+            &mut stderr,
+            "{} ",
+            humantime::format_rfc3339_seconds(event.timestamp)
+        ).unwrap();
+        self.print_meta(&mut stderr, event.static_meta).unwrap();
+        write!(
+            &mut stderr,
+            "{}",
+            Style::new().bold().paint(format!("{}", event.message))
+        ).unwrap();
+        self.print_kvs(&mut stderr, event.fields(), ", ").unwrap();
+        write!(&mut stderr, "\n").unwrap();
+    }
+
+    #[inline]
+    fn enter(&self, span: &tokio_trace::Span, _at: SystemTime) {
+        let mut stderr = self.stderr.lock();
+        let indent = self.print_indent(&mut stderr).unwrap();
+        self.print_kvs(&mut stderr, span.fields(), "").unwrap();
+        write!(&mut stderr, "\n").unwrap();
+        self.indent
+            .compare_and_swap(indent, indent + self.indent_amount, Ordering::Release);
+    }
+
+    #[inline]
+    fn exit(&self, _span: &tokio_trace::Span, _at: SystemTime) {
+        self.indent.fetch_sub(self.indent_amount, Ordering::Release);
+    }
+}