2025-02-19 14:23:58 +08:00
|
|
|
package web
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
"net/http"
|
2025-03-08 10:59:31 +08:00
|
|
|
"proxy-server/server/pkg/env"
|
|
|
|
|
"strconv"
|
2025-02-25 14:48:50 +08:00
|
|
|
|
|
|
|
|
"github.com/gin-gonic/gin"
|
2025-02-19 14:23:58 +08:00
|
|
|
)
|
|
|
|
|
|
2025-05-14 15:13:44 +08:00
|
|
|
type Server struct {
|
|
|
|
|
http *http.Server
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func New() *Server {
|
|
|
|
|
return &Server{}
|
|
|
|
|
}
|
2025-02-19 14:23:58 +08:00
|
|
|
|
2025-05-14 15:13:44 +08:00
|
|
|
func (s *Server) Run() error {
|
2025-03-08 10:59:31 +08:00
|
|
|
address := ":" + strconv.Itoa(int(env.AppWebPort))
|
2025-02-19 14:23:58 +08:00
|
|
|
engine := gin.Default()
|
2025-05-14 15:13:44 +08:00
|
|
|
s.http = &http.Server{Addr: address, Handler: engine}
|
2025-02-19 14:23:58 +08:00
|
|
|
|
2025-03-08 10:59:31 +08:00
|
|
|
// 配置中间件和路由
|
|
|
|
|
Router(engine)
|
|
|
|
|
|
2025-02-19 14:23:58 +08:00
|
|
|
// 启动服务
|
2025-05-14 15:13:44 +08:00
|
|
|
err := s.http.ListenAndServe()
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (s *Server) Stop() error {
|
|
|
|
|
err := s.http.Shutdown(context.Background())
|
2025-02-19 14:23:58 +08:00
|
|
|
if err != nil {
|
2025-05-14 15:13:44 +08:00
|
|
|
return err
|
2025-02-19 14:23:58 +08:00
|
|
|
}
|
|
|
|
|
|
2025-03-08 10:59:31 +08:00
|
|
|
return nil
|
2025-02-19 14:23:58 +08:00
|
|
|
}
|