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.
68 lines
1.9 KiB
68 lines
1.9 KiB
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) |
|
} |
|
}
|
|
|