rust下根据protobuf的消息名创建对象实例

系统管理员 2022-09-11 03:19 142阅读 0赞

在C++里面, 我们可以根据一个消息的名称, 动态的创建一个实例

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 </td> 
 <td> <code>google::protobuf::Descriptor* desc =</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>google::protobuf::DescriptorPool::generated_pool()</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>-&gt;FindMessageTypeByName(</code><code>"mypkg.MyType"</code><code>);</code> <code>google::protobuf::Message* message =</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>google::protobuf::MessageFactory::generated_factory()</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>-&gt;GetPrototype(desc)-&gt;New();</code> </td> 
 </tr> 
 </tbody> 
</table>

这个在protobuf里面是集成进去了, 在其他语言也有类似的东西.

通过这个, 我们就让轻松实现编解码库, 而不需去构造一个映射表.

但是, 但是在rust里面, 是没有这种东西的. 比较难的地方是rust全局变量必须要实现Send trait, 否则是不能被共享的, 这样做确实安全, 但是对于我们实现MessageFactory就变得困难.

好在rust有thread\_local和build.rs, 我们可以通过build.rs在编译proto文件的时候去遍历, 把每个消息添加到一个thread\_local的hash map里面去, 从而曲线救国.

不说实现细节, 可以自己去看源码, 这边说如何使用和集成.

1. 创建一个子工程, 名字叫proto

然后将依赖添加进去:

[dependencies]
    protobuf = "2.8.0"
    
    [build-dependencies]
    protoc-rust = "2.8.0"
    protobuf_message_factory = "0.1.2"

2. 把所有的**.proto**文件都添加到src目录下面去

3. 添加一个build.rs文件

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 </td> 
 <td> <code>extern</code>&nbsp;<code>crate protobuf_message_factory;</code> <code>use protobuf_message_factory::*;</code> <code>...</code> <code>fn main() { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>let proto_path =&nbsp;</code><code>"src/"</code><code>;</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>let proto_files = get_protos_info(proto_path);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>let proto_messages = get_proto_list(&amp;proto_files);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>//!!!&nbsp;&nbsp; this is importent.&nbsp;&nbsp; !!!</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>protoc_rust::run(protoc_rust::Args { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>out_dir: proto_path,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>input: &amp;protos,</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>includes: &amp;[proto_path],</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>customize: Customize { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>..Default::</code><code>default</code><code>()</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>},</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>}).expect(</code><code>"protoc"</code><code>);</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>//now generate factory codes</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>generate_factory_file(proto_path, &amp;proto_files);</code> <code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

然后把build.rs添加到toml里面去

4. 到主工程里面去, 添加对proto工程的依赖

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 </td> 
 <td> <code>[dependencies]</code> <code>proto = {version=</code><code>"^0"</code><code>, path=</code><code>"proto"</code><code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

这时候, 就可以在主工程里面使用proto了

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 </td> 
 <td> <code>extern</code>&nbsp;<code>crate proto;</code> <code>use proto::factory::*;</code> <code>use proto::rpc::*;</code> <code>use local_ipaddress;</code> <code>fn main() { </code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>let desc = get_descriptor(</code><code>"mypkg.MyType"</code><code>.to_string()).unwrap();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>println!(</code><code>"{}"</code><code>, desc.full_name());</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>let msg = desc.new_instance();</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>println!(</code><code>"msg: {:?}"</code><code>, msg);</code> <code>}</code> </td> 
 </tr> 
 </tbody> 
</table>

cargo run:

就可以看到

mypkg.MyType

msg:

这时候就可以拿到MessageDesriptor, 通过这个对象可以new一个instance

仓库在这里[https://crates.io/crates/protobuf\_message\_factory][https_crates.io_crates_protobuf_message_factory]

关键字: MessageFactory, Protobuf, Rust

[https_crates.io_crates_protobuf_message_factory]: https://crates.io/crates/protobuf_message_factory