subscriber: Add MakeWriter impl for Option

tracing-subscriber/src/fmt/writer.rs

@@ -1095,6 +1095,27 @@ where
     }
 }
 
+impl<'a, M> MakeWriter<'a> for Option<M>
+where
+    M: MakeWriter<'a> + 'static,
+{
+    type Writer = OptionalWriter<M::Writer>;
+
+    fn make_writer(&'a self) -> Self::Writer {
+        match self {
+            Some(inner) => OptionalWriter::some(inner.make_writer()),
+            None => OptionalWriter::none(),
+        }
+    }
+
+    fn make_writer_for(&'a self, meta: &Metadata<'_>) -> Self::Writer {
+        match self {
+            Some(inner) => OptionalWriter::some(inner.make_writer_for(meta)),
+            None => OptionalWriter::none(),
+        }
+    }
+}
+
 // === impl WriteAdaptor ===
 
 #[cfg(any(feature = "json", feature = "time"))]
@@ -1391,4 +1412,80 @@ mod test {
         has_lines(&a_buf, &lines[..]);
         has_lines(&b_buf, &lines[..]);
     }
+
+    #[test]
+    fn option_some_makewriter() {
+        let buf = Arc::new(Mutex::new(Vec::new()));
+        let make_writer = Some(MockMakeWriter::new(buf.clone()));
+
+        let lines = &[(Level::INFO, "hello"), (Level::INFO, "world")];
+        let c = {
+            #[cfg(feature = "ansi")]
+            let f = Format::default().without_time().with_ansi(false);
+            #[cfg(not(feature = "ansi"))]
+            let f = Format::default().without_time();
+            Collector::builder()
+                .event_format(f)
+                .with_writer(make_writer)
+                .with_max_level(Level::TRACE)
+                .finish()
+        };
+        let _s = tracing::collect::set_default(c);
+        info!("hello");
+        info!("world");
+        has_lines(&buf, &lines[..]);
+    }
+
+    #[test]
+    fn option_none_makewriter() {
+        let make_writer = Option::<MockMakeWriter>::None;
+
+        let c = {
+            #[cfg(feature = "ansi")]
+            let f = Format::default().without_time().with_ansi(false);
+            #[cfg(not(feature = "ansi"))]
+            let f = Format::default().without_time();
+            Collector::builder()
+                .event_format(f)
+                .with_writer(make_writer)
+                .with_max_level(Level::TRACE)
+                .finish()
+        };
+        let _s = tracing::collect::set_default(c);
+        info!("hello");
+        info!("world");
+    }
+
+    #[test]
+    fn multi_tee() {
+        let always_buf = Arc::new(Mutex::new(Vec::new()));
+        let some_buf = Arc::new(Mutex::new(Vec::new()));
+
+        let always_make_writer = MockMakeWriter::new(always_buf.clone());
+        let some_make_writer = Some(MockMakeWriter::new(some_buf.clone()));
+        let none_make_writer = Option::<MockMakeWriter>::None;
+
+        let make_writer = always_make_writer
+            .and(some_make_writer)
+            .and(none_make_writer);
+
+        let lines = &[(Level::INFO, "hello"), (Level::INFO, "world")];
+        let c = {
+            #[cfg(feature = "ansi")]
+            let f = Format::default().without_time().with_ansi(false);
+            #[cfg(not(feature = "ansi"))]
+            let f = Format::default().without_time();
+            Collector::builder()
+                .event_format(f)
+                .with_writer(make_writer)
+                .with_max_level(Level::TRACE)
+                .finish()
+        };
+        let _s = tracing::collect::set_default(c);
+        info!("hello");
+        info!("world");
+
+        has_lines(&always_buf, &lines[..]);
+        has_lines(&some_buf, &lines[..]);
+    }
 }