84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package api
|
||
|
||
import (
|
||
"easyvqd/internal/core/host"
|
||
"easyvqd/internal/core/media"
|
||
"easyvqd/internal/core/vqd"
|
||
"easyvqd/internal/core/vqd/store/audioencodedb"
|
||
"easyvqd/internal/core/vqdsdk"
|
||
"net/http"
|
||
|
||
"easyvqd/domain/uniqueid"
|
||
"easyvqd/domain/uniqueid/store/uniqueiddb"
|
||
"easyvqd/domain/version/versionapi"
|
||
"easyvqd/internal/conf"
|
||
|
||
"git.lnton.com/lnton/pkg/orm"
|
||
"git.lnton.com/lnton/pkg/web"
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/google/wire"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
var (
|
||
ProviderVersionSet = wire.NewSet(versionapi.NewVersionCore)
|
||
ProviderSet = wire.NewSet(
|
||
wire.Struct(new(Usecase), "*"),
|
||
NewHTTPHandler,
|
||
NewVqdTaskCore,
|
||
NewVqdTaskAPI,
|
||
versionapi.New,
|
||
host.NewCore,
|
||
media.NewCore,
|
||
)
|
||
)
|
||
|
||
type Usecase struct {
|
||
Conf *conf.Bootstrap
|
||
DB *gorm.DB
|
||
Version versionapi.API
|
||
VqdTaskCore *vqd.Core
|
||
VqdTaskAPI VqdTaskAPI
|
||
HostCore *host.Core
|
||
MediaCore *media.Core
|
||
VqdSdkCore *vqdsdk.Core
|
||
}
|
||
|
||
// NewHTTPHandler 生成Gin框架路由内容
|
||
func NewHTTPHandler(uc *Usecase) http.Handler {
|
||
cfg := uc.Conf
|
||
// 检查是否设置了 JWT 密钥，如果未设置，则生成一个长度为 32 的随机字符串作为密钥
|
||
if cfg.Server.HTTP.JwtSecret == "" {
|
||
uc.Conf.Server.HTTP.JwtSecret = orm.GenerateRandomString(32)
|
||
}
|
||
// 如果不处于调试模式，将 Gin 设置为发布模式
|
||
if !uc.Conf.Debug {
|
||
gin.SetMode(gin.ReleaseMode)
|
||
}
|
||
g := gin.New()
|
||
// 处理未找到路由的情况，返回 JSON 格式的 404 错误信息
|
||
|
||
// 如果启用了 Pprof，设置 Pprof 监控
|
||
if cfg.Server.HTTP.PProf.Enabled {
|
||
web.SetupPProf(g, &cfg.Server.HTTP.PProf.AccessIps)
|
||
}
|
||
|
||
setupRouter(g, uc)
|
||
RegisterPluginGRPC(uc.HostCore, uc.MediaCore, uc)
|
||
// 这个一定要在最后执行，避免在迁移过程中发生panic导致迁移一半
|
||
uc.Version.RecordVersion()
|
||
return g
|
||
}
|
||
|
||
// NewUniqueID 生成唯一 id
|
||
func NewUniqueID(db *gorm.DB) uniqueid.Core {
|
||
store := uniqueiddb.NewDB(db).AutoMigrate(orm.GetEnabledAutoMigrate())
|
||
return uniqueid.NewCore(store, 6)
|
||
}
|
||
|
||
// NewVqdTaskCore 推流任务
|
||
func NewVqdTaskCore(db *gorm.DB) *vqd.Core {
|
||
store := audioencodedb.NewDB(db).AutoMigrate(orm.EnabledAutoMigrate)
|
||
return vqd.NewCore(store)
|
||
}
|