feat: 跟踪数据列表查询接口

src/modules/trace/controller/trace_data.go

@@ -0,0 +1,62 @@
+package controller
+
+import (
+	"strings"
+
+	"be.ems/src/framework/i18n"
+	"be.ems/src/framework/utils/ctx"
+	"be.ems/src/framework/utils/parse"
+	"be.ems/src/framework/vo/result"
+	traceService "be.ems/src/modules/trace/service"
+	"github.com/gin-gonic/gin"
+)
+
+// 实例化控制层 TraceDataController 结构体
+var NewTraceData = &TraceDataController{
+	traceDataService: traceService.NewTraceData,
+}
+
+// 跟踪任务
+//
+// PATH /data
+type TraceDataController struct {
+	// 跟踪_数据信息服务
+	traceDataService *traceService.TraceData
+}
+
+// 跟踪任务列表
+//
+// GET /list
+func (s *TraceDataController) List(c *gin.Context) {
+	query := ctx.QueryMap(c)
+
+	// 查询数据
+	data := s.traceDataService.SelectPage(query)
+	c.JSON(200, result.Ok(data))
+}
+
+// 跟踪任务删除
+//
+// DELETE /:ids
+func (s *TraceDataController) Remove(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	rowIds := c.Param("ids")
+	if rowIds == "" {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+	// 处理字符转id数组后去重
+	ids := strings.Split(rowIds, ",")
+	uniqueIDs := parse.RemoveDuplicates(ids)
+	if len(uniqueIDs) <= 0 {
+		c.JSON(200, result.Err(nil))
+		return
+	}
+	rows, err := s.traceDataService.DeleteByIds(uniqueIDs)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
+		return
+	}
+	msg := i18n.TTemplate(language, "app.common.deleteSuccess", map[string]any{"num": rows})
+	c.JSON(200, result.OkMsg(msg))
+}

src/modules/trace/model/trace_data.go

@@ -0,0 +1,23 @@
+package model
+
+// TraceData 跟踪_数据 trace_data
+type TraceData struct {
+	ID        string `json:"id" gorm:"column:id;primaryKey;autoIncrement"`
+	TaskId    int64  `json:"taskId" gorm:"task_id"` // 任务ID
+	Imsi      string `json:"imsi" gorm:"imsi"`
+	Msisdn    string `json:"msisdn" gorm:"msisdn"`    // 可能存在
+	SrcAddr   string `json:"srcAddr" gorm:"src_addr"` // 源地址带端口
+	DstAddr   string `json:"dstAddr" gorm:"dst_addr"` // 目标地址带端口
+	IfType    int64  `json:"ifType" gorm:"if_type"`   // 接口类型，未分类
+	MsgType   int64  `json:"msgType" gorm:"msg_type"`
+	MsgDirect int64  `json:"msgDirect" gorm:"msg_direct"`
+	Length    int64  `json:"length" gorm:"length"`       // 去除头后的原始数据byte长度
+	Timestamp int64  `json:"timestamp" gorm:"timestamp"` // 毫秒
+	RawMsg    string `json:"rawMsg" gorm:"raw_msg"`      // 去除头后的原始数据byteBase64
+	DecMsg    string `json:"decMsg" gorm:"dec_msg"`      // TCP内容消息
+}
+
+// TableName 表名称
+func (*TraceData) TableName() string {
+	return "trace_data"
+}

src/modules/trace/repository/trace_data.go

@@ -0,0 +1,246 @@
+package repository
+
+import (
+	"strings"
+
+	"be.ems/src/framework/datasource"
+	"be.ems/src/framework/logger"
+	"be.ems/src/framework/utils/parse"
+	"be.ems/src/framework/utils/repo"
+	"be.ems/src/modules/trace/model"
+)
+
+// 实例化数据层 TraceData 结构体
+var NewTraceData = &TraceData{
+	selectSql: `select id, task_id, imsi, msisdn, src_addr, dst_addr, if_type, msg_type, msg_direct, length, timestamp, raw_msg, dec_msg from trace_data`,
+
+	resultMap: map[string]string{
+		"id":         "ID",
+		"task_id":    "TaskId",
+		"imsi":       "Imsi",
+		"msisdn":     "Msisdn",
+		"src_addr":   "SrcAddr",
+		"dst_addr":   "DstAddr",
+		"if_type":    "IfType",
+		"msg_type":   "MsgType",
+		"msg_direct": "MsgDirect",
+		"length":     "Length",
+		"timestamp":  "Timestamp",
+		"raw_msg":    "RawMsg",
+		"dec_msg":    "DecMsg",
+	},
+}
+
+// CDREventIMSImpl 跟踪_任务给HRL网元 数据层处理
+type TraceData struct {
+	// 查询视图对象SQL
+	selectSql string
+	// 结果字段与实体映射
+	resultMap map[string]string
+}
+
+// convertResultRows 将结果记录转实体结果组
+func (r *TraceData) convertResultRows(rows []map[string]any) []model.TraceData {
+	arr := make([]model.TraceData, 0)
+	for _, row := range rows {
+		item := model.TraceData{}
+		for key, value := range row {
+			if keyMapper, ok := r.resultMap[key]; ok {
+				repo.SetFieldValue(&item, keyMapper, value)
+			}
+		}
+		arr = append(arr, item)
+	}
+	return arr
+}
+
+// SelectPage 根据条件分页查询
+func (r *TraceData) SelectPage(query map[string]any) map[string]any {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if v, ok := query["imsi"]; ok && v != "" {
+		conditions = append(conditions, "imsi like concat(?, '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["msisdn"]; ok && v != "" {
+		conditions = append(conditions, "msisdn like concat(?, '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["startTime"]; ok && v != "" {
+		conditions = append(conditions, "timestamp >= ?")
+		params = append(params, v)
+	}
+	if v, ok := query["endTime"]; ok && v != "" {
+		conditions = append(conditions, "timestamp <= ?")
+		params = append(params, v)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	result := map[string]any{
+		"total": 0,
+		"rows":  []model.TraceData{},
+	}
+
+	// 查询数量 长度为0直接返回
+	totalSql := "select count(1) as 'total' from trace_data"
+	totalRows, err := datasource.RawDB("", totalSql+whereSql, params)
+	if err != nil {
+		logger.Errorf("total err => %v", err)
+		return result
+	}
+	total := parse.Number(totalRows[0]["total"])
+	if total == 0 {
+		return result
+	} else {
+		result["total"] = total
+	}
+
+	// 分页
+	pageNum, pageSize := repo.PageNumSize(query["pageNum"], query["pageSize"])
+	pageSql := " limit ?,? "
+	params = append(params, pageNum*pageSize)
+	params = append(params, pageSize)
+
+	// 查询数据
+	querySql := r.selectSql + whereSql + pageSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	result["rows"] = r.convertResultRows(results)
+	return result
+}
+
+// SelectList 根据实体查询
+func (r *TraceData) SelectList(data model.TraceData) []model.TraceData {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if data.Imsi != "" {
+		conditions = append(conditions, "imsi = ?")
+		params = append(params, data.Imsi)
+	}
+	if data.Msisdn != "" {
+		conditions = append(conditions, "msisdn = ?")
+		params = append(params, data.Msisdn)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	// 查询数据
+	querySql := r.selectSql + whereSql + " order by id desc "
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	return r.convertResultRows(results)
+}
+
+// SelectByIds 通过ID查询
+func (r *TraceData) SelectByIds(ids []string) []model.TraceData {
+	placeholder := repo.KeyPlaceholderByQuery(len(ids))
+	querySql := r.selectSql + " where id in (" + placeholder + ")"
+	parameters := repo.ConvertIdsSlice(ids)
+	results, err := datasource.RawDB("", querySql, parameters)
+	if err != nil {
+		logger.Errorf("query err => %v", err)
+		return []model.TraceData{}
+	}
+	// 转换实体
+	return r.convertResultRows(results)
+}
+
+// Insert 新增信息
+func (r *TraceData) Insert(data model.TraceData) string {
+	// 参数拼接
+	params := make(map[string]any)
+	if data.TaskId > 0 {
+		params["task_id"] = data.TaskId
+	}
+	if data.Imsi != "" {
+		params["imsi"] = data.Imsi
+	}
+	if data.Msisdn != "" {
+		params["msisdn"] = data.Msisdn
+	}
+	if data.SrcAddr != "" {
+		params["src_addr"] = data.SrcAddr
+	}
+	if data.DstAddr != "" {
+		params["dst_addr"] = data.DstAddr
+	}
+	if data.IfType > -1 {
+		params["if_type"] = data.IfType
+	}
+	if data.MsgType > -1 {
+		params["msg_type"] = data.MsgType
+	}
+	if data.MsgDirect > -1 {
+		params["msg_direct"] = data.MsgDirect
+	}
+	if data.Length > 0 {
+		params["length"] = data.Length
+	}
+	if data.Timestamp > 0 {
+		params["timestamp"] = data.Timestamp
+	}
+	if data.RawMsg != "" {
+		params["raw_msg"] = data.RawMsg
+	}
+	if data.DecMsg != "" {
+		params["dec_msg"] = data.DecMsg
+	}
+
+	// 构建执行语句
+	keys, placeholder, values := repo.KeyPlaceholderValueByInsert(params)
+	sql := "insert into trace_data (" + strings.Join(keys, ",") + ")values(" + placeholder + ")"
+
+	db := datasource.DefaultDB()
+	// 开启事务
+	tx := db.Begin()
+	// 执行插入
+	err := tx.Exec(sql, values...).Error
+	if err != nil {
+		logger.Errorf("insert row : %v", err.Error())
+		tx.Rollback()
+		return ""
+	}
+	// 获取生成的自增 ID
+	var insertedID string
+	err = tx.Raw("select last_insert_id()").Row().Scan(&insertedID)
+	if err != nil {
+		logger.Errorf("insert last id : %v", err.Error())
+		tx.Rollback()
+		return ""
+	}
+	// 提交事务
+	tx.Commit()
+	return insertedID
+}
+
+// DeleteByIds 批量删除信息
+func (r *TraceData) DeleteByIds(ids []string) int64 {
+	placeholder := repo.KeyPlaceholderByQuery(len(ids))
+	sql := "delete from trace_data where id in (" + placeholder + ")"
+	parameters := repo.ConvertIdsSlice(ids)
+	results, err := datasource.ExecDB("", sql, parameters)
+	if err != nil {
+		logger.Errorf("delete err => %v", err)
+		return 0
+	}
+	return results
+}

src/modules/trace/service/trace_data.go

@@ -0,0 +1,54 @@
+package service
+
+import (
+	"fmt"
+
+	"be.ems/src/modules/trace/model"
+	"be.ems/src/modules/trace/repository"
+)
+
+// 实例化数据层 TraceData 结构体
+var NewTraceData = &TraceData{
+	traceDataRepository: repository.NewTraceData,
+}
+
+// TraceData 跟踪_数据 服务层处理
+type TraceData struct {
+	// 跟踪_数据信息
+	traceDataRepository *repository.TraceData
+}
+
+// SelectPage 根据条件分页查询
+func (r *TraceData) SelectPage(query map[string]any) map[string]any {
+	return r.traceDataRepository.SelectPage(query)
+}
+
+// SelectById 通过ID查询
+func (r *TraceData) SelectById(id string) model.TraceData {
+	tasks := r.traceDataRepository.SelectByIds([]string{id})
+	if len(tasks) > 0 {
+		return tasks[0]
+	}
+	return model.TraceData{}
+}
+
+// Insert 新增信息
+func (r *TraceData) Insert(task model.TraceData) string {
+	return r.traceDataRepository.Insert(task)
+}
+
+// DeleteByIds 批量删除信息
+func (r *TraceData) DeleteByIds(ids []string) (int64, error) {
+	// 检查是否存在
+	rows := r.traceDataRepository.SelectByIds(ids)
+	if len(rows) <= 0 {
+		return 0, fmt.Errorf("not data")
+	}
+
+	if len(rows) == len(ids) {
+		rows := r.traceDataRepository.DeleteByIds(ids)
+		return rows, nil
+	}
+	// 删除信息失败！
+	return 0, fmt.Errorf("delete fail")
+}