MyException - 我的异常网
当前位置:我的异常网» Go » 2018 Golang Update(五)Configuration

2018 Golang Update(五)Configuration

www.MyException.Cn  网友分享于:2018-03-16  浏览:0次
2018 Golang Update(5)Configuration
2018 Golang Update(5)Configuration

Add the Dependency to Dep
>dep ensure -add github.com/spf13/viper

All the Viper method
https://godoc.org/github.com/spf13/viper#pkg-variables

Here is How I use that, when I run the command, I can do as follow to pass the ENV
>ENVIRONMENT=PROD bin/restful_go_api

Add dependency to use Viper
>dep ensure -add github.com/spf13/viper

The Configuration style will be YAML style config-prod.yaml as follow
restful_go_api/conf/config-prod.yaml
restful_go_api/conf/config-dev.yaml
restful_go_api/conf/config-stage.yaml

http:
  port: 8081
gin:
  debug: false
database:
  type: mysql
  host: 192.168.1.109
  user: watermonitor
  password: xxxxxx
  port: 3306
  name: watermonitor

here is the config module for init the Viper
restful_go_api/src/sillycat.com/restful_go_api/config/vipconfig.go
package config

import (
    "fmt"
    "github.com/spf13/viper"
    "log"
    "os"
)

const appName = "config"

func InitViperConfig() {
    log.Println("Start to init the config--------")
    viper.SetDefault("http.port", "8080")

    if os.Getenv("ENVIRONMENT") == "PROD" {
        log.Println("system is running under PROD mode")
        viper.SetConfigName(appName + "-prod") // name of config file (without extension)
    } else if os.Getenv("ENVIRONMENT") == "STAGE" {
        log.Println("system is running under STAGE mode")
        viper.SetConfigName(appName + "-stage") // name of config file (without extension)
    } else {
        log.Println("system is running under DEV mode")
        viper.SetConfigName(appName + "-dev") // name of config file (without extension)
    }
    viper.AddConfigPath("/etc/" + appName + "/") // path to look for the config file in
    viper.AddConfigPath("./")                    // optionally look for config in the working directory
    viper.AddConfigPath("./conf/")
    viper.AutomaticEnv()
    err := viper.ReadInConfig() // Find and read the config file
    if err != nil {             // Handle errors reading the config file
        panic(fmt.Errorf("Fatal error config file: %s \n", err))
    }
}

In the database init class, here is how I use that
restful_go_api/src/sillycat.com/restful_go_api/common/db.go
package common

import (
    "fmt"
    _ "github.com/go-sql-driver/mysql"
    "github.com/go-xorm/core"
    "github.com/go-xorm/xorm"
    "github.com/spf13/viper"
    "sillycat.com/restful_go_api/config"
)

var engine = InitDatabase()

func InitDatabase() *xorm.Engine {

    config.InitViperConfig()

    var engine *xorm.Engine
    var err error

    dbType := viper.GetString("database.type")
    dbUser := viper.GetString("database.user")
    dbPwd := viper.GetString("database.password")
    dbHost := viper.GetString("database.host")
    dbPort := viper.GetString("database.port")
    dbName := viper.GetString("database.name")

    engine, err = xorm.NewEngine(dbType, dbUser+":"+dbPwd+"@tcp("+dbHost+":"+dbPort+")/"+dbName+"?charset=utf8")
    if err != nil {
        fmt.Println(err)
    }
    engine.Ping() //ping

    engine.ShowSQL(true) //show SQL
    engine.Logger().SetLevel(core.LOG_DEBUG)
    return engine
}

func GetDatabase() *xorm.Engine {
    return engine
}

In the main.go, here is how I use that configuration
restful_go_api/src/sillycat.com/restful_go_api/main.go

package main

import (
    "github.com/gin-gonic/gin"
    "github.com/spf13/viper"
    "log"
    "sillycat.com/restful_go_api/waterrecord"
)

var DB = make(map[string]string)

func SetupRouter() *gin.Engine {

    // Disable Console Color
    // gin.DisableConsoleColor()
    if viper.GetBool("gin.debug") {
        log.Println("Gin is under debug mode")
    } else {
        log.Println("Gin is under prod mode")
        gin.SetMode(gin.ReleaseMode)
    }
    router := gin.Default()

    // Ping test
    router.GET("/ping", func(c *gin.Context) {
        c.String(200, "pong")
    })

    v1 := router.Group("api/v1")
    {
        v1.GET("/waterrecords", waterrecord.GetWaterRecords)
        v1.GET("/waterrecords/:id", waterrecord.GetWaterRecord)
        v1.POST("/waterrecords", waterrecord.PostWaterRecord)
        v1.PUT("/waterrecords/:id", waterrecord.UpdateWaterRecord)
        v1.DELETE("/waterrecords/:id", waterrecord.DeleteWaterRecord)
    }

    return router
}

func main() {
    router := SetupRouter()
    // Listen and Server in 0.0.0.0:8080
    port := viper.GetString("http.port")
    router.Run(":" + port)
}




References:
https://medium.com/@felipedutratine/manage-config-in-golang-to-get-variables-from-file-and-env-variables-33d876887152
https://github.com/spf13/viper

文章评论

初级 vs 高级开发者 哪个性价比更高?
初级 vs 高级开发者 哪个性价比更高?
程序员应该关注的一些事儿
程序员应该关注的一些事儿
总结2014中国互联网十大段子
总结2014中国互联网十大段子
程序员周末都喜欢做什么?
程序员周末都喜欢做什么?
我是如何打败拖延症的
我是如何打败拖延症的
Google伦敦新总部 犹如星级庄园
Google伦敦新总部 犹如星级庄园
老美怎么看待阿里赴美上市
老美怎么看待阿里赴美上市
那些性感的让人尖叫的程序员
那些性感的让人尖叫的程序员
程序员的鄙视链
程序员的鄙视链
Web开发者需具备的8个好习惯
Web开发者需具备的8个好习惯
程序员和编码员之间的区别
程序员和编码员之间的区别
我跳槽是因为他们的显示器更大
我跳槽是因为他们的显示器更大
代码女神横空出世
代码女神横空出世
一个程序员的时间管理
一个程序员的时间管理
团队中“技术大拿”并非越多越好
团队中“技术大拿”并非越多越好
写给自己也写给你 自己到底该何去何从
写给自己也写给你 自己到底该何去何从
中美印日四国程序员比较
中美印日四国程序员比较
看13位CEO、创始人和高管如何提高工作效率
看13位CEO、创始人和高管如何提高工作效率
每天工作4小时的程序员
每天工作4小时的程序员
那些争议最大的编程观点
那些争议最大的编程观点
当下全球最炙手可热的八位少年创业者
当下全球最炙手可热的八位少年创业者
10个帮程序员减压放松的网站
10个帮程序员减压放松的网站
程序员的一天:一寸光阴一寸金
程序员的一天:一寸光阴一寸金
聊聊HTTPS和SSL/TLS协议
聊聊HTTPS和SSL/TLS协议
为啥Android手机总会越用越慢?
为啥Android手机总会越用越慢?
5款最佳正则表达式编辑调试器
5款最佳正则表达式编辑调试器
Java程序员必看电影
Java程序员必看电影
十大编程算法助程序员走上高手之路
十大编程算法助程序员走上高手之路
如何区分一个程序员是“老手“还是“新手“?
如何区分一个程序员是“老手“还是“新手“?
Java 与 .NET 的平台发展之争
Java 与 .NET 的平台发展之争
“懒”出效率是程序员的美德
“懒”出效率是程序员的美德
科技史上最臭名昭著的13大罪犯
科技史上最臭名昭著的13大罪犯
程序员眼里IE浏览器是什么样的
程序员眼里IE浏览器是什么样的
程序员必看的十大电影
程序员必看的十大电影
如何成为一名黑客
如何成为一名黑客
不懂技术不要对懂技术的人说这很容易实现
不懂技术不要对懂技术的人说这很容易实现
“肮脏的”IT工作排行榜
“肮脏的”IT工作排行榜
做程序猿的老婆应该注意的一些事情
做程序猿的老婆应该注意的一些事情
程序员最害怕的5件事 你中招了吗?
程序员最害怕的5件事 你中招了吗?
2013年美国开发者薪资调查报告
2013年美国开发者薪资调查报告
 程序员的样子
程序员的样子
亲爱的项目经理,我恨你
亲爱的项目经理,我恨你
2013年中国软件开发者薪资调查报告
2013年中国软件开发者薪资调查报告
编程语言是女人
编程语言是女人
Web开发人员为什么越来越懒了?
Web开发人员为什么越来越懒了?
我的丈夫是个程序员
我的丈夫是个程序员
60个开发者不容错过的免费资源库
60个开发者不容错过的免费资源库
要嫁就嫁程序猿—钱多话少死的早
要嫁就嫁程序猿—钱多话少死的早
鲜为人知的编程真相
鲜为人知的编程真相
软件开发程序错误异常ExceptionCopyright © 2009-2015 MyException 版权所有