24w14a

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/json"
+	"flag"
 	"fmt"
 	"log"
 	"net/http"
@@ -21,6 +22,11 @@ var (
 	configfile = "/data/ghproxy/config/config.yaml"
 )
 
+func ReadFlag() {
+	cfgfile := flag.String("cfg", configfile, "config file path")
+	configfile = *cfgfile
+}
+
 func loadConfig() {
 	var err error
 	// 初始化配置
@@ -48,6 +54,7 @@ func Loadlist(cfg *config.Config) {
 }
 
 func init() {
+	ReadFlag()
 	loadConfig()
 	setupLogger()
 	Loadlist(cfg)