Nulo/kawipiko
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[server] Add support for threads argument to dummy server.

Browse source
This commit is contained in:
Ciprian Dorin Craciun 2021-12-19 12:35:15 +02:00
parent 37590c0d75
commit 4876ce43e2
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
sources/cmd/server-dummy.go
View file

@ -6,6 +6,7 @@ import "fmt"
import "os" import "os"
import "runtime" import "runtime"
import "runtime/debug" import "runtime/debug"
import "strconv"
import "time" import "time"
import "github.com/valyala/fasthttp" import "github.com/valyala/fasthttp"
@ -17,7 +18,7 @@ import "github.com/valyala/fasthttp/reuseport"
func main () () { func main () () {
runtime.GOMAXPROCS (2) runtime.GOMAXPROCS (1)
debug.SetGCPercent (-1) debug.SetGCPercent (-1)
debug.SetMaxThreads (128) debug.SetMaxThreads (128)
@ -25,6 +26,7 @@ func main () () {
_endpoint := "127.0.0.1:8080" _endpoint := "127.0.0.1:8080"
_threads := 1
_timeouts := false _timeouts := false
switch len (os.Args) { switch len (os.Args) {
@ -32,16 +34,28 @@ func main () () {
// NOP // NOP
case 2 : case 2 :
_endpoint = os.Args[1] _endpoint = os.Args[1]
case 3 :
_endpoint = os.Args[1]
if _threads_0, _error := strconv.Atoi (os.Args[2]); (_error == nil) && (_threads > 0) {
_threads = _threads_0
} else {
panic ("[40396d14] invalid arguments!")
}
default : default :
panic ("[60023f00] invalid arguments!") panic ("[60023f00] invalid arguments!")
} }
if _threads > 1 {
runtime.GOMAXPROCS (_threads)
debug.SetMaxThreads (int (128 * (_threads / 64 + 1)))
}
_listener, _error := reuseport.Listen ("tcp4", _endpoint) _listener, _error := reuseport.Listen ("tcp4", _endpoint)
if _error != nil { if _error != nil {
panic (fmt.Sprintf ("[8c30a625] failed to listen: %s", _error)) panic (fmt.Sprintf ("[8c30a625] failed to listen: %s", _error))
} }
fmt.Fprintf (os.Stderr, "[ii] [04fa2421] listening on `http://%s/`...\n", _endpoint) fmt.Fprintf (os.Stderr, "[ii] [04fa2421] listening on `http://%s/` (using %d threads)...\n", _endpoint, _threads)
_server := & fasthttp.Server { _server := & fasthttp.Server {