99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"91porn-server/app/appg"
|
|
"91porn-server/app/middleware/authuser"
|
|
"91porn-server/app/router"
|
|
"91porn-server/common"
|
|
"91porn-server/common/cachev2"
|
|
"91porn-server/common/conf"
|
|
"91porn-server/common/db"
|
|
"91porn-server/common/log"
|
|
"91porn-server/common/services"
|
|
"91porn-server/models"
|
|
"91porn-server/models/Init"
|
|
|
|
fw "git.ruishengces.com/pub_tool/autotest"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type Env struct {
|
|
AppToken string // APP 用户 token
|
|
TestUID uint64
|
|
}
|
|
|
|
func (e *Env) Start(cfg fw.Config, base *fw.BaseEnv) {
|
|
time.Local, _ = time.LoadLocation("Asia/Shanghai")
|
|
|
|
// 读取配置
|
|
appCfg := appg.GlobalConfig{}
|
|
if err := conf.LoadJSON("../../app/config/app.json", &appCfg); err != nil {
|
|
panic("load config error: " + err.Error())
|
|
}
|
|
appg.Conf = &appCfg
|
|
log.Init(log.Options{Level: appCfg.Log.Level, DisableStack: appCfg.Log.DisableStacktrace})
|
|
common.InitGinSecret(appCfg.Base.ResKey, appCfg.Base.ResKey360, appCfg.Base.Env)
|
|
|
|
// 初始化 MongoDB
|
|
ds := db.InitDS(appg.InitMongodbParam(models.RegisterPool))
|
|
for k, v := range ds {
|
|
switch k {
|
|
case models.VideoDb:
|
|
appg.VideoDB = v
|
|
case models.LogDb:
|
|
appg.LogDB = v
|
|
case models.StatDb:
|
|
appg.StatDB = v
|
|
}
|
|
}
|
|
|
|
// 初始化 Redis
|
|
appg.InitRedis()
|
|
services.Init(appg.Redis)
|
|
cachev2.Init(appg.Redis, log.ZapLog)
|
|
|
|
// 初始化 Models
|
|
Init.InitMongo()
|
|
|
|
// 生成 APP 测试 token
|
|
e.TestUID = 300100
|
|
var err error
|
|
e.AppToken, err = authuser.GenToken(&authuser.Claims{
|
|
UID: e.TestUID,
|
|
Timestamp: time.Now().Unix(),
|
|
Type: 0,
|
|
})
|
|
if err != nil {
|
|
panic("gen token error: " + err.Error())
|
|
}
|
|
// 缓存到 Redis
|
|
tokenKey := fmt.Sprintf("token:user:%d", e.TestUID)
|
|
_ = appg.Redis.Set(tokenKey, e.AppToken, 24*time.Hour)
|
|
|
|
// 启动 HTTP 服务
|
|
gin.SetMode(gin.TestMode)
|
|
base.AppBaseURL = base.StartHTTP("APP", 13051, router.SetupGin())
|
|
}
|
|
|
|
func (e *Env) SetupData() {
|
|
// 基础数据由各 Suite 的 Setup 负责
|
|
}
|
|
|
|
func (e *Env) CleanData() {}
|
|
|
|
func (e *Env) Destroy() {
|
|
db.CloseDS()
|
|
_ = appg.RedisClose()
|
|
}
|
|
|
|
func (e *Env) Meta() []fw.ReportMeta {
|
|
return []fw.ReportMeta{
|
|
{Key: "项目", Value: "91porn-server"},
|
|
{Key: "模块", Value: "签到"},
|
|
{Key: "时间", Value: time.Now().Format("2006-01-02 15:04:05")},
|
|
}
|
|
}
|