1个不稳定版本
0.6.1 | 2024年3月21日 |
---|
435 在 编程语言 中
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.rs
、lib.rs
或mod.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
字段设置为date
或date-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.rs
和 Generator
函数 cli
和 httpmock
分别实现。
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.rs
、lib.rs
或 mod.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"
(如上所述的 chrono
、uuid
、base64
和 rand
)
请注意,虽然 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