78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package http
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/inconshreveable/log15"
|
|
"gitlab.33.cn/chat/dtalk/pkg/api"
|
|
"gitlab.33.cn/chat/dtalk/service/oss/service"
|
|
"net/http"
|
|
|
|
swaggerFiles "github.com/swaggo/files"
|
|
ginSwagger "github.com/swaggo/gin-swagger"
|
|
_ "gitlab.33.cn/chat/dtalk/service/oss/docs"
|
|
)
|
|
|
|
var (
|
|
svc *service.Service
|
|
log = log15.New("module", "oss/http")
|
|
)
|
|
|
|
func Init(s *service.Service) *http.Server {
|
|
addr := s.Config().Server.Addr
|
|
engine := Default()
|
|
InitService(s)
|
|
SetupEngine(engine)
|
|
|
|
srv := &http.Server{
|
|
Addr: addr,
|
|
Handler: engine,
|
|
}
|
|
go func() {
|
|
// service connections
|
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Error("engineInner.Start() error(%v)", err)
|
|
panic(err)
|
|
}
|
|
}()
|
|
return srv
|
|
}
|
|
|
|
// Default returns an Engine instance with the Logger and Recovery middleware already attached.
|
|
func Default() *gin.Engine {
|
|
router := gin.New()
|
|
// LoggerWithFormatter middleware will write the logs to gin.DefaultWriter
|
|
// By default gin.DefaultWriter = os.Stdout
|
|
router.Use(gin.LoggerWithFormatter(api.Chat33GinLogFormatter))
|
|
router.Use(gin.Recovery())
|
|
return router
|
|
}
|
|
|
|
func InitService(s *service.Service) {
|
|
svc = s
|
|
}
|
|
|
|
// SetupEngine
|
|
// @title 云存储服务接口
|
|
// @version 1.0
|
|
// @host 127.0.0.1:18005
|
|
func SetupEngine(e *gin.Engine) *gin.Engine {
|
|
root := e.Group("/", api.RespMiddleWare())
|
|
//获取服务器列表
|
|
root.Use(api.AuthMiddleWare())
|
|
{
|
|
root.POST("/get-token", GetOssToken)
|
|
root.GET("/get-token", GetOssToken)
|
|
root.POST("/get-huaweiyun-token", GetHuaweiyunOssToken)
|
|
root.GET("/get-huaweiyun-token", GetHuaweiyunOssToken)
|
|
root.POST("/upload", Upload)
|
|
root.POST("/init-multipart-upload", InitMultipartUpload)
|
|
root.POST("/upload-part", UploadPart)
|
|
root.POST("/complete-multipart-upload", CompleteMultipartUpload)
|
|
root.POST("/abort-multipart-upload", AbortMultipartUpload)
|
|
root.POST("/get-host", GetHost)
|
|
}
|
|
|
|
e.GET("/swagger/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
|
|
return e
|
|
}
|