Golang根据参数引入相应配置文件的实现方式

1) config目录 创建dev.env和prod.env

env文件示例
dev.env

#gin
GIN_DEBUG=true
GIN_PORT=9000

prod.env

#gin
GIN_DEBUG=false
GIN_PORT=9527

2) 使用godotenv引入对应的文件

没有这个包的话需要先引入

go get github.com/joho/godotenv
package main

import (
	"flag"
	"log"
	"os"
	
	"github.com/gin-gonic/gin"
	"github.com/joho/godotenv"
)

var envArgs string

func init() {
	// 默认为dev环境
	flag.StringVar(&envArgs, "env", "dev", "运行环境")
}

func main() {
	// 需要先解析才能使用
	flag.Parse()
	err := godotenv.Load("./config/" + string(envArgs) + ".env")
	if err != nil {
		log.Fatal("Error loading .env file")
	}
	ginPort := os.GetEnv("GIN_PORT")
	debugFlag := os.GetEnv("GIN_DEBUG")
	if debugFlag == "true" {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}
	r := gin.Default()
	r.Run(":" + ginPort)
}

命令行执行

#prod环境
go run main.go -env=prod
#dev环境 如果env不传默认为dev
go run main.go -env=dev
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值