47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"runtime"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
func main() {
|
|
capacity, err := strconv.Atoi(os.Args[1])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
runtime.GOMAXPROCS(2)
|
|
app := fiber.New(fiber.Config{
|
|
Concurrency: capacity,
|
|
BodyLimit: 65536,
|
|
ReadBufferSize: 8192,
|
|
ReadTimeout: 5 * time.Second,
|
|
WriteTimeout: 5 * time.Second,
|
|
IdleTimeout: 5 * time.Second,
|
|
CaseSensitive: true,
|
|
StrictRouting: true,
|
|
})
|
|
app.Get("/json", func(context fiber.Ctx) error {
|
|
context.Set("Content-Type", "application/json")
|
|
return context.SendString(`{"ok":true}`)
|
|
})
|
|
app.Post("/echo", func(context fiber.Ctx) error {
|
|
context.Set("Content-Type", "application/octet-stream")
|
|
return context.Send(context.BodyRaw())
|
|
})
|
|
bound, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Printf("READY|%d\n", bound.Addr().(*net.TCPAddr).Port)
|
|
if err := app.Listener(bound, fiber.ListenConfig{DisableStartupMessage: true}); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|