Rust如何存储共享trait的stdout、sink常量 解决if分支类型不兼容问题

Rust是静态强类型语言，要求if表达式的所有分支返回值类型完全一致。std::io::stdout()返回std::io::Stdout类型，std::io::sink()返回std::io::Sink类型，两者是不同的结构体，自然会报类型不兼容错误。

根据你的场景需求，可以选择以下三种方案：

方案1：特征对象动态分发（最通用，推荐绝大多数场景使用） #

如果只需要用到std::io::Write特征的通用写入能力，直接把两个值装箱成Box<dyn Write>特征对象即可，完全屏蔽底层实现类型，符合你要的抽象设计要求，运行时只有极小的动态分发开销，几乎可以忽略：

use std::io::Write;

let mut out: Box<dyn Write> = if std::env::var("LOG").is_ok() {
    Box::new(std::io::stdout())
} else {
    Box::new(std::io::sink())
};

// 后续可以正常调用所有Write特征的方法
writeln!(out, "测试输出").unwrap();

方案2：自定义枚举静态分发（零开销，适合性能敏感场景） #

如果完全不想付出堆分配和动态分发的开销，可以自己定义枚举覆盖所有可能的写入类型，手动实现Write特征转发：

use std::io::{Write, Stdout, Sink, Result};

enum Output {
    Stdout(Stdout),
    Sink(Sink),
}

impl Write for Output {
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
        match self {
            Output::Stdout(s) => s.write(buf),
            Output::Sink(s) => s.write(buf),
        }
    }

    fn flush(&mut self) -> Result<()> {
        match self {
            Output::Stdout(s) => s.flush(),
            Output::Sink(s) => s.flush(),
        }
    }
}

// 使用示例
let mut out = if std::env::var("LOG").is_ok() {
    Output::Stdout(std::io::stdout())
} else {
    Output::Sink(std::io::sink())
};

writeln!(out, "测试输出").unwrap();

方案3：第三方库通用枚举（无需手动实现特征，依赖额外库） #

如果项目已经引入了either库，可以直接使用通用Either类型，它会自动为两个实现了相同特征的类型转发特征实现：

use either::Either;
use std::io::Write;

let mut out = if std::env::var("LOG").is_ok() {
    Either::Left(std::io::stdout())
} else {
    Either::Right(std::io::sink())
};

writeln!(out, "测试输出").unwrap();

内容的提问来源于stack exchange，提问作者Martin Cowie