feat: 跟踪任务相关接口初始化

This commit is contained in:
TsMask
2024-09-20 17:28:00 +08:00
parent 08dee2a3c2
commit 3b507c447c

View File

@@ -5,6 +5,7 @@ import (
"be.ems/src/framework/middleware"
"be.ems/src/framework/middleware/collectlogs"
"be.ems/src/modules/trace/controller"
"be.ems/src/modules/trace/service"
"github.com/gin-gonic/gin"
)
@@ -13,6 +14,9 @@ import (
func Setup(router *gin.Engine) {
logger.Infof("开始加载 ====> trace 模块路由")
// 启动时需要的初始参数
InitLoad()
traceGroup := router.Group("/trace")
// 信令抓包
@@ -39,32 +43,76 @@ func Setup(router *gin.Engine) {
)
}
// 跟踪任务 网元HLR
// 跟踪任务 网元HLR (免登录)
taskHLRGroup := traceGroup.Group("/task/hlr")
{
taskHLRGroup.GET("/list",
middleware.PreAuthorize(nil),
controller.NewTraceTaskHlr.List,
)
taskHLRGroup.DELETE("/:ids",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_DELETE)),
controller.NewTraceTaskHlr.Remove,
)
taskHLRGroup.POST("/start",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_INSERT)),
controller.NewTraceTaskHlr.Start,
)
taskHLRGroup.POST("/stop",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_UPDATE)),
controller.NewTraceTaskHlr.Stop,
)
taskHLRGroup.POST("/file",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_OTHER)),
controller.NewTraceTaskHlr.File,
)
taskHLRGroup.GET("/filePull",
controller.NewTraceTaskHlr.FilePull,
)
}
// 跟踪任务
taskGroup := traceGroup.Group("/task")
{
taskGroup.GET("/list",
middleware.PreAuthorize(nil),
controller.NewTraceTask.List,
)
taskGroup.GET("/:id",
middleware.PreAuthorize(nil),
controller.NewTraceTask.Info,
)
taskGroup.POST("",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_INSERT)),
controller.NewTraceTask.Add,
)
taskGroup.PUT("",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_UPDATE)),
controller.NewTraceTask.Edit,
)
taskGroup.DELETE("/:ids",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.task", collectlogs.BUSINESS_TYPE_DELETE)),
controller.NewTraceTask.Remove,
)
}
// 跟踪数据
taskDataGroup := traceGroup.Group("/data")
{
taskDataGroup.GET("/list",
middleware.PreAuthorize(nil),
controller.NewTraceData.List,
)
taskDataGroup.DELETE("/:ids",
middleware.PreAuthorize(nil),
collectlogs.OperateLog(collectlogs.OptionNew("log.operate.title.taskData", collectlogs.BUSINESS_TYPE_DELETE)),
controller.NewTraceData.Remove,
)
}
}
// InitLoad 初始参数
func InitLoad() {
// 创建跟踪任务信令数据通道UDP
if err := service.NewTraceTask.CreateUDP(); err != nil {
logger.Errorf("socket udp init fail: %s", err.Error())
}
}