77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
// entry function
|
||
package main
|
||
|
||
import (
|
||
"context"
|
||
"flag"
|
||
"time"
|
||
|
||
"modelRT/config"
|
||
"modelRT/database"
|
||
"modelRT/handler"
|
||
"modelRT/log"
|
||
"modelRT/middleware"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"go.uber.org/zap"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
var limiter *middleware.Limiter
|
||
|
||
func init() {
|
||
limiter = middleware.NewLimiter(10, 1*time.Minute) // 设置限流器，允许每分钟最多请求10次
|
||
}
|
||
|
||
var (
|
||
modelRTConfigDir = flag.String("modelRT_config_dir", "./config", "config file dir of model runtime service")
|
||
modelRTConfigName = flag.String("modelRT_config_name", "config", "config file name of model runtime service")
|
||
modelRTConfigType = flag.String("modelRT_config_type", "yaml", "config file type of model runtime service")
|
||
)
|
||
|
||
var (
|
||
modelRTConfig config.ModelRTConfig
|
||
postgresDBClient *gorm.DB
|
||
logger *zap.Logger
|
||
)
|
||
|
||
// TODO 使用 wire 依赖注入
|
||
func main() {
|
||
flag.Parse()
|
||
ctx := context.TODO()
|
||
|
||
modelRTConfig = config.ReadAndInitConfig(*modelRTConfigDir, *modelRTConfigName, *modelRTConfigType)
|
||
// TODO 创建 Redis 与 pg client
|
||
// init postgresDBClient
|
||
postgresDBClient = database.GetPostgresDBInstance(ctx, modelRTConfig.PostgresDBURI)
|
||
|
||
defer func() {
|
||
sqlDB, err := postgresDBClient.DB()
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
sqlDB.Close()
|
||
}()
|
||
|
||
// init logger
|
||
logger = log.GetLoggerInstance(modelRTConfig.LCfg)
|
||
defer logger.Sync()
|
||
|
||
// load circuit diagram from postgres
|
||
database.LoadCircuitDiagramFromPostgresDB(ctx, logger)
|
||
// TODO 从 pg 加载到 redis
|
||
engine := gin.Default()
|
||
engine.Use(limiter.Middleware)
|
||
engine.GET("/model/model_load", handler.ModelLoad)
|
||
engine.POST("/model/model_create", nil)
|
||
engine.POST("/model/model_update", nil)
|
||
engine.POST("/model/model_delete", nil)
|
||
|
||
// start route with 8080 port
|
||
engine.Run(":8080")
|
||
|
||
// Redis hashmap 母线模型、异步电动机模型
|
||
|
||
// kv key name value busx
|
||
}
|