巧妙解决命令行字符串参数到自定义结构体的转换难题
本文提供一种基于Go语言map的高效方案,解决将命令行参数(字符串类型)转换为自定义结构体类型的问题。 假设程序接收-m参数,其值(例如“h1”)代表需要获取特定poc结构体信息,而getpocinfo函数需要poc类型参数。 直接转换字符串到结构体不可行,因此需要建立字符串标识符与poc结构体实例的映射。
我们使用map[string]Poc来存储字符串标识符和对应poc结构体实例。 以下代码演示了该方法:
package mainimport ( "flag" "fmt")type Poc struct { method string headers headers path string body string expression int}type headers struct { UserAgent string Accept string XForwardedFor string ContentType string Referer string AcceptLanguage string Cookie string}func main() { var module string flag.StringVar(&module, "m", "H1", "module: all") flag.Parse() pocMap := map[string]Poc{ "H1": {method: "GET", path: "/resin-doc/viewfile/?file=index.jsp", body: "", expression: 200, headers: headers{UserAgent: "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36"}}, // ... 添加更多Poc结构体实例 } poc, ok := pocMap[module] if !ok { fmt.Println("无效的模块名称") return } method, path, header, body := getpocinfo(poc) fmt.Println(method + ":" + path + ":" + header + ":" + body)}func getpocinfo(id Poc) (method, path, header, body string) { method = id.method path = id.path header = id.headers.UserAgent body = id.body return}
登录后复制
此方案通过pocMap快速查找对应的poc结构体,避免了不安全的直接类型转换。 如果module值不在pocMap中,程序会输出错误信息,保证了程序的健壮性。 这种方法比直接字符串转换更可靠、更清晰。
以上就是如何高效地将命令行字符串参数转换为自定义结构体?的详细内容,更多请关注【创想鸟】其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至253000106@qq.com举报,一经查实,本站将立刻删除。
发布者:PHP中文网,转转请注明出处:https://www.chuangxiangniao.com/p/2539423.html