openbilibili/app/interface/main/upload/http/http.go

package http

import (
	"net/http"

	"go-common/app/interface/main/upload/conf"
	xanti "go-common/app/interface/main/upload/http/antispam"
	"go-common/app/interface/main/upload/service"
	"go-common/library/log"
	bm "go-common/library/net/http/blademaster"
	"go-common/library/net/http/blademaster/middleware/auth"
	"go-common/library/net/http/blademaster/middleware/verify"
)

var (
	uploadSvr    *service.Service
	authInterSvr *auth.Auth
	authSvr      *auth.Auth
	verifySvr    *verify.Verify
	anti         *xanti.Antispam
)

// Init init http
func Init(c *conf.Config, s *service.Service) {
	initService(c, s)
	engine := bm.DefaultServer(c.BM)
	initRouter(engine)
	// init Outer serve
	if err := engine.Start(); err != nil {
		log.Error("xhttp.Serve error(%v)", err)
		panic(err)
	}
}

func initService(c *conf.Config, s *service.Service) {
	uploadSvr = s
	authInterSvr = auth.New(c.AuthInter)
	authSvr = auth.New(c.AuthOut)
	verifySvr = verify.New(nil)
	anti = xanti.New(c.Antispam, s.GetRateLimit) //mid+dir 限流
}

func initRouter(e *bm.Engine) {
	e.Ping(ping)
	uploadInternal := e.Group("/x/internal")
	{
		uploadInternal.POST("/upload", verifySvr.Verify, internalUpload)
		uploadInternal.POST("/upload/image", authInterSvr.User, anti.Handler(), internalUploadImage)
		uploadInternal.POST("/upload/admin/image", verifySvr.Verify, anti.Handler(), internalUploadAdminImage)
		uploadInternal.POST("/image/gen", verifySvr.Verify, anti.Handler(), genImageUpload)
	}

	upload := e.Group("/x/upload")
	{
		upload.POST("/image", uploadImagePublic)
		upload.POST("/app/image", authSvr.UserMobile, anti.Handler(), uploadMobileImage)
		upload.POST("/web/image", authSvr.UserWeb, anti.Handler(), uploadWebImage)
	}
}

// ping check server ok.
func ping(c *bm.Context) {
	var err error
	if err = uploadSvr.Ping(c); err != nil {
		log.Error("upload service ping error(%v)", err)
		c.AbortWithStatus(http.StatusServiceUnavailable)
	}
}