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

Add Settings StartupBanner & HideStartupBanner

pull/425/head
Juan Riquelme González 2020-05-29 23:41:03 +02:00
parent c9664609ee
commit d247e2a961
1 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
app.go
View File

@ -28,6 +28,8 @@ import (
// Version of current package
const Version = "1.10.1"
var fiberASCIIBanner = fmt.Sprintf(" _______ __\n ____ / ____(_) /_ ___ _____\n_____ / /_ / / __ \\/ _ \\/ ___/\n __ / __/ / / /_/ / __/ /\n /_/ /_/_.___/\\___/_/ v%s\n", Version)
// Map is a shortcut for map[string]interface{}, usefull for JSON returns
type Map map[string]interface{}
@ -91,8 +93,14 @@ type Settings struct {
// By default all header names are normalized: conteNT-tYPE -> Content-Type
DisableHeaderNormalizing bool // default: false
// When set to true, it will not print out the fiber ASCII and "listening" on message
DisableStartupMessage bool
// Startup banner
StartupBanner string // default: fiberASCIIBanner
// When set to true, it will hide the startup banner
HideStartupBanner bool // default: false
// When set to true, it will not print out the StartupBanner and "listening" on message
DisableStartupMessage bool // default: false
// Templates is the interface that wraps the Render function.
Templates Templates
@ -166,9 +174,10 @@ func New(settings ...*Settings) *App {
},
// Set default settings
Settings: &Settings{
Prefork: utils.GetArgument("-prefork"),
BodyLimit: 4 * 1024 * 1024,
Concurrency: 256 * 1024,
Prefork: utils.GetArgument("-prefork"),
BodyLimit: 4 * 1024 * 1024,
Concurrency: 256 * 1024,
StartupBanner: fiberASCIIBanner,
},
}
// Overwrite settings if provided
@ -188,6 +197,9 @@ func New(settings ...*Settings) *App {
getBytes = getBytesImmutable
getString = getStringImmutable
}
if app.Settings.StartupBanner == "" {
app.Settings.StartupBanner = fiberASCIIBanner
}
}
// Initialize app
return app.init()
@ -300,9 +312,11 @@ func (app *App) Serve(ln net.Listener, tlsconfig ...*tls.Config) error {
if len(tlsconfig) > 0 {
ln = tls.NewListener(ln, tlsconfig[0])
}
// Print listening message
// Print startup message
if !app.Settings.DisableStartupMessage {
fmt.Printf(" _______ __\n ____ / ____(_) /_ ___ _____\n_____ / /_ / / __ \\/ _ \\/ ___/\n __ / __/ / / /_/ / __/ /\n /_/ /_/_.___/\\___/_/ v%s\n", Version)
if !app.Settings.HideStartupBanner {
fmt.Print(app.Settings.StartupBanner)
}
fmt.Printf("Started listening on %s\n", ln.Addr().String())
}
return app.server.Serve(ln)
@ -341,9 +355,13 @@ func (app *App) Listen(address interface{}, tlsconfig ...*tls.Config) error {
if len(tlsconfig) > 0 {
ln = tls.NewListener(ln, tlsconfig[0])
}
// Print listening message
// Print startup message
if !app.Settings.DisableStartupMessage && !utils.GetArgument("-child") {
fmt.Printf(" _______ __\n ____ / ____(_) /_ ___ _____\n_____ / /_ / / __ \\/ _ \\/ ___/\n __ / __/ / / /_/ / __/ /\n /_/ /_/_.___/\\___/_/ v%s\n", Version)
if !app.Settings.HideStartupBanner {
if !app.Settings.HideStartupBanner {
fmt.Print(app.Settings.StartupBanner)
}
}
fmt.Printf("Started listening on %s\n", ln.Addr().String())
}
return app.server.Serve(ln)