mirror of https://github.com/rocboss/paopao-ce
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.2 KiB
104 lines
2.2 KiB
// Copyright 2022 ROC. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/Masterminds/semver/v3"
|
|
"github.com/fatih/color"
|
|
sentrygin "github.com/getsentry/sentry-go/gin"
|
|
"github.com/gin-contrib/cors"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/rocboss/paopao-ce/internal/conf"
|
|
"github.com/rocboss/paopao-ce/internal/servants"
|
|
)
|
|
|
|
var (
|
|
_ Service = (*webService)(nil)
|
|
)
|
|
|
|
type webService struct {
|
|
*baseHttpService
|
|
}
|
|
|
|
func (s *webService) Name() string {
|
|
return "WebService"
|
|
}
|
|
|
|
func (s *webService) Version() *semver.Version {
|
|
return semver.MustParse("v0.1.0")
|
|
}
|
|
|
|
func (s *webService) OnInit() error {
|
|
s.registerRoute(s, servants.RegisterWebServants)
|
|
return nil
|
|
}
|
|
|
|
func (s *webService) String() string {
|
|
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.WebServerSetting.HttpIp, conf.WebServerSetting.HttpPort))
|
|
}
|
|
|
|
func newWebEngine() *gin.Engine {
|
|
e := gin.New()
|
|
e.HandleMethodNotAllowed = true
|
|
e.Use(gin.Logger())
|
|
e.Use(gin.Recovery())
|
|
|
|
// 跨域配置
|
|
corsConfig := cors.DefaultConfig()
|
|
corsConfig.AllowAllOrigins = true
|
|
corsConfig.AddAllowHeaders("Authorization")
|
|
e.Use(cors.New(corsConfig))
|
|
// 使用Sentry hook
|
|
if conf.UseSentryGin() {
|
|
e.Use(sentrygin.New(sentrygin.Options{
|
|
Repanic: true,
|
|
}))
|
|
}
|
|
|
|
// 默认404
|
|
e.NoRoute(func(c *gin.Context) {
|
|
c.JSON(http.StatusNotFound, gin.H{
|
|
"code": 404,
|
|
"msg": "Not Found",
|
|
})
|
|
})
|
|
|
|
// 默认405
|
|
e.NoMethod(func(c *gin.Context) {
|
|
c.JSON(http.StatusMethodNotAllowed, gin.H{
|
|
"code": 405,
|
|
"msg": "Method Not Allowed",
|
|
})
|
|
})
|
|
|
|
return e
|
|
}
|
|
|
|
func newWebService() Service {
|
|
addr := conf.WebServerSetting.HttpIp + ":" + conf.WebServerSetting.HttpPort
|
|
server := httpServers.from(addr, func() *httpServer {
|
|
engine := newWebEngine()
|
|
return &httpServer{
|
|
baseServer: newBaseServe(),
|
|
e: engine,
|
|
server: &http.Server{
|
|
Addr: addr,
|
|
Handler: engine,
|
|
ReadTimeout: conf.WebServerSetting.GetReadTimeout(),
|
|
WriteTimeout: conf.WebServerSetting.GetWriteTimeout(),
|
|
MaxHeaderBytes: 1 << 20,
|
|
},
|
|
}
|
|
})
|
|
return &webService{
|
|
baseHttpService: &baseHttpService{
|
|
server: server,
|
|
},
|
|
}
|
|
}
|