mirror of
https://gitee.com/XM-GO/PandaX.git
synced 2026-04-23 10:58:35 +08:00
【更新】更新restful
This commit is contained in:
59
pkg/transport/http_server.go
Normal file
59
pkg/transport/http_server.go
Normal file
@@ -0,0 +1,59 @@
|
||||
package transport
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/XM-GO/PandaKit/logger"
|
||||
"github.com/emicklei/go-restful/v3"
|
||||
"net/http"
|
||||
"pandax/pkg/global"
|
||||
)
|
||||
|
||||
type HttpServer struct {
|
||||
Addr string
|
||||
srv *http.Server
|
||||
|
||||
Container *restful.Container
|
||||
}
|
||||
|
||||
func NewHttpServer(addr string) *HttpServer {
|
||||
c := restful.NewContainer()
|
||||
c.EnableContentEncoding(true)
|
||||
restful.TraceLogger(&httpLog{})
|
||||
restful.SetLogger(&httpLog{})
|
||||
return &HttpServer{
|
||||
Addr: addr,
|
||||
Container: c,
|
||||
srv: &http.Server{
|
||||
Addr: addr,
|
||||
Handler: c,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (s *HttpServer) Type() Type {
|
||||
return TypeHTTP
|
||||
}
|
||||
|
||||
func (s *HttpServer) Start(ctx context.Context) error {
|
||||
global.Log.Debugf("HTTP Server listen: %s", s.Addr)
|
||||
go func() {
|
||||
if err := s.srv.ListenAndServe(); err != nil {
|
||||
global.Log.Errorf("error http serve: %s", err)
|
||||
}
|
||||
}()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *HttpServer) Stop(ctx context.Context) error {
|
||||
return s.srv.Shutdown(ctx)
|
||||
}
|
||||
|
||||
type httpLog struct{}
|
||||
|
||||
func (t *httpLog) Print(v ...any) {
|
||||
logger.Log.Debug(v...)
|
||||
}
|
||||
|
||||
func (t *httpLog) Printf(format string, v ...any) {
|
||||
logger.Log.Debugf(format, v...)
|
||||
}
|
||||
Reference in New Issue
Block a user