TcpStream::connect匹配分支类型不兼容问题求助

问题场景 #

在用Rust编写基础网络代码时，处理TcpStream::connect()的Result类型时遇到了match分支类型不兼容的错误，代码示例如下：

use std::net::TcpStream;

fn main() {
    let mut server = match TcpStream::connect("127.0.0.1:23456"){
        Ok(x) => x,
        Err(x) => println!("Could not connect to server: {x}"),
    };
}

编译器报错信息：

error[E0308]: `match` arms have incompatible types
 --> src/main.rs:8:19
  |
6 |       let mut server = match TcpStream::connect("127.0.0.1:23456"){
  |  ______________________-
7 | |         Ok(x) => x,
  | |                  - this is found to be of type `TcpStream`
8 | |         Err(x) => println!("Could not connect to server: {x}"),
  | |                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ 
                        expected struct `TcpStream`, found `()`
9 | |     };
  | |_____- `match` arms have incompatible types
  |

问题原因 #

Rust的match表达式要求所有分支返回完全相同的类型。你将match的结果赋值给了server变量，该变量的类型由Ok分支的TcpStream确定，但Err分支中println!()的返回值是单元类型()，两者类型不匹配，因此编译器报错。

你之前可能处理的是无需将match结果赋值给变量的场景——比如直接在match分支内完成所有逻辑，不需要后续使用返回值，这种情况下Err分支不需要返回和Ok分支相同的类型。但现在你需要把结果赋值给变量，所有分支必须返回同一类型，或者在Err分支直接终止程序（此时不需要返回值）。

解决方案 #

1. 在Err分支终止程序 #

如果连接失败后无需继续执行后续逻辑，可以在Err分支中打印错误并终止程序，这样编译器就不会要求分支返回相同类型：

use std::net::TcpStream;
use std::process;

fn main() {
    let mut server = match TcpStream::connect("127.0.0.1:23456"){
        Ok(x) => x,
        Err(x) => {
            eprintln!("Could not connect to server: {x}");
            process::exit(1);
        },
    };
    // 后续可正常使用server
}

也可以用unwrap_or_else简化写法：

use std::net::TcpStream;
use std::process;

fn main() {
    let mut server = TcpStream::connect("127.0.0.1:23456")
        .unwrap_or_else(|err| {
            eprintln!("Could not connect to server: {err}");
            process::exit(1);
        });
}

2. 让main函数返回Result #

将main函数的返回类型改为Result<(), Box<dyn std::error::Error>>，可以用?操作符自动处理错误，无需手动编写match：

use std::net::TcpStream;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let mut server = TcpStream::connect("127.0.0.1:23456")?;
    // 执行后续网络操作
    Ok(())
}

这种方式下，若connect失败，?会自动将错误向上返回，由Rust运行时输出错误信息并终止程序。

3. 包装为Option类型 #

如果连接失败后需要继续执行其他逻辑，可以将结果包装为Option<TcpStream>，Err分支返回None：

use std::net::TcpStream;

fn main() {
    let server = match TcpStream::connect("127.0.0.1:23456"){
        Ok(x) => Some(x),
        Err(x) => {
            println!("Could not connect to server: {x}");
            None
        },
    };
    
    if let Some(mut stream) = server {
        // 处理连接成功的逻辑
    } else {
        // 处理连接失败的后续逻辑
    }
}

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