summaryrefslogtreecommitdiff
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go10
1 files changed, 7 insertions, 3 deletions
diff --git a/main.go b/main.go
index f4480f6..4cbc7c9 100644
--- a/main.go
+++ b/main.go
@@ -9,15 +9,19 @@ import (
"log"
"net/http"
"os"
+ "runtime"
"strings"
"time"
)
func main() {
- if len(os.Args) != 2 {
+ wrksNum := flag.Int("n", runtime.NumCPU(), "Number of HTTP-workers to start")
+ flag.Parse()
+
+ if flag.NArg() != 1 {
printUsage()
}
- luaExe := os.Args[1]
+ luaExe := flag.Arg(0)
mustExist(luaExe)
httpClient := &http.Client{}
@@ -135,7 +139,7 @@ func main() {
module["http"] = httpModule
// start workers
- for i := 0; i < 1; i++ {
+ for i := 0; i < *wrksNum; i++ {
wrk := NewWorker()
wrks = append(wrks, wrk)
go func () {