切换开发与生产环境

main.go

var ENV string
var PORT string

func init() {
	flag.StringVar(&ENV, "module", "", "运行开发环境")
	flag.StringVar(&PORT, "port", "4000", "启用的端口")
	flag.Parse()
}

func main() {
	// 如果输入go run . -module=pro则进入生产模式,使用生产模式的环境变量
	if ENV == "pro" {
		err := godotenv.Load("./config/db.production.yaml", "./config/minio.production.yaml")
		if err != nil {
			log.Fatal(err)
		}
	}
	err := godotenv.Load("./config/db.development.yaml", "./config/minio.development.yaml")
	if err != nil {
		log.Fatal(err)
	}

	router.Server(PORT)
}