package nf import ( "context" "crypto/tls" "errors" "fmt" "io" "log" "net" "net/http" "strings" ) type App struct { *RouterGroup config *Config router *router groups []*RouterGroup server *http.Server } func (a *App) ServeHTTP(writer http.ResponseWriter, request *http.Request) { c := newContext(a, writer, request) for _, group := range a.groups { if strings.HasPrefix(request.URL.Path, group.prefix) { c.handlers = append(c.handlers, group.middlewares...) } } if err := a.router.handle(c); err != nil { var ne = &Err{} if errors.As(err, ne) { writer.WriteHeader(ne.Status) } else { writer.WriteHeader(500) } _, _ = writer.Write([]byte(err.Error())) } } func (a *App) run(ln net.Listener) error { srv := &http.Server{Handler: a} if a.config.DisableHttpErrorLog { srv.ErrorLog = log.New(io.Discard, "", 0) } a.server = srv if !a.config.DisableBanner { fmt.Println(banner + "nf serve at: " + ln.Addr().String() + "\n") } err := a.server.Serve(ln) if !errors.Is(err, http.ErrServerClosed) || a.config.ErrServeClose { return err } return nil } func (a *App) Run(address string) error { ln, err := net.Listen("tcp", address) if err != nil { return err } return a.run(ln) } func (a *App) RunTLS(address string, tlsConfig *tls.Config) error { ln, err := tls.Listen("tcp", address, tlsConfig) if err != nil { return err } return a.run(ln) } func (a *App) RunListener(ln net.Listener) error { a.server = &http.Server{Addr: ln.Addr().String()} return a.run(ln) } func (a *App) Shutdown(ctx context.Context) error { return a.server.Shutdown(ctx) }