mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 05:32:02 +08:00
655fb1dcca
* Add READ_BUFFER_SIZE, CONCURRENCY,DISABLE_KEEPALIVE to ENV * Add TestGuessSupportedFormat
110 lines
3.3 KiB
Go
110 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"strconv"
|
|
"webp_server_go/config"
|
|
"webp_server_go/encoder"
|
|
"webp_server_go/handler"
|
|
"webp_server_go/helper"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/etag"
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
|
"github.com/gofiber/fiber/v2/middleware/recover"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
ReadBufferSizeStr = helper.GetEnv("READ_BUFFER_SIZE", "4096") // Default: 4096
|
|
ReadBufferSize, _ = strconv.Atoi(ReadBufferSizeStr)
|
|
ConcurrencyStr = helper.GetEnv("CONCURRENCY", "262144") // Default: 256 * 1024
|
|
Concurrency, _ = strconv.Atoi(ConcurrencyStr)
|
|
DisableKeepaliveStr = helper.GetEnv("DISABLE_KEEPALIVE", "false") // Default: false
|
|
DisableKeepalive, _ = strconv.ParseBool(DisableKeepaliveStr)
|
|
)
|
|
|
|
// https://docs.gofiber.io/api/fiber
|
|
var app = fiber.New(fiber.Config{
|
|
ServerHeader: "WebP Server Go",
|
|
AppName: "WebP Server Go",
|
|
DisableStartupMessage: true,
|
|
ProxyHeader: "X-Real-IP",
|
|
ReadBufferSize: ReadBufferSize, // per-connection buffer size for requests' reading. This also limits the maximum header size. Increase this buffer if your clients send multi-KB RequestURIs and/or multi-KB headers (for example, BIG cookies).
|
|
Concurrency: Concurrency, // Maximum number of concurrent connections.
|
|
DisableKeepalive: DisableKeepalive, // Disable keep-alive connections, the server will close incoming connections after sending the first response to the client
|
|
})
|
|
|
|
func setupLogger() {
|
|
log.SetOutput(os.Stdout)
|
|
log.SetReportCaller(true)
|
|
formatter := &log.TextFormatter{
|
|
EnvironmentOverrideColors: true,
|
|
FullTimestamp: true,
|
|
TimestampFormat: config.TimeDateFormat,
|
|
CallerPrettyfier: func(f *runtime.Frame) (string, string) {
|
|
return fmt.Sprintf("[%d:%s]", f.Line, f.Function), ""
|
|
},
|
|
}
|
|
log.SetFormatter(formatter)
|
|
log.SetLevel(log.InfoLevel)
|
|
|
|
// fiber logger format
|
|
app.Use(logger.New(logger.Config{
|
|
Format: config.FiberLogFormat,
|
|
TimeFormat: config.TimeDateFormat,
|
|
}))
|
|
app.Use(recover.New(recover.Config{}))
|
|
log.Infoln("WebP Server Go ready.")
|
|
}
|
|
|
|
func init() {
|
|
// Our banner
|
|
banner := fmt.Sprintf(`
|
|
▌ ▌ ▌ ▛▀▖ ▞▀▖ ▞▀▖
|
|
▌▖▌▞▀▖▛▀▖▙▄▘ ▚▄ ▞▀▖▙▀▖▌ ▌▞▀▖▙▀▖ ▌▄▖▞▀▖
|
|
▙▚▌▛▀ ▌ ▌▌ ▖ ▌▛▀ ▌ ▐▐ ▛▀ ▌ ▌ ▌▌ ▌
|
|
▘ ▘▝▀▘▀▀ ▘ ▝▀ ▝▀▘▘ ▘ ▝▀▘▘ ▝▀ ▝▀
|
|
|
|
WebP Server Go - v%s
|
|
Develop by WebP Server team. https://github.com/webp-sh`, config.Version)
|
|
// main init is the last one to be called
|
|
flag.Parse()
|
|
// process cli params
|
|
if config.DumpConfig {
|
|
fmt.Println(config.SampleConfig)
|
|
os.Exit(0)
|
|
}
|
|
if config.DumpSystemd {
|
|
fmt.Println(config.SampleSystemd)
|
|
os.Exit(0)
|
|
}
|
|
if config.ShowVersion {
|
|
fmt.Printf("\n %c[1;32m%s%c[0m\n\n", 0x1B, banner+"", 0x1B)
|
|
os.Exit(0)
|
|
}
|
|
config.LoadConfig()
|
|
fmt.Printf("\n %c[1;32m%s%c[0m\n\n", 0x1B, banner, 0x1B)
|
|
setupLogger()
|
|
}
|
|
|
|
func main() {
|
|
if config.Prefetch {
|
|
go encoder.PrefetchImages()
|
|
}
|
|
app.Use(etag.New(etag.Config{
|
|
Weak: true,
|
|
}))
|
|
|
|
listenAddress := config.Config.Host + ":" + config.Config.Port
|
|
app.Get("/*", handler.Convert)
|
|
|
|
fmt.Println("WebP Server Go is Running on http://" + listenAddress)
|
|
|
|
_ = app.Listen(listenAddress)
|
|
|
|
}
|