编写 function 接受两种参数类型但做同样的事情

Question

这是我的src/main.rs文件：

use chrono_tz::Tz;
use chrono::FixedOffset;
use chrono::{NaiveDateTime, TimeZone, NaiveDate};


fn my_func(from_tz: Tz, ndt: NaiveDateTime){
    let res = from_tz.from_local_datetime(&ndt);
    println!("res: {:?}", res);
}

fn main() {
    let area_location: Tz = "UTC".parse().unwrap();
    let hour = 3600;
    let fixed_offset: FixedOffset = FixedOffset::east_opt(5 * hour).unwrap();
    let ndt = NaiveDate::from_ymd_opt(2038, 1, 19).unwrap().and_hms_opt(3, 14, 08).unwrap();
    my_func(area_location, ndt)
}

和Cargo.toml ：

[package]
name = "tmp"
version = "0.1.0"
edition = "2021"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
chrono = "0.4.23"
chrono-tz = "0.8.1"
regex = "1.7.1"

代码运行良好并打印

res: Single(2038-01-19T03:14:08UTC)

但是，我也可以将main的最后一行更改为my_func(fixed_offset, ndt) ，并将my_func中from_tz的类型更改为FixedOffset ，代码仍然可以正常运行，打印

res: Single(2038-01-19T03:14:08+05:00)

在我的应用程序中，我不知道我会收到Tz还是FixedOffset 。 无论哪种方式，我都想将其传递给my_func ，并在其中执行from_tz.from_local_datetime(&ndt) 。

我怎样才能让my_func接受Tz或FixedOffset ？

Answer 1

在您的情况下，所讨论的方法具有TimeZone特征，因此您可以只接受绑定到该特征的通用参数：

fn my_func<T: TimeZone>(from_tz: T, ndt: NaiveDateTime){
    let res = from_tz.from_local_datetime(&ndt);
    println!("res: {:?}", res);
}