Go YAML 解析：必填字段

从现在开始，努力学习吧！本文《Go YAML 解析：必填字段》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

摘要： 我需要将 yaml 格式的数据解析为 golang 结构。是否有一种方法（库、属性）使某些字段成为必需字段，即如果某些字段不存在，则使 unmarshal 函数返回错误？

示例想要什么： 此代码中的解组函数应引发错误，因为输入数据不包含“b”字段。

package main

import (
    "fmt"
    "gopkg.in/yaml.v2"
)

type TestStruct struct {
    FieldA string `yaml:"a"`
    FieldB string `yaml:"b"`
}

func main() {
    input := []byte(`{a: 1}`)
    var output TestStruct

    _ = yaml.Unmarshal(input, &output)
}

解决方案

您可以使用此 library 的 haszero 方法来检查结构中是否存在缺失值。这将返回 true 或 false，具体取决于结构是否完全填充。请参阅演示 example 来了解一下。

但是如果您特别需要知道缺少哪个字段，则需要检查该值是否为 nil，如下例所示。

package main
import (
   "fmt"
   "errors"
  "gopkg.in/yaml.v2"
)

type TestStruct struct {
  FieldA  string `yaml:"a"`
  FieldB  string `yaml:"b"`
}

func main() {
  input := []byte(`{a: 1}`)

  var output TestStruct 
  if err := output.ParseFromFile(input); err != nil {
     fmt.Println(err)
  }
  fmt.Println(output)     
}

func (output *TestStruct) ParseFromFile(data []byte) error {

  if err := yaml.Unmarshal(data, output); err != nil {
    return err
  }

  if output.FieldA == "" {
    return errors.New("Blank Field A")
  }
  if output.FieldB == "" {
    return errors.New("Blank Field B")
  }

  return nil
}

演示example如果需要专门返回错误

理论要掌握，实操不能落！以上关于《Go YAML 解析：必填字段》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！