Golang 在包之间共享配置

Question

所以我刚开始学习 Go 编程语言，并且花了几个小时查看示例、参考资料等。正如你们中的大多数人会同意的那样，学习一门语言没有比潜入并做点什么更好的方法了，这就是我目前正在尝试做的事情。我正在构建一个 Restful Web 服务。我已经设法让基础知识运行以及插入数据库、注册路由等。但是在过去的两天里，我一直在努力实现应用程序配置/属性。可能只是因为我是新手，所以我的 Go 项目架构都是错误的，因此为什么我会遇到这样的困难。不用多说，这里是我的项目结构

src
   server
      database
         dbaccess.go
         dbcomm.go
      handling
         handler.go
         handlercomm.go
      models
         config.go
         response.go
         user.go
      routing
         routes.go
      main.go

这是我的 config.go

package models

import (
   "io/ioutil"
   "encoding/json"
)

type Config struct  {
   Db map[string]string `json:"db"`
   Server map[string]string `json:"server"`
}


func NewConfig(fname string) *Config{
   data,err := ioutil.ReadFile(fname)
   if err != nil{
      panic(err)
   }
   config := Config{}
   err = json.Unmarshal(data,&config)
   if err != nil {
   panic(err)
}
return config

这是我的主线

func main(){
    args := os.Args[1:]
    if len(args) == 0{
       fmt.Println("********************\nMust specify a config file   in args\n********************")
    os.Exit(1)
   }

   config := models.NewConfig(args[0])
   port := config.Server["PORT"]

   router := routing.NewRouter()
   fmt.Printf(  "-------------------------------------------------\n"+
        "Listening and Serving on Port %s\n"+
        "-------------------------------------------------",port)

   log.Fatal(http.ListenAndServe(":"+port,router))
 }

最后这是我的路线被映射的地方

type Route struct {
   Name string
   Method string
   Pattern string
   HandlerFunc http.HandlerFunc
}

var routes = []Route{
   Route{
    "signup",
    "POST",
    "/signup",
    handling.PostSignUpUser,
   },

   Route{
    "authenticate",
    "POST",
    "/login",
    handling.PostLogin,
   },
}

func NewRouter() *mux.Router{
 router :=  mux.NewRouter().StrictSlash(true)
 for _,route := range routes{       
    router.Methods(route.Method)
          .Path(route.Pattern)
          .Name(route.Name)
          .Handler(route.HandlerFunc)
}

return router
}

正如您在我的 Main 中看到的，我从一个文件中初始化了相关配置，这很好。但问题是我将如何在数据库包中使用与 main 相同的配置对象，因为我需要设置主机、端口等？我可以再次解析文件，但如果我可以从一开始就共享一个对象，我会更喜欢。请指出正确的方向

Answer 1

我的建议是在config.go 中声明一个全局变量并使用init() 函数对其进行初始化。这样，您知道该变量将始终在任何包导入它时被初始化。这是一些代码：

package models

import (
   "io/ioutil"
   "encoding/json"
)


var (

    Configuration Config 
)


init() {

    args := os.Args[1:]
    if len(args) == 0{
       fmt.Println("********************\nMust specify a config file   in args\n********************")
       os.Exit(1)
   }

   Configuration = NewConfig(args[0]) // configuration initialized here
}

type Config struct  {
   Db map[string]string `json:"db"`
   Server map[string]string `json:"server"`
}


func NewConfig(fname string) *Config{
   data,err := ioutil.ReadFile(fname)
   if err != nil{
      panic(err)
   }
   config := Config{}
   err = json.Unmarshal(data,&config)
   if err != nil {
      panic(err)
   }
   return config
}

var() 将在init() 之前运行，但init() 将在在导入它的包中的代码之前运行。因此，如果main.go 导入了models 包，那么models 中的init() 将在main.go 中的任何代码之前运行，因此变量Configuration 将在使用之前被初始化。

Effective Go explanation of init()

Answer 2

现在你想要的只是提供一个变量可以在另一个包中使用，解决方法很简单，记住如果你声明一个变量名以大写字母开头：[AZ]，这个变量可以在go中的另一个包。

所以您只需将main.go 中的config 重命名为Config 并将其提取为全局变量：

var Config *models.Config
func main(){
args := os.Args[1:]
if len(args) == 0{
   fmt.Println("********************\nMust specify a config file   in args\n********************")
os.Exit(1)
}

Config = models.NewConfig(args[0])
port := Config.Server["PORT"]

router := routing.NewRouter()
fmt.Printf(  "-------------------------------------------------\n"+
    "Listening and Serving on Port %s\n"+
    "-------------------------------------------------",port)

log.Fatal(http.ListenAndServe(":"+port,router))
}

当你想在另一个包中使用它时，只需调用<package name>.Config，包名就是你的main.go所属的包名，在你的情况下可能是main。

Answer 3

我为此构建了一个库，一个工具箱工厂（单例）和一个不可知的配置解析器（yaml、json 和 toml），看看它，观看示例：SpareBox