feat: statistics: finish dev

This commit is contained in:
fallen-angle
2022-01-23 18:28:01 +08:00
parent 582807ae10
commit d844711191
29 changed files with 1670 additions and 51 deletions

View File

@@ -22,4 +22,5 @@ func initConfig() {
}
global.ServerSettings = serverConfig
color.Blue("Already read config from %s", v.ConfigFileUsed())
color.Blue("Env: %s", global.ServerSettings.Env)
}

14
initialize/cron.go Normal file
View File

@@ -0,0 +1,14 @@
package initialize
import (
"nCovTrack-Backend/service/statistics"
"github.com/robfig/cron/v3"
)
func initCron() {
c := cron.New()
//c.AddFunc("@every 10s", func() { global.Redis.Set("OK", time.Now().String(), time.Duration(10*time.Hour)) })
c.AddFunc("@every 10m", statistics.CacheNCov)
c.Start()
}

View File

@@ -7,5 +7,9 @@ func Initialize() *gin.Engine {
initLogger()
initHttpClient()
initMySQL()
return initRouter()
initRedis()
initCron()
g := initRouter()
initSwagger()
return g
}

21
initialize/redis.go Normal file
View File

@@ -0,0 +1,21 @@
package initialize
import (
"fmt"
"nCovTrack-Backend/global"
"github.com/go-redis/redis"
)
func initRedis() {
rds := redis.NewClient(&redis.Options{
Addr: fmt.Sprintf("%s:%d", global.ServerSettings.Redis.Host, global.ServerSettings.Redis.Port),
Password: global.ServerSettings.Redis.Password,
DB: 0,
})
_, err := rds.Ping().Result()
if err != nil {
panic(err)
}
global.Redis = rds
}

View File

@@ -12,7 +12,7 @@ func initRouter() *gin.Engine {
r := gin.New()
r.Use(middleware.Cors(), middleware.RequestLog(), middleware.GinRecovery(true))
//r.Use(middleware.Cors(), middleware.RequestLog(), gin.Recovery())
global.RootRouter = r.Group(global.ServerSettings.UrlPrefix)
global.RootRouter = r.Group("/api" + global.ServerSettings.UrlPrefix)
router.BusiRouter()
return r
}

14
initialize/swagger.go Normal file
View File

@@ -0,0 +1,14 @@
package initialize
import (
"nCovTrack-Backend/docs"
"nCovTrack-Backend/global"
swaggerFile "github.com/swaggo/files"
swagger "github.com/swaggo/gin-swagger"
)
func initSwagger() {
docs.SwaggerInfo.BasePath = "/api" + global.ServerSettings.UrlPrefix
global.RootRouter.GET("/swagger/*any", swagger.WrapHandler(swaggerFile.Handler))
}