Go中如何解析YAML字符串提取files与folders列表？

问题核心 #

你遇到的问题本质是YAML结构存在层级嵌套，直接用yaml.Unmarshal解析到[]string必然失败——目标数据在project节点下，并非顶层列表。下面提供两种可行方案：

方案1：定义结构体解析（推荐） #

这是Go处理结构化配置最规范的方式，只需定义匹配YAML层级的结构体，就能精准提取目标字段：

package main

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

// 定义匹配YAML结构的结构体，无关字段可忽略
type Config struct {
	Project struct {
		Files   []string `yaml:"files"`
		Folders []string `yaml:"folders"`
	} `yaml:"project"`
}

func main() {
	// 模拟收到的JSON响应
	jsonResp := []byte(`{"config.yml": "project: 
  files: 
    - file1
    - file2
  folders: 
    - folder1
    - folder2
  random1: 
  random2:
    - redundant1"}`)

	// 第一步：解析JSON，提取YAML字符串
	var jsonMap map[string]string
	if err := json.Unmarshal(jsonResp, &jsonMap); err != nil {
		panic(err)
	}
	yamlStr := jsonMap["config.yml"]

	// 第二步：解析YAML到结构体
	var config Config
	if err := yaml.Unmarshal([]byte(yamlStr), &config); err != nil {
		panic(err)
	}

	// 提取目标列表
	fmt.Println("Files:", config.Project.Files)
	fmt.Println("Folders:", config.Project.Folders)
}

优势 #

• 类型安全，编译期即可发现字段匹配错误
• 代码可读性高，结构逻辑清晰
• 自动忽略无关字段（如示例中的random1、random2）

方案2：用嵌套Map解析（灵活场景） #

如果不想定义结构体，也可以将YAML解析为嵌套map[string]interface{}，逐层断言取值：

package main

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

func main() {
	jsonResp := []byte(`{"config.yml": "project: 
  files: 
    - file1
    - file2
  folders: 
    - folder1
    - folder2
  random1: 
  random2:
    - redundant1"}`)

	// 提取YAML字符串
	var jsonMap map[string]string
	if err := json.Unmarshal(jsonResp, &jsonMap); err != nil {
		panic(err)
	}
	yamlStr := jsonMap["config.yml"]

	// 解析YAML到嵌套Map
	var yamlMap map[string]interface{}
	if err := yaml.Unmarshal([]byte(yamlStr), &yamlMap); err != nil {
		panic(err)
	}

	// 逐层断言取值
	project, ok := yamlMap["project"].(map[interface{}]interface{})
	if !ok {
		panic("invalid project structure")
	}

	// 提取files列表
	files := make([]string, 0)
	if filesRaw, ok := project["files"].([]interface{}); ok {
		for _, f := range filesRaw {
			if s, ok := f.(string); ok {
				files = append(files, s)
			}
		}
	}

	// 提取folders列表
	folders := make([]string, 0)
	if foldersRaw, ok := project["folders"].([]interface{}); ok {
		for _, f := range foldersRaw {
			if s, ok := f.(string); ok {
				folders = append(folders, s)
			}
		}
	}

	fmt.Println("Files:", files)
	fmt.Println("Folders:", folders)
}

注意点 #

• 需要多次类型断言，代码冗余且易出错
• 无类型安全保障，仅适合临时调试或结构极不稳定的场景

你之前尝试失败的原因 #

你直接将YAML字符串解析到[]string，但YAML顶层是project映射节点，并非列表，类型不匹配导致Unmarshal失败。只要匹配好层级结构，无论是结构体还是嵌套Map都能解决问题。

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