使用命令传递启动参数

Makefile

@@ -19,7 +19,7 @@ build:
 	go mod tidy
 	go build -o dist/main bin/main.go
 
-# 更新部署到服务器
+# 更新部署到服务器(生产和测试)
 update: build
 	scp dist/main $(HOST):~/webp/main_new
 	ssh ${HOST} "mv ~/webp/main ~/webp/main_old"

bin/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"flag"
 	"fmt"
 	"log"
 	"net/http"
@@ -217,7 +218,10 @@ func (image *Image) GetSimilarImagesIdList(collection_name string) (ids []int64)
 func main() {
 	runtime.GOMAXPROCS(runtime.NumCPU() - 1)
 
-	viper.SetConfigFile("./data/config.yaml")
+	configFilePath := flag.String("config", "./data/config.yaml", "配置文件路径")
+	flag.Parse()
+
+	viper.SetConfigFile(*configFilePath)
 	if err := viper.ReadInConfig(); err != nil {
 		log.Println("读取配置文件失败", err)
 	}