如何让Go Cobra顶级命令的PersistentFlag仅在start子命令中必填？

场景说明 #

我们有个顶级命令prog，下面包含start和stop两个子命令，在prog里定义了一个叫database的持久化Flag（PersistentFlag）。现在要实现：这个Flag在start命令里是必填项，在stop命令里可选填。

踩过的坑 #

1. 直接在顶级命令标记必填
   使用progCmd.MarkPersistentFlagRequired("database")的话，start和stop两个子命令都会强制要求传入database，不符合仅限制start的需求。
2. 直接在start命令标记必填
   执行startCmd.MarkPersistentFlagRequired("database")会报错：

   no such flag -database
   

   原因是database是父命令prog的持久化Flag，start命令只是继承了它的使用权限，但自身的Flag集合里并没有这个Flag的本地引用，直接标记会找不到对应的Flag。

正确实现方式 #

先把父级的databaseFlag添加到start命令的本地Flag集合中，再标记为必填即可，代码示例：

// 从父命令获取database持久化Flag
dbFlag := progCmd.PersistentFlags().Lookup("database")
// 将该Flag加入start命令的本地Flag集合
startCmd.Flags().AddFlag(dbFlag)
// 标记start命令的database为必填
startCmd.MarkFlagRequired("database")

或者更简洁的写法，无需单独引用父命令对象：

startCmd.Flags().AddFlag(startCmd.Parent().PersistentFlags().Lookup("database"))
startCmd.MarkFlagRequired("database")

处理后效果：

• 运行prog start时必须携带--database参数，否则会触发必填校验报错
• 运行prog stop时可选择携带或不携带--database参数，不影响命令执行

内容的提问来源于stack exchange，提问作者thiagowfx