func TestYaml(t *testing.T) { f, err := os.Create("testyaml.conf") if err != nil { t.Fatal(err) } _, err = f.WriteString(yamlcontext) if err != nil { f.Close() t.Fatal(err) } f.Close() defer os.Remove("testyaml.conf") yamlconf, err := config.NewConfig("yaml", "testyaml.conf") if err != nil { t.Fatal(err) } if yamlconf.String("appname") != "beeapi" { t.Fatal("appname not equal to beeapi") } if port, err := yamlconf.Int("httpport"); err != nil || port != 8080 { t.Error(port) t.Fatal(err) } if port, err := yamlconf.Int64("mysqlport"); err != nil || port != 3600 { t.Error(port) t.Fatal(err) } if pi, err := yamlconf.Float("PI"); err != nil || pi != 3.1415976 { t.Error(pi) t.Fatal(err) } if yamlconf.String("runmode") != "dev" { t.Fatal("runmode not equal to dev") } if v, err := yamlconf.Bool("autorender"); err != nil || v != false { t.Error(v) t.Fatal(err) } if v, err := yamlconf.Bool("copyrequestbody"); err != nil || v != true { t.Error(v) t.Fatal(err) } if err = yamlconf.Set("name", "henrylee2cn"); err != nil { t.Fatal(err) } if yamlconf.String("name") != "henrylee2cn" { t.Fatal("get name error") } }
func readConfig() Config { iniconf, err := config.NewConfig("ini", "conf/app.conf") if err != nil { log.Panicln(err) } return Config{ AppName: iniconf.DefaultString("appname", "thinkgo"), RunMode: iniconf.DefaultString("runmode", "debug"), HttpAddr: iniconf.DefaultString("httpaddr", "0.0.0.0"), HttpPort: iniconf.DefaultInt("httpport", 8080), TplSuffex: iniconf.DefaultString("tplsuffex", ".html"), TplLeft: iniconf.DefaultString("tplleft", "{{{"), TplRight: iniconf.DefaultString("tplright", "}}}"), } }
func getConfig() Config { iniconf, err := config.NewConfig("ini", "conf/app.conf") if err != nil { fmt.Println("\n 请确保在项目目录下运行,且存在配置文件 conf/app.conf") os.Exit(1) } var logLevel log.Level switch strings.ToUpper(iniconf.String("loglevel")) { case "TRACE": logLevel = log.TRACE case "DEBUG": logLevel = log.DEBUG case "INFO": logLevel = log.INFO case "NOTICE": logLevel = log.NOTICE case "WARN": logLevel = log.WARN case "ERROR": logLevel = log.ERROR case "FATAL": logLevel = log.FATAL case "OFF": logLevel = log.OFF default: logLevel = log.DEBUG } defaultModule := iniconf.DefaultString("defmodule", "home") return Config{ AppName: iniconf.DefaultString("appname", "thinkgo"), Debug: iniconf.DefaultBool("debug", true), LogLevel: logLevel, HttpAddr: iniconf.DefaultString("httpaddr", "0.0.0.0"), HttpPort: iniconf.DefaultInt("httpport", 8080), TplSuffix: iniconf.DefaultString("tplsuffex", ".html"), TplLeft: iniconf.DefaultString("tplleft", "{{{"), TplRight: iniconf.DefaultString("tplright", "}}}"), DefaultModule: SnakeString(strings.Trim(defaultModule, "/")), } }