freeBuf
主站

分类

漏洞 工具 极客 Web安全 系统安全 网络安全 无线安全 设备/客户端安全 数据安全 安全管理 企业安全 工控安全

特色

头条 人物志 活动 视频 观点 招聘 报告 资讯 区块链安全 标准与合规 容器安全 公开课

官方公众号企业安全新浪微博

FreeBuf.COM网络安全行业门户,每日发布专业的安全资讯、技术剖析。

FreeBuf+小程序

FreeBuf+小程序

文盘Rust -- 配置文件解析
2022-08-05 18:13:19
所属地 北京

处理配置文件是应用开发的常规操作。成熟的开发语言都有自己处理配置文件的套路。golang 有 viper 这样的成熟第三方库来处理配置文件。rust 的第三方库并不成熟。

这篇文章我们来聊聊 rust 如何处理配置文件。

处理yaml配置文件的流程

配置文件的作用是一系列应用程序相应功能的开关。在应用启动前配置,应用启动时加载,以备运行时使用。

我们依旧用[interactcli-rs](https://github.com/jiashiwen/interactcli-rs) 为例,说明一下配置文件的处理过程。

解析配置文件的主要逻辑在 src/configure 目录。

定义 config 结构体

首先,定义一个结构体用来承载配置项。由于 Config struct 需要与yaml文件交互,我们定义一个具备序列化与反序列化能力的结构体。

```rust#[derive(Debug, PartialEq, Serialize, Deserialize, Clone)]pub struct Config {pub server: String,pub token: String,}```

为 Config 结构体定义必要的功能

```rustimpl Config {pub fn default() -> Self {Self {server: "http://127.0.0.1:8080".to_string(),token: "".to_string(),}}pub fn set_self(&mut self, config: Config) {self.server = config.server;self.token = config.token;}pub fn get_config_image(&self) -> Self {self.clone()}pub fn flush_to_file(&self, path: String) -> Result<()> {let yml = serde_yaml::to_string(&self)?;fs::write(path, yml)?;Ok(())}}```

利用 lazy_static 初始化配置项单例

```rustlazy_static::lazy_static! {static ref GLOBAL_CONFIG: Mutex<Config> = {let global_config = Config::default();Mutex::new(global_config)};static ref CONFIG_FILE_PATH: RwLock<String> = RwLock::new({let path = "".to_string();path});}```

加载配置文件

[interactcli-rs](https://github.com/jiashiwen/interactcli-rs) 是一个命令行程序。加载配置文件的策略为:当指定配置文件位置时,则按给定路径加载配置;如果未指定配置文件则按照默认路径加载,此时若默认配置文件不存在则终止程序。

src/cmd/rootcmd.rs 中的 cmd_match 函数包含上面的逻辑。


```rustfn cmd_match(matches: &ArgMatches) {if let Some(c) = matches.value_of("config") {set_config_file_path(c.to_string());set_config_from_file(&get_config_file_path());} else {set_config_from_file("");}......```

后记

手工处理配置文件还是比较繁琐。尤其在配置文件的书写上,必须明确配置每一个配置项,即使配置项为空也需填写。为了保证配置文件的配置项齐全,我们为Config struct 定义了 flush_to_file 函数,用来生成配置文件。

由于rust的生态较 golang 以及 java 的生态还很年轻,第三方的工具包不及两者完善。在配置文件的处理上比较繁琐,很多地方需要手工处理,但是已基本满足要求。

咱们下期见。

# 数据安全 # Rust # 配置文件 # YAML
本文为 独立观点,未经允许不得转载,授权请联系FreeBuf客服小蜜蜂,微信:freebee2022
被以下专辑收录,发现更多精彩内容
+ 收入我的专辑
+ 加入我的收藏
相关推荐
  • 0 文章数
  • 0 关注者
文章目录