1个不稳定版本

0.6.1 2024年3月21日

435编程语言

MPL-2.0 许可证

250KB
5K SLoC

Progenitor

Progenitor是一个Rust包,用于从OpenAPI 3.0.x规范中的API描述生成具有特定观点的客户端。它使用Rust futures进行异步API调用,并使用Streams进行分页接口。

它生成一个名为Client的类型,其方法对应于OpenAPI文档中指定的操作。

Progenitor还可以生成一个CLI来与OpenAPI服务实例交互,并提供httpmock助手以创建强类型OpenAPI服务的模拟。

主要目标是Dropshot生成的API输出的OpenAPI文档,但也可以用于许多OpenAPI文档。由于OpenAPI涵盖了广泛的API,Progenitor可能无法处理某些OpenAPI文档。如果您遇到问题,可以通过提交包含产生问题的OpenAPI文档的问题来帮助该项目。

使用Progenitor

有三种不同的方式使用progenitor包。您选择的方式将取决于您的用例和偏好。

使用Progenitor最简单的方法是通过它的generate_api!宏。

在源文件(通常是main.rslib.rsmod.rs)中简单地调用宏

generate_api!("path/to/openapi_document.json");

您需要将以下内容添加到Cargo.toml

[dependencies]
futures = "0.3"
progenitor = { git = "https://github.com/oxidecomputer/progenitor" }
reqwest = { version = "0.11", features = ["json", "stream"] }
serde = { version = "1.0", features = ["derive"] }

此外,如果OpenAPI文档包含字符串类型,并且format字段设置为datedate-time,则包括

[dependencies]
chrono = { version = "0.4", features = ["serde"] }

类似地,如果有一个format字段设置为uuid

[dependencies]
uuid = { version = "1.0.0", features = ["serde", "v4"] }

并且存在任何WebSocket通道端点

[dependencies]
base64 = "0.21"
rand = "0.8"

如果类型包含正则表达式验证

[dependencies]
regress = "0.4.1"

该宏有一些额外的选项来控制生成的代码

generate_api!(
    spec = "path/to/openapi_document.json",      // The OpenAPI document
    interface = Builder,                         // Choose positional (default) or builder style
    tags = Separate,                             // Tags may be Merged or Separate (default)
    inner_type = my_client::InnerType,           // Client inner type available to pre and post hooks
    pre_hook = closure::or::path::to::function,  // Hook invoked before issuing the HTTP request
    post_hook = closure::or::path::to::function, // Hook invoked prior to receiving the HTTP response
    derives = [ schemars::JsonSchema ],          // Additional derive macros applied to generated types
);

请注意,当OpenAPI文档的spec更改时(当其mtime更新时),宏将被重新评估。

build.rs

Progenitor 包含一个适用于在 build.rs 文件中使用的前端接口。虽然比宏稍微复杂一些,但构建器的好处在于可以直观地看到生成的代码。生成 CLI 和 httpmock 辅助工具的功能仅通过 build.rsGenerator 函数 clihttpmock 分别实现。

build.rs 文件应大致如下所示

fn main() {
    let src = "../sample_openapi/keeper.json";
    println!("cargo:rerun-if-changed={}", src);
    let file = std::fs::File::open(src).unwrap();
    let spec = serde_json::from_reader(file).unwrap();
    let mut generator = progenitor::Generator::default();

    let tokens = generator.generate_tokens(&spec).unwrap();
    let ast = syn::parse2(tokens).unwrap();
    let content = prettyplease::unparse(&ast);

    let mut out_file = std::path::Path::new(&std::env::var("OUT_DIR").unwrap()).to_path_buf();
    out_file.push("codegen.rs");

    std::fs::write(out_file, content).unwrap();
}

在源文件中(通常是 main.rslib.rsmod.rs)包含生成的代码

include!(concat!(env!("OUT_DIR"), "/codegen.rs"));

您需要将以下内容添加到Cargo.toml

[dependencies]
futures = "0.3"
progenitor-client = { git = "https://github.com/oxidecomputer/progenitor" }
reqwest = { version = "0.11", features = ["json", "stream"] }
serde = { version = "1.0", features = ["derive"] }

[build-dependencies]
prettyplease = "0.1.25"
progenitor = { git = "https://github.com/oxidecomputer/progenitor" }
serde_json = "1.0"
syn = "1.0"

(如上所述的 chronouuidbase64rand)

请注意,虽然 progenitor 被用于 build.rs,但所需的生成代码需要 progenitor-client

静态包

Progenitor 可以运行以生成生成的客户端的独立包。这确保了不会有意外变化(例如,来自对 Progenitor 的更新)。然而,这是使用 Progenitor 最手动的方法。

用法

cargo progenitor

Options:
    -i INPUT            OpenAPI definition document (JSON or YAML)
    -o OUTPUT           Generated Rust crate directory
    -n CRATE            Target Rust crate name
    -v VERSION          Target Rust crate version

例如

cargo install cargo-progenitor
cargo progenitor -i sample_openapi/keeper.json -o keeper -n keeper -v 0.1.0

...或者在本库中

cargo run --bin cargo-progenitor -- progenitor -i sample_openapi/keeper.json -o keeper -n keeper -v 0.1.0

这将在指定的目录中生成一个包。

选项 --license--registry-name 也可以用于在发布静态包之前改进元数据。

如果 Progenitor 从已发布的版本构建,默认情况下将使用已发布的 progenitor-client 包。但是,当使用从存储库构建的 Progenitor 时,默认情况下将自动将 progenitor-client 内联到静态包中。可以使用命令行标志 --include-client 来覆盖默认行为。

要确保输出没有对 Progenitor 的持续依赖,请启用 --include-client

以下是生成的 Cargo.toml 的摘录

[dependencies]
bytes = "1.3.0"
chrono = { version = "0.4.23", default-features=false, features = ["serde"] }
futures-core = "0.3.25"
percent-encoding = "2.2.0"
reqwest = { version = "0.11.13", default-features=false, features = ["json", "stream"] }
serde = { version = "1.0.152", features = ["derive"] }
serde_urlencoded = "0.7.1"

在生成的 Cargo.toml 中的依赖项版本与构建 Progenitor 时使用的版本相同。

请注意,存在对 percent-encoding 的依赖,并且宏和由 build.rs 生成的客户端是从 progenitor-client 中包含的。

生成样式

Progenitor 可以生成两种不同的接口样式:位置和构建器(如下所述)。选择只是偏好问题,许多偏好取决于 API 和个人品味。

位置(当前默认)

"位置"样式生成接受按顺序传递参数的 Client 方法,例如

impl Client {
    pub async fn instance_create<'a>(
        &'a self,
        organization_name: &'a types::Name,
        project_name: &'a types::Name,
        body: &'a types::InstanceCreate,
    ) -> Result<ResponseValue<types::Instance>, Error<types::Error>> {
        // ...
    }
}

调用者通过指定位置参数来调用此接口

let result = client.instance_create(org, proj, body).await?;

请注意,每个参数的类型必须精确匹配--没有隐式转换。

构建器

"构建器"样式生成生成构建器结构的 Client 方法。API 参数应用于该构建器,然后执行构建器(通过一个 send 方法)。代码更复杂,但更易于编写更简单、更易于阅读的消费者

impl Client
    pub fn instance_create(&self) -> builder::InstanceCreate {
        builder::InstanceCreate::new(self)
    }
}

mod builder {
    pub struct InstanceCreate<'a> {
        client: &'a super::Client,
        organization_name: Result<types::Name, String>,
        project_name: Result<types::Name, String>,
        body: Result<types::InstanceCreate, String>,
    }

    impl<'a> InstanceCreate<'a> {
        pub fn new(client: &'a super::Client) -> Self {
            // ...
        }

        pub fn organization_name<V>(mut self, value: V) -> Self
        where
            V: TryInto<types::Name>,
        {
            // ...
        }

        pub fn project_name<V>(mut self, value: V) -> Self
        where
            V: TryInto<types::Name>,
        {
            // ...
        }

        pub fn body<V>(mut self, value: V) -> Self
        where
            V: TryInto<types::InstanceCreate>,
        {
            // ...
        }

        pub async fn send(self) ->
            Result<ResponseValue<types::Instance>, Error<types::Error>>
        {
            // ...
        }
    }
}

请注意,与位置生成不同,消费者可以提供兼容的(而不是不可变的)参数

let result = client
    .instance_create()
    .organization_name("org")
    .project_name("proj")
    .body(body)
    .send()
    .await?;

字符串参数将隐式调用 TryFrom::try_from()。失败的转换或缺少必需的参数将导致从 send()

生成的 结构体 类型也有构建器,以便可以就地构造 body 参数

let result = client
    .instance_create()
    .organization_name("org")
    .project_name("proj")
    .body(types::InstanceCreate::builder()
        .name("...")
        .description("...")
        .hostname("...")
        .ncpus(types::InstanceCpuCount(4))
        .memory(types::ByteCount(1024 * 1024 * 1024)),
    )
    .send()
    .await?;

消费者无需指定不需要的参数或API指定默认值的结构体属性。真方便!

依赖项

~12–29MB
~429K SLoC