Golang中如何处理动态YAML配置文件

为什么不使用YAML文件中的集合？https://yaml.org/spec/1.2/spec.html#id2759963

在Go中处理动态YAML配置，可以使用map[string]interface{}或自定义结构体配合切片。以下是两种解决方案：

方案1：使用map动态解析

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"

	"gopkg.in/yaml.v2"
)

type Config struct {
	ProcessingFolder string                 `yaml:"processingfolder"`
	Drives           map[string]string      `yaml:"drives"`
}

func (c *Config) LoadConfig(filePath string) error {
	yamlFile, err := ioutil.ReadFile(filePath)
	if err != nil {
		return fmt.Errorf("读取配置文件失败: %v", err)
	}

	err = yaml.Unmarshal(yamlFile, c)
	if err != nil {
		return fmt.Errorf("解析YAML失败: %v", err)
	}

	// 验证处理文件夹
	if _, err := os.Stat(c.ProcessingFolder); os.IsNotExist(err) {
		return fmt.Errorf("处理文件夹不存在: %s", c.ProcessingFolder)
	}

	// 验证所有驱动器文件夹
	for name, path := range c.Drives {
		if _, err := os.Stat(path); os.IsNotExist(err) {
			return fmt.Errorf("驱动器文件夹不存在 [%s]: %s", name, path)
		}
	}

	return nil
}

func main() {
	var config Config
	err := config.LoadConfig("./config.yml")
	if err != nil {
		fmt.Printf("配置加载失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("处理文件夹: %s\n", config.ProcessingFolder)
	fmt.Println("监控的驱动器:")
	for name, path := range config.Drives {
		fmt.Printf("  %s: %s\n", name, path)
	}

	// 示例：处理所有文件夹
	for folderName, folderPath := range config.Drives {
		fmt.Printf("开始处理文件夹 %s: %s\n", folderName, folderPath)
		// 这里添加你的文件夹处理逻辑
	}
}

方案2：使用切片存储文件夹配置

package main

import (
	"fmt"
	"io/ioutil"
	"os"

	"gopkg.in/yaml.v2"
)

type FolderConfig struct {
	Name string `yaml:"name"`
	Path string `yaml:"path"`
}

type Config struct {
	ProcessingFolder string         `yaml:"processingfolder"`
	Drives           []FolderConfig `yaml:"drives"`
}

func (c *Config) LoadConfig(filePath string) error {
	yamlFile, err := ioutil.ReadFile(filePath)
	if err != nil {
		return fmt.Errorf("读取配置文件失败: %v", err)
	}

	err = yaml.Unmarshal(yamlFile, c)
	if err != nil {
		return fmt.Errorf("解析YAML失败: %v", err)
	}

	if _, err := os.Stat(c.ProcessingFolder); os.IsNotExist(err) {
		return fmt.Errorf("处理文件夹不存在: %s", c.ProcessingFolder)
	}

	for _, drive := range c.Drives {
		if _, err := os.Stat(drive.Path); os.IsNotExist(err) {
			return fmt.Errorf("文件夹不存在 [%s]: %s", drive.Name, drive.Path)
		}
	}

	return nil
}

func main() {
	var config Config
	err := config.LoadConfig("./config.yml")
	if err != nil {
		fmt.Printf("配置加载失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("处理文件夹: %s\n", config.ProcessingFolder)
	fmt.Println("监控的文件夹:")
	for _, drive := range config.Drives {
		fmt.Printf("  %s: %s\n", drive.Name, drive.Path)
	}
}

对应的YAML配置文件：

processingfolder: C:\Processing\
drives:
  - name: folder1
    path: F:\data\
  - name: folder2
    path: G:\data\
  - name: folder3
    path: H:\data\

方案3：混合模式（支持命名和顺序访问）

package main

import (
	"fmt"
	"io/ioutil"
	"os"

	"gopkg.in/yaml.v2"
)

type Config struct {
	ProcessingFolder string            `yaml:"processingfolder"`
	Drives           map[string]string `yaml:"drives"`
	DriveList        []string          `yaml:"-"` // 从Drives map派生
}

func (c *Config) LoadConfig(filePath string) error {
	yamlFile, err := ioutil.ReadFile(filePath)
	if err != nil {
		return fmt.Errorf("读取配置文件失败: %v", err)
	}

	err = yaml.Unmarshal(yamlFile, c)
	if err != nil {
		return fmt.Errorf("解析YAML失败: %v", err)
	}

	if _, err := os.Stat(c.ProcessingFolder); os.IsNotExist(err) {
		return fmt.Errorf("处理文件夹不存在: %s", c.ProcessingFolder)
	}

	// 构建文件夹列表
	c.DriveList = make([]string, 0, len(c.Drives))
	for name, path := range c.Drives {
		if _, err := os.Stat(path); os.IsNotExist(err) {
			return fmt.Errorf("文件夹不存在 [%s]: %s", name, path)
		}
		c.DriveList = append(c.DriveList, name)
	}

	return nil
}

func (c *Config) GetDrivePath(name string) (string, bool) {
	path, exists := c.Drives[name]
	return path, exists
}

func (c *Config) GetDriveByIndex(index int) (string, string, bool) {
	if index < 0 || index >= len(c.DriveList) {
		return "", "", false
	}
	name := c.DriveList[index]
	path := c.Drives[name]
	return name, path, true
}

func main() {
	var config Config
	err := config.LoadConfig("./config.yml")
	if err != nil {
		fmt.Printf("配置加载失败: %v\n", err)
		os.Exit(1)
	}

	// 按名称访问
	if path, exists := config.GetDrivePath("folder1"); exists {
		fmt.Printf("folder1路径: %s\n", path)
	}

	// 按索引遍历
	for i := 0; i < len(config.DriveList); i++ {
		name, path, exists := config.GetDriveByIndex(i)
		if exists {
			fmt.Printf("文件夹 %d: %s -> %s\n", i, name, path)
		}
	}
}

使用map的方案最灵活，可以动态处理任意数量的文件夹，无需预定义结构体字段。YAML配置文件保持你原有的格式不变。