GOMODULES
/
fiber
mirror of https://github.com/gofiber/fiber.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add Prefork support for Windows

pull/502/head
Fenny 2020-06-22 15:55:24 +02:00
parent 9c2a36be17
commit a72ce156c5
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
app.go
View File

@ -13,8 +13,8 @@ import (
"net"
"net/http"
"net/http/httputil"
"os"
"reflect"
"runtime"
"sort"
"strconv"
"strings"
@ -215,8 +215,7 @@ var (
func init() { //nolint:gochecknoinits
// Definition flag to not break the program when the user adds their own flags
// and runs `flag.Parse()`
flag.BoolVar(&prefork, childFlag[1:], false, "Is a child process")
flag.BoolVar(&child, preforkFlag[1:], false, "use prefork")
flag.BoolVar(&prefork, preforkFlag[1:], false, "use prefork")
}
// New creates a new Fiber named instance.
@ -444,7 +443,11 @@ func (app *App) Listen(address interface{}, tlsconfig ...*tls.Config) error {
// Update fiber server settings
app.init()
// Start prefork
if app.Settings.Prefork && runtime.NumCPU() > 1 {
if app.Settings.Prefork {
// Print startup message
if !app.Settings.DisableStartupMessage {
startupMessage(addr)
}
pf := fprefork.New(app.server) // fasthttp/prefork
pf.Reuseport = true
pf.Network = "tcp4"
@ -452,10 +455,6 @@ func (app *App) Listen(address interface{}, tlsconfig ...*tls.Config) error {
if len(tlsconfig) > 0 {
lnn = tls.NewListener(lnn, tlsconfig[0])
}
// Print startup message
if !app.Settings.DisableStartupMessage && !utils.GetArgument(childFlag) {
startupMessage(lnn.Addr().String())
}
return app.server.Serve(lnn)
}
return pf.ListenAndServe(addr)
@ -470,7 +469,7 @@ func (app *App) Listen(address interface{}, tlsconfig ...*tls.Config) error {
ln = tls.NewListener(ln, tlsconfig[0])
}
// Print startup message
if !app.Settings.DisableStartupMessage && !utils.GetArgument(childFlag) {
if !app.Settings.DisableStartupMessage {
startupMessage(ln.Addr().String())
}
// Start listening
@ -611,6 +610,10 @@ func (app *App) init() *App {
}
func startupMessage(addr string) {
fmt.Printf(" _______ __\n ____ / ____(_) /_ ___ _____\n_____ / /_ / / __ \\/ _ \\/ ___/\n __ / __/ / / /_/ / __/ /\n /_/ /_/_.___/\\___/_/ v%s\n", Version)
fmt.Printf("Started listening on %s\n", addr)
if fprefork.IsChild() {
fmt.Printf("Launched child proc #%v\n", os.Getpid())
} else {
fmt.Printf(" _______ __\n ____ / ____(_) /_ ___ _____\n_____ / /_ / / __ \\/ _ \\/ ___/\n __ / __/ / / /_/ / __/ /\n /_/ /_/_.___/\\___/_/ v%s\n", Version)
fmt.Printf("Started listening on %s\n", addr)
}
}