Golang -embed结合viper打包静态文件

李若盛开 / 2023-05-19 / 原文

代码含有viper独立方式和go embed方式

package main

import (
    "bytes"
    "embed"
    "fmt"
    "time"
    "github.com/spf13/viper"
)

//go:embed config.yaml
var f []byte

func main() {
    // config := viper.New()
    // config.AddConfigPath("./")         //设置读取的文件路径
    // config.SetConfigName("config")    //设置读取的文件名
    // config.SetConfigType("yaml")      //设置文件的类型
    //尝试进行配置读取
    // if err := config.ReadInConfig(); err != nil {
    // fmt.Println("error: ", err)
    // fmt.Println("config:", config.GetString("SignalingServer"))
    
    //embed 方式
    config := viper.New()
    config.SetConfigType("yaml")
    config.ReadConfig(bytes.NewReader(f))

    fmt.Println("config:", config.Get("SignalingServer"))

}