69 lines
1.9 KiB
Go
69 lines
1.9 KiB
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)
|
|
}
|
|
}
|