main.go 使用 APIConfig 传参

main.go

@@ -1,12 +1,29 @@
 package main
 
 import (
+	"flag"
 	"log"
 	"msw-open-music/internal/pkg/api"
 )
 
+var DatabaseName string
+var Listen string
+var Token string
+
+func init() {
+	flag.StringVar(&DatabaseName, "db", "/tmp/music.sqlite3", "sqlite3 database file path")
+	flag.StringVar(&Listen, "listen", ":8080", "http server listening")
+	flag.StringVar(&Token, "token", "mikusavetheworld", "secret token")
+}
+
 func main() {
-	api, err := api.NewAPI("/tmp/test.sqlite3", ":8080")
+	flag.Parse()
+	apiConfig := api.APIConfig{
+		DatabaseName: DatabaseName,
+		Addr: Listen,
+		Token: Token,
+	}
+	api, err := api.NewAPI(apiConfig)
 	if err != nil {
 		log.Fatal(err)
 	}