support passing CLI arguments to lua script

1 files changed, 4 insertions, 4 deletions

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)
 			}