From 51872975d12c358b2ff0af9f78cda3c9490e486d Mon Sep 17 00:00:00 2001
From: unwox <me@unwox.com>
Date: Fri, 27 Sep 2024 15:31:54 +0600
Subject: support passing CLI arguments to lua script

---
 main.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index eaa3ec6..9a2ad52 100644
--- a/main.go
+++ b/main.go
@@ -23,11 +23,11 @@ func main() {
 	wrksNum := flag.Int("n", runtime.NumCPU(), "Number of HTTP-workers to start")
 	flag.Parse()
 
-	if flag.NArg() != 1 {
+	if flag.NArg() == 0 {
 		printUsage()
 	}
-	luaExe := flag.Arg(0)
-	mustExist(luaExe)
+	luaArgv := flag.Args()
+	mustExist(luaArgv[0])
 
 	httpClient := &http.Client{}
 
@@ -346,7 +346,7 @@ func main() {
 		wrk := NewWorker()
 		wrks = append(wrks, wrk)
 		go func () {
-			err := wrk.Start(luaExe, module)
+			err := wrk.Start(luaArgv, module)
 			if err != nil {
 				log.Fatal(err)
 			}
-- 
cgit v1.2.3