package core import ( "git.noahlan.cn/northlan/nnet/internal/env" "git.noahlan.cn/northlan/nnet/log" "git.noahlan.cn/northlan/nnet/packet" "git.noahlan.cn/northlan/nnet/serialize" "net/http" "time" ) type ( // RunOption defines the method to customize a Server. RunOption func(*Server) Server struct { ngin *engine router Router } ) // NewServer returns a server with given config of c and options defined in opts. // Be aware that later RunOption might overwrite previous one that write the same option. func NewServer(c EngineConf, opts ...RunOption) *Server { s := &Server{ ngin: newEngine(c), router: NewRouter(), } opts = append([]RunOption{WithNotFoundHandler(nil)}, opts...) for _, opt := range opts { opt(s) } return s } // AddRoutes add given routes into the Server. func (s *Server) AddRoutes(rs []Route) { s.ngin.addRoutes(rs...) } // AddRoute adds given route into the Server. func (s *Server) AddRoute(r Route) { s.AddRoutes([]Route{r}) } // Start starts the Server. // Graceful shutdown is enabled by default. func (s *Server) Start() { if err := s.ngin.serve(s.router); err != nil { log.Error(err) panic(err) } } // Stop stops the Server. func (s *Server) Stop() { s.ngin.close() } // Use adds the given middleware in the Server. func (s *Server) Use(middleware Middleware) { s.ngin.use(middleware) } // ToMiddleware converts the given handler to a Middleware. func ToMiddleware(handler func(next Handler) Handler) Middleware { return func(next HandlerFunc) HandlerFunc { return handler(next).Handle } } // WithMiddlewares adds given middlewares to given routes. func WithMiddlewares(ms []Middleware, rs ...Route) []Route { for i := len(ms) - 1; i >= 0; i-- { rs = WithMiddleware(ms[i], rs...) } return rs } // WithMiddleware adds given middleware to given route. func WithMiddleware(middleware Middleware, rs ...Route) []Route { routes := make([]Route, len(rs)) for i := range rs { route := rs[i] routes[i] = Route{ Matches: route.Matches, Handler: middleware(route.Handler), } } return routes } // WithNotFoundHandler returns a RunOption with not found handler set to given handler. func WithNotFoundHandler(handler Handler) RunOption { return func(server *Server) { notFoundHandler := server.ngin.notFoundHandler(handler) server.router.SetNotFoundHandler(notFoundHandler) } } func WithRouter(router Router) RunOption { return func(server *Server) { server.router = router } } func WithPacker(packer packet.Packer) RunOption { return func(server *Server) { server.ngin.packer = packer } } func WithProcessor(p Processor) RunOption { return func(server *Server) { server.ngin.processor = p } } func WithSerializer(s serialize.Serializer) RunOption { return func(server *Server) { server.ngin.serializer = s } } func WithLogger(logger log.Logger) RunOption { return func(_ *Server) { log.SetLogger(logger) } } // WithTimerPrecision 设置Timer精度 // 注:精度需大于1ms, 并且不能在运行时更改 // 默认精度是 time.Second func WithTimerPrecision(precision time.Duration) RunOption { if precision < time.Millisecond { panic("time precision can not less than a Millisecond") } return func(_ *Server) { env.TimerPrecision = precision } } func WithPipeline(pipeline Pipeline) RunOption { return func(server *Server) { server.ngin.pipeline = pipeline } } func WithHeartbeatInterval(d time.Duration) RunOption { return func(server *Server) { server.ngin.heartbeatInterval = d } } type WSOption func(opts *wsOptions) // WithWebsocket 开启Websocket, 参数是websocket的相关参数 nnet.WSOption func WithWebsocket(wsOpts ...WSOption) RunOption { return func(server *Server) { for _, opt := range wsOpts { opt(&server.ngin.wsOpt) } server.ngin.wsOpt.IsWebsocket = true } } // WithWSPath 设置websocket的path func WithWSPath(path string) WSOption { return func(opts *wsOptions) { opts.WebsocketPath = path } } // WithWSTLSConfig 设置websocket的证书和密钥 func WithWSTLSConfig(certificate, key string) WSOption { return func(opts *wsOptions) { opts.TLSCertificate = certificate opts.TLSKey = key } } func WithWSCheckOriginFunc(fn func(*http.Request) bool) WSOption { return func(opts *wsOptions) { if fn != nil { opts.CheckOrigin = fn } } }