登录
首页 >  Golang >  Go问答

将命令行参数转换为用作结构字段

来源:stackoverflow

时间:2024-04-23 23:33:35 156浏览 收藏

golang学习网今天将给大家带来《将命令行参数转换为用作结构字段》,感兴趣的朋友请继续看下去吧!以下内容将会涉及到等等知识点,如果你是正在学习Golang或者已经是大佬级别了,都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家!

问题内容

我正在尝试打印一个 json(转换为结构)字段,特别是作为来自终端的输入。

type foo struct{ 
field1 string
field2 int
}

这是我创建的结构示例,我想将“field1”作为命令行参数,并通过到达解码的 json 输出将其打印到 html 正文或控制台。

通常我可以用这种方式打印它,但我找不到将 os.args[1] 值转换为已定义的结构体字段的方法。

var res *foo
_=json.newdecoder(resp.body).decode(&res)
fmt.println(res.field1)

总之,有没有办法将任何字符串变量转换为函数/方法。提前非常感谢

牧师。注意:例如。 我将写入终端:

go run main.go "field1"

程序会做这样的事情

fmt.Fprintf(writer, "option.field is %v", someFuncThatConcatenatesThisTwo("option."+ os.Args[1]))

顺便说一句,有多个字段需要调用。预定义当然可以解决特定情况,但我想学习的是其他东西。


正确答案


总之,有没有办法将任何字符串转换为函数/方法。

我不确定您想在这里实现什么目标。这没有任何意义。

一般来说,要使用命令行参数填充结构字段,您可以执行如下操作。

package main

import (
    "fmt"
    "log"
    "strconv"
    "os"
)

type options struct {
    field1 string
    field2 int64
}

func main() {
    if len(os.args) < 2 {
        log.fatal("missing two required positional arguments: field1 and field2")
    }

    opts := &options{}
    opts.field1 = os.args[1]

    var err error
    opts.field2, err = strconv.parseint(os.args[2], 10, 64)
    if err != nil {
        log.fatalf("failed to parse integer value: %v", os.args[2])
    }

    fmt.println(opts)
}

为了让您的生活变得更加轻松,您可以使用 flag(或 pflag)包将输入参数声明为命令行标志。

import (
    "flag"
    "fmt"
)

type options struct {
    field1 string
    field2 int
}

var opts options

func init() {
    flag.stringvar(&opts.field1, "field1", "", "help message for field1")
    flag.intvar(&opts.field2, "field2", 0, "help message for field2")
    flag.parse()
}

func main() {
   fmt.println(opts)
}

或者,就像 @jomaar 回答的那样,您可以使用像 go-arg 这样的帮助程序库来避免手动声明命令行标志。另一种选择是 go-flags

编辑

经过进一步澄清,您似乎希望使用编写器有选择地写入结构体的字段,并且希望使用位置命令行参数来指定要写入的字段。

我认为 map 将是在这种情况下更适合存储选项的数据结构,因为它允许您简单地使用 string 键引用字段。

import (
    "fmt"
    "os"
)

func main() {
    options := map[string]interface{} {
        "field1": "some-value",
        "field2": 1,
    }

    for _, arg := range os.args[1:] {
        if val, ok := options[arg]; ok {
            fmt.println(val)
        }
    }
}

如果您想继续使用结构体,可以使用 reflect 包。

import (
    "fmt"
    "os"
    "reflect"
)

type Options struct {
    Field1 string
    Field2 int
}

func main() {
    opts := &Options{Field1: "some-value", Field2: 1}

    for _, arg := range os.Args[1:] {
        fmt.Println(getAttr(opts, arg))
    }
}

// copied from https://stackoverflow.com/a/66470232/2410641
func getAttr(obj interface{}, fieldName string) (reflect.Value, error) {
    pointToStruct := reflect.ValueOf(obj) // addressable
    curStruct := pointToStruct.Elem()
    if curStruct.Kind() != reflect.Struct {
        return reflect.Value{}, fmt.Errorf("obj is not a struct")
    }
    curField := curStruct.FieldByName(fieldName) // type: reflect.Value
    if !curField.IsValid() {
        return reflect.Value{}, fmt.Errorf("field not found: %s", fieldName)
    }
    return curField, nil
}

go演示:https://play.golang.org/p/sch53l2bq4O

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于Golang的相关知识,也可关注golang学习网公众号。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>