登录
首页 >  Golang >  Go问答

我应该如何在 YAML 中编写和解组 byte64 编码值

来源:stackoverflow

时间:2024-04-10 16:27:34 354浏览 收藏

学习Golang要努力,但是不要急!今天的这篇文章《我应该如何在 YAML 中编写和解组 byte64 编码值》将会介绍到等等知识点,如果你想深入学习Golang,可以关注我!我会持续更新相关文章的,希望对大家都能有所帮助!

问题内容

我的 yaml 文件的一部分

rules:
- action:
    count: {}
  name: rulenumbertwo
  priority: 123
  statement:
    bytematchstatement:
      fieldtomatch:
        singleheader:
          name: foobar
      positionalconstraint: contains
      searchstring: [103, 105, 122, 122, 98, 117, 122, 122]

上面的结果是搜索字符串值的相应 ascii 字符串(“fizzbuzz”)。但我怎样才能把它写成人类可读的格式呢?手册说搜索字符串应该“自动”转换,但如果我不将其写为 ascii 代码数组,它会抛出错误。

我的代码(尝试解组它,如果我直接写入数组,它会起作用,但它不是人类可读的):

d := &wafv2.updaterulegroupinput{}
err = yaml.unmarshal(buf, d)

我想简单地编写yaml,如下

rules:
- action:
    count: {}
  name: rulenumbertwo
  priority: 123
  statement:
    bytematchstatement:
      fieldtomatch:
        singleheader:
          name: foobar
      positionalconstraint: contains
      searchstring: fizzbuzz

或者可能使用模板(但为什么它应该很容易)

searchstring: {{ convertStringToByteArray "fizzbuzz" }}

解决方案


包含 searchstring 的类型 bytematchstatement 没有声明任何自定义从 yaml 加载的方式,因此它无法从 yaml 标量加载 []byte。您链接的手册描述了该类型的语义,但既没有定义数据的存储方式(根据用例,将 string 的内容存储给用户作为 []byte 确实有意义),也没有定义数据的存储方式可以从 yaml 加载。一般来说,(反)序列化往往会暴露有关数据模型的事实,否则这些事实将是实现细节,这就是您在这里遇到的问题。

简单的修复方法是声明一些 type searchstringtype []byte,将其用于 searchstring 字段,并在那里声明一个自定义 unmarshalyaml 方法来处理到 []byte 的转换。但是,这在您的代码中是不可能的,因为您无法控制类型。

一个可能但有些粗糙的修复方法是预处理 yaml 输入:

type loadableupdaterulegroupinput wafv2.updaterulegroupinput

func (l *loadableupdaterulegroupinput) unmarshalyaml(n *yaml.node) error {
    preprocsearchstring(n)
    return n.decode((*wafv2.updaterulegroupinput)(l))
}

func preprocsearchstring(n *yaml.node) {
    switch n.kind {
    case yaml.sequencenode:
        for _, item := range n.content {
            preprocsearchstring(item)
        }
    case yaml.mappingnode:
        for i := 0; i < len(n.content); i += 2 {
            if n.content[i].kind == yaml.scalarnode &&
                n.content[i].value == "searchstring" {
                tobytesequence(n.content[i+1])
            } else {
                preprocsearchstring(n.content[i+1])
            }
        }
    }
}

func tobytesequence(n *yaml.node) {
    n.kind = yaml.sequencenode
    raw := []byte(n.value)
    n.content = make([]*yaml.node, len(raw))
    for i := range raw {
        n.content[i] = &yaml.node{kind: yaml.scalarnode, value: strconv.itoa(int(raw[i]))}
    }
}

通过在解组时使用此新子类型,searchstring 的 yaml 标量值将转换为您将编写的 yaml 序列节点。当您加载 yaml 节点时,它将正确加载到 []byte

d := &LoadableUpdateRuleGroupInput{}
err = yaml.Unmarshal(buf, d)

此代码需要 gopkg.in/yaml.v3v2 无法为您提供对解组的更多控制。

今天关于《我应该如何在 YAML 中编写和解组 byte64 编码值》的内容介绍就到此结束,如果有什么疑问或者建议,可以在golang学习网公众号下多多回复交流;文中若有不正之处,也希望回复留言以告知!

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