This commit is contained in:
zhuyasen
2023-03-17 00:13:48 +08:00
parent 0314755988
commit 43d1fa8e6e
2 changed files with 10 additions and 4 deletions

View File

@@ -9,6 +9,7 @@ import (
"github.com/zhufuyi/sponge/pkg/gin/handlerfunc"
"github.com/zhufuyi/sponge/pkg/gin/middleware"
"github.com/zhufuyi/sponge/pkg/gin/validator"
"github.com/zhufuyi/sponge/pkg/logger"
"github.com/gin-gonic/gin"
"github.com/gin-gonic/gin/binding"
@@ -18,12 +19,14 @@ import (
var staticFS embed.FS // index.html in the static directory
// NewRouter create a router
func NewRouter() *gin.Engine {
func NewRouter(isLog ...bool) *gin.Engine {
gin.SetMode(gin.ReleaseMode)
r := gin.New()
r.Use(gin.Recovery())
r.Use(middleware.Cors())
//r.Use(middleware.Logging(middleware.WithLog(logger.Get())))
if len(isLog) > 0 && isLog[0] {
r.Use(middleware.Logging(middleware.WithLog(logger.Get())))
}
binding.Validator = validator.Init()
// solve vue using history route 404 problem
@@ -43,10 +46,10 @@ func NewRouter() *gin.Engine {
}
// RunHTTPServer run http server
func RunHTTPServer(addr string) {
func RunHTTPServer(addr string, isLog ...bool) {
initRecord()
router := NewRouter()
router := NewRouter(isLog...)
server := &http.Server{
Addr: addr,
Handler: router,

View File

@@ -28,3 +28,6 @@ fi
cd ~/app/${serviceName}-binary
chmod +x run.sh
./run.sh
checkResult $?
echo "server directory is ~/app/${serviceName}-binary"