打造灵活高效的命令行工具:从待办清单到 Markdown 预览
1. 利用环境变量提升待办工具的灵活性
待办工具已经有了不少实用功能,但用户还不能自行选择保存待办事项列表的文件。为了让工具更灵活,我们可以使用环境变量。
在 Go 语言里, os 包提供了处理环境和环境变量的函数。我们会用 os.Getenv("TODO_FILENAME") 来获取名为 TODO_FILENAME 的环境变量的值。
要给待办工具添加这个功能,需要对 main.go 文件做两处修改:
- 把定义 todoFileName 的行从常量改成变量,这样在定义了环境变量时就可以修改它。当用户没设置环境变量时,这行代码就是默认的文件名:
// Default file name
var todoFileName = ".todo.json"
- 在
main函数里,在实例化todo.List类型之前添加以下代码。目的是检查是否设置了环境变量TODO_FILENAME,如果设置了就把它的值赋给变量todoFileName,否则变量会保持默认值:
// Check if the user defined the ENV
超级会员免费看
订阅专栏 解锁全文
40

被折叠的 条评论
为什么被折叠?



