131 lines
3.8 KiB
Go
131 lines
3.8 KiB
Go
package controller
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"be.ems/src/framework/i18n"
|
|
"be.ems/src/framework/utils/ctx"
|
|
"be.ems/src/framework/vo/result"
|
|
neService "be.ems/src/modules/network_element/service"
|
|
traceService "be.ems/src/modules/trace/service"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gin-gonic/gin/binding"
|
|
)
|
|
|
|
// 实例化控制层 TCPdumpController 结构体
|
|
var NewTCPdump = &TCPdumpController{
|
|
tcpdumpService: traceService.NewTCPdump,
|
|
neInfoService: neService.NewNeInfo,
|
|
}
|
|
|
|
// 信令抓包
|
|
//
|
|
// PATH /tcpdump
|
|
type TCPdumpController struct {
|
|
tcpdumpService *traceService.TCPdump // 信令抓包服务
|
|
neInfoService *neService.NeInfo // 网元信息服务
|
|
}
|
|
|
|
// 网元抓包PACP 开始
|
|
//
|
|
// POST /start
|
|
func (s *TCPdumpController) DumpStart(c *gin.Context) {
|
|
language := ctx.AcceptLanguage(c)
|
|
var body struct {
|
|
NeType string `json:"neType" binding:"required"` // 网元类型
|
|
NeId string `json:"neId" binding:"required"` // 网元ID
|
|
Cmd string `json:"cmd" binding:"required"` // 命令 "-n -s 0 -v"
|
|
}
|
|
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
|
if err != nil {
|
|
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
|
return
|
|
}
|
|
|
|
taskCode, err := s.tcpdumpService.DumpStart(body.NeType, body.NeId, body.Cmd)
|
|
if err != nil {
|
|
c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
c.JSON(200, result.OkData(taskCode))
|
|
}
|
|
|
|
// 网元抓包PACP 结束
|
|
//
|
|
// POST /stop
|
|
func (s *TCPdumpController) DumpStop(c *gin.Context) {
|
|
language := ctx.AcceptLanguage(c)
|
|
var body struct {
|
|
NeType string `json:"neType" binding:"required"` // 网元类型
|
|
NeId string `json:"neId" binding:"required"` // 网元ID
|
|
TaskCode string `json:"taskCode" binding:"required"` // 任务码，停止任务并查看日志信息
|
|
}
|
|
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
|
if err != nil {
|
|
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
|
return
|
|
}
|
|
|
|
taskLog, err := s.tcpdumpService.DumpStop(body.NeType, body.NeId, body.TaskCode)
|
|
if err != nil {
|
|
c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
c.JSON(200, result.OkMsg(taskLog))
|
|
}
|
|
|
|
// 网元抓包PACP 下载
|
|
//
|
|
// GET /download
|
|
func (s *TCPdumpController) DumpDownload(c *gin.Context) {
|
|
language := ctx.AcceptLanguage(c)
|
|
var query struct {
|
|
NeType string `form:"neType" binding:"required"` // 网元类型
|
|
NeID string `form:"neId" binding:"required"` // 网元ID
|
|
TaskCode string `form:"taskCode" binding:"required"` // 任务码，停止任务并查看日志信息
|
|
DelTemp bool `form:"delTemp"` // 完成后是否删除本地临时zip文件
|
|
}
|
|
if err := c.ShouldBindQuery(&query); err != nil {
|
|
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
|
return
|
|
}
|
|
|
|
zipFilePath, err := s.tcpdumpService.DumpDownload(query.NeType, query.NeID, query.TaskCode)
|
|
if err != nil {
|
|
c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
|
|
defer func() {
|
|
if query.DelTemp {
|
|
_ = os.Remove(zipFilePath)
|
|
}
|
|
}()
|
|
c.FileAttachment(zipFilePath, filepath.Base(zipFilePath))
|
|
}
|
|
|
|
// UPF标准版内部抓包
|
|
//
|
|
// POST /upf
|
|
func (s *TCPdumpController) UPFTrace(c *gin.Context) {
|
|
language := ctx.AcceptLanguage(c)
|
|
var body struct {
|
|
NeType string `json:"neType" binding:"required"` // 网元类型
|
|
NeId string `json:"neId" binding:"required"` // 网元ID
|
|
Cmd string `json:"cmd" binding:"required"` // 命令
|
|
}
|
|
err := c.ShouldBindBodyWith(&body, binding.JSON)
|
|
if err != nil {
|
|
c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
|
|
return
|
|
}
|
|
|
|
msg, err := s.tcpdumpService.UPFTrace(body.NeType, body.NeId, body.Cmd)
|
|
if err != nil {
|
|
c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
|
|
return
|
|
}
|
|
c.JSON(200, result.OkData(msg))
|
|
}
|