在 Rust 中使用 prost 和 tonic 时如何导入 .proto?

How to import .proto when working with prost and tonic in rust?

我正在尝试使用 crates prost 和 tonic 在 Rust 中构建一个简单的 gRPC 客户端。我的原型定义非常简单,但我突然坚持使用从其他原型导入的消息。

// file src/protos/types.proto
syntax = "proto3";
package Types;

message Message1 {
    uint32 value1 = 1;
    bytes value2 = 2;
}

message Message2 {
    uint32 value1 = 1;
    uint32 value2 = 2;
    uint32 value3 = 3;
    uint32 value4 = 4;
}
// file src/protos/service.proto
syntax = "proto3";
import "types.proto";
package Service;

service Worker {
    rpc Do (Request) returns (Reply);
}

message Request {
    Types.Message1 message1 = 1;
    Types.Message2 message2 = 2;
}

message Reply {
    bool success = 1;
}

我的build.rs很直白:

fn main() -> Result<(), Box<dyn std::error::Error>> {
    tonic_build::compile_protos("src/protos/types.proto")?;
    tonic_build::compile_protos("src/protos/service.proto")?;
    Ok(())
}

当我在 main.rs 中包含原型时,问题就开始了:

pub mod service {
    tonic::include_proto!("types");
    tonic::include_proto!("service");
}

编译失败,出现以下错误:

 --> D:\temp\rust-proto\target\debug\build\rust-proto-11c38604fbc7ce30\out/service.rs:4:48     
  |
4 |     pub message1: ::std::option::Option<super::types::Message1>,
  |                                                ^^^^^ maybe a missing crate `types`?

这里可能出了什么问题?!我将我的游乐场项目上传到 github 以防它可能有用。

对tonic了解不多,但是一定要把include放在proto包对应的modules中,例如:

pub mod types {
    tonic::include_proto!("types");
}
pub mod service {
    tonic::include_proto!("service");
}

fn main() {
    let msg = types::Message1::default();
    println!("Hello, world! {:?}", msg);
}

编译正确。

顺便说一句,您可以在此处的设置中的以下位置检查生成的 Rust 代码:

D:\temp\rust-proto\target\debug\build\rust-proto-11c38604fbc7ce30\out