feat: 添加udm-voip/volte功能接口

2025-04-24 10:55:05 +08:00
parent cb1b0cde55
commit 081ef5c9cc
24 changed files with 2117 additions and 102 deletions
--- a/src/modules/network_data/repository/udm_auth.go
+++ b/src/modules/network_data/repository/udm_auth.go
@@ -18,7 +18,7 @@ type UDMAuthUser struct{}
 // ClearAndInsert 清空ne_id后新增实体
 func (r *UDMAuthUser) ClearAndInsert(neId string, uArr []model.UDMAuthUser) int64 {
 	// 不指定neID时，用 TRUNCATE 清空表快
-	// _, err := datasource.ExecDB("", "TRUNCATE TABLE u_auth_user", nil)
+	// _, err := datasource.ExecDB("", "TRUNCATE TABLE udm_auth", nil)
 	result := db.DB("").Where("ne_id = ?", neId).Unscoped().Delete(&model.UDMAuthUser{})
 	if result.Error != nil {
 		logger.Errorf("Delete err => %v", result.Error)
@@ -39,6 +39,7 @@ func (r *UDMAuthUser) SelectPage(query map[string]string) (int64, []model.UDMAut
 	if v, ok := query["imsis"]; ok && v != "" {
 		arr := strings.Split(v, ",")
 		tx = tx.Where("imsi in ?", arr)
+		// 勾选时，pageSize为勾选的数量
 		query["pageSize"] = fmt.Sprint(len(arr))
 	}

@@ -126,7 +127,7 @@ func (r *UDMAuthUser) Delete(imsi, neId string) int64 {
 }

 // DeletePrefixByIMSI 删除前缀匹配的实体
-func (r *UDMAuthUser) DeletePrefixByIMSI(neId, imsi string) int64 {
+func (r *UDMAuthUser) DeletePrefixByIMSI(imsi, neId string) int64 {
 	tx := db.DB("").Where("imsi like ? and ne_id = ?", fmt.Sprintf("%s%%", imsi), neId).Delete(&model.UDMAuthUser{})
 	if err := tx.Error; err != nil {
 		logger.Errorf("DeletePrefixByIMSI err => %v", err)
--- a/src/modules/network_data/repository/udm_sub.go
+++ b/src/modules/network_data/repository/udm_sub.go
@@ -18,7 +18,7 @@ type UDMSubUser struct{}
 // ClearAndInsert 清空ne_id后新增实体
 func (r *UDMSubUser) ClearAndInsert(neId string, u []model.UDMSubUser) int64 {
 	// 不指定neID时，用 TRUNCATE 清空表快
-	// _, err := datasource.ExecDB("", "TRUNCATE TABLE u_sub_user", nil)
+	// _, err := datasource.ExecDB("", "TRUNCATE TABLE udm_sub", nil)
 	result := db.DB("").Where("ne_id = ?", neId).Unscoped().Delete(&model.UDMSubUser{})
 	if result.Error != nil {
 		logger.Errorf("Delete err => %v", result.Error)
@@ -42,6 +42,7 @@ func (r *UDMSubUser) SelectPage(query map[string]string) (int64, []model.UDMSubU
 	if v, ok := query["imsis"]; ok && v != "" {
 		arr := strings.Split(v, ",")
 		tx = tx.Where("imsi in ?", arr)
+		// 勾选时，pageSize为勾选的数量
 		query["pageSize"] = fmt.Sprint(len(arr))
 	}

--- a/src/modules/network_data/repository/udm_voip.go
+++ b/src/modules/network_data/repository/udm_voip.go
@@ -0,0 +1,138 @@
+package repository
+
+import (
+	"fmt"
+	"strings"
+
+	"be.ems/src/framework/database/db"
+	"be.ems/src/framework/logger"
+	"be.ems/src/modules/network_data/model"
+)
+
+// 实例化数据层 UDMVOIPUser 结构体
+var NewUDMVOIPUser = &UDMVOIPUser{}
+
+// UDMVOIPUser UDMVOIP用户信息表 数据层处理
+type UDMVOIPUser struct{}
+
+// ClearAndInsert 清空ne_id后新增实体
+func (r UDMVOIPUser) ClearAndInsert(neId string, uArr []model.UDMVOIPUser) int64 {
+	// 不指定neID时，用 TRUNCATE 清空表快
+	// _, err := datasource.ExecDB("", "TRUNCATE TABLE udm_voip", nil)
+	result := db.DB("").Where("ne_id = ?", neId).Unscoped().Delete(&model.UDMVOIPUser{})
+	if result.Error != nil {
+		logger.Errorf("Delete err => %v", result.Error)
+	}
+	return r.Inserts(uArr)
+}
+
+// SelectPage 根据条件分页查询
+func (r UDMVOIPUser) SelectPage(query map[string]string) (int64, []model.UDMVOIPUser) {
+	tx := db.DB("").Model(&model.UDMVOIPUser{})
+	// 查询条件拼接
+	if v, ok := query["username"]; ok && v != "" {
+		tx = tx.Where("username like ?", fmt.Sprintf("%%%s%%", v))
+	}
+	if v, ok := query["neId"]; ok && v != "" {
+		tx = tx.Where("ne_id = ?", v)
+	}
+	if v, ok := query["usernames"]; ok && v != "" {
+		arr := strings.Split(v, ",")
+		tx = tx.Where("username in ?", arr)
+		// 勾选时，pageSize为勾选的数量
+		query["pageSize"] = fmt.Sprint(len(arr))
+	}
+
+	var total int64 = 0
+	rows := []model.UDMVOIPUser{}
+
+	// 查询数量 长度为0直接返回
+	if err := tx.Count(&total).Error; err != nil || total <= 0 {
+		return total, rows
+	}
+
+	// 分页
+	pageNum, pageSize := db.PageNumSize(query["pageNum"], query["pageSize"])
+	tx = tx.Offset(int(pageNum * pageSize)).Limit(int(pageSize))
+
+	// 排序
+	if v, ok := query["sortField"]; ok && v != "" {
+		sortSql := v
+		if o, ok := query["sortOrder"]; ok && o != "" {
+			if o == "desc" {
+				sortSql += " desc "
+			} else {
+				sortSql += " asc "
+			}
+		}
+		tx = tx.Order(sortSql)
+	} else {
+		tx = tx.Order("username asc")
+	}
+
+	// 查询数据
+	if err := tx.Find(&rows).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+
+	return total, rows
+}
+
+// SelectList 根据实体查询
+func (r UDMVOIPUser) SelectList(u model.UDMVOIPUser) []model.UDMVOIPUser {
+	tx := db.DB("").Model(&model.UDMVOIPUser{})
+	// 查询条件拼接
+	if u.UserName != "" {
+		tx = tx.Where("username = ?", u.UserName)
+	}
+	if u.NeId != "" {
+		tx = tx.Where("ne_id = ?", u.NeId)
+	}
+
+	// 查询数据
+	arr := []model.UDMVOIPUser{}
+	if err := tx.Order("username asc").Find(&arr).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+	return arr
+}
+
+// SelectByUserNameAndNeID 通过username和ne_id查询
+func (r UDMVOIPUser) SelectByUserNameAndNeID(username, neId string) model.UDMVOIPUser {
+	tx := db.DB("").Model(&model.UDMVOIPUser{})
+	item := model.UDMVOIPUser{}
+	// 查询条件拼接
+	tx = tx.Where("username = ? and ne_id = ?", username, neId)
+	// 查询数据
+	if err := tx.Order("username asc").Limit(1).Find(&item).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+	return item
+}
+
+// Insert 批量添加
+func (r UDMVOIPUser) Inserts(uArr []model.UDMVOIPUser) int64 {
+	tx := db.DB("").CreateInBatches(uArr, 500)
+	if err := tx.Error; err != nil {
+		logger.Errorf("CreateInBatches err => %v", err)
+	}
+	return tx.RowsAffected
+}
+
+// Delete 删除实体
+func (r UDMVOIPUser) Delete(username, neId string) int64 {
+	tx := db.DB("").Where("username = ? and ne_id = ?", username, neId).Delete(&model.UDMVOIPUser{})
+	if err := tx.Error; err != nil {
+		logger.Errorf("Delete err => %v", err)
+	}
+	return tx.RowsAffected
+}
+
+// DeletePrefixByUserName 删除前缀匹配的实体
+func (r UDMVOIPUser) DeletePrefixByUserName(username, neId string) int64 {
+	tx := db.DB("").Where("username like ? and ne_id = ?", fmt.Sprintf("%s%%", username), neId).Delete(&model.UDMVOIPUser{})
+	if err := tx.Error; err != nil {
+		logger.Errorf("DeletePrefixByUserName err => %v", err)
+	}
+	return tx.RowsAffected
+}
--- a/src/modules/network_data/repository/udm_volte_ims.go
+++ b/src/modules/network_data/repository/udm_volte_ims.go
@@ -0,0 +1,150 @@
+package repository
+
+import (
+	"fmt"
+	"strings"
+
+	"be.ems/src/framework/database/db"
+	"be.ems/src/framework/logger"
+	"be.ems/src/modules/network_data/model"
+)
+
+// 实例化数据层 UDMVolteIMSUser 结构体
+var NewUDMVolteIMSUser = &UDMVolteIMSUser{}
+
+// UDMVolteIMSUser UDMVOIP用户信息表 数据层处理
+type UDMVolteIMSUser struct{}
+
+// ClearAndInsert 清空ne_id后新增实体
+func (r UDMVolteIMSUser) ClearAndInsert(neId string, uArr []model.UDMVolteIMSUser) int64 {
+	// 不指定neID时，用 TRUNCATE 清空表快
+	// _, err := datasource.ExecDB("", "TRUNCATE TABLE udm_volte_ims", nil)
+	result := db.DB("").Where("ne_id = ?", neId).Unscoped().Delete(&model.UDMVolteIMSUser{})
+	if result.Error != nil {
+		logger.Errorf("Delete err => %v", result.Error)
+	}
+	return r.Inserts(uArr)
+}
+
+// SelectPage 根据条件分页查询
+func (r UDMVolteIMSUser) SelectPage(query map[string]string) (int64, []model.UDMVolteIMSUser) {
+	tx := db.DB("").Model(&model.UDMVolteIMSUser{})
+	// 查询条件拼接
+	if v, ok := query["imsi"]; ok && v != "" {
+		tx = tx.Where("imsi like ?", fmt.Sprintf("%%%s%%", v))
+	}
+	if v, ok := query["msisdn"]; ok && v != "" {
+		tx = tx.Where("msisdn like ?", fmt.Sprintf("%%%s%%", v))
+	}
+	if v, ok := query["neId"]; ok && v != "" {
+		tx = tx.Where("ne_id = ?", v)
+	}
+	if v, ok := query["tag"]; ok && v != "" {
+		tx = tx.Where("tag = ?", v)
+	}
+	if v, ok := query["vni"]; ok && v != "" {
+		tx = tx.Where("vni like ?", fmt.Sprintf("%%%s%%", v))
+	}
+	if v, ok := query["imsis"]; ok && v != "" {
+		arr := strings.Split(v, ",")
+		tx = tx.Where("imsi in ?", arr)
+		// 勾选时，pageSize为勾选的数量
+		query["pageSize"] = fmt.Sprint(len(arr))
+	}
+
+	var total int64 = 0
+	rows := []model.UDMVolteIMSUser{}
+
+	// 查询数量 长度为0直接返回
+	if err := tx.Count(&total).Error; err != nil || total <= 0 {
+		return total, rows
+	}
+
+	// 分页
+	pageNum, pageSize := db.PageNumSize(query["pageNum"], query["pageSize"])
+	tx = tx.Offset(int(pageNum * pageSize)).Limit(int(pageSize))
+
+	// 排序
+	if v, ok := query["sortField"]; ok && v != "" {
+		sortSql := v
+		if o, ok := query["sortOrder"]; ok && o != "" {
+			if o == "desc" {
+				sortSql += " desc "
+			} else {
+				sortSql += " asc "
+			}
+		}
+		tx = tx.Order(sortSql)
+	} else {
+		tx = tx.Order("imsi asc")
+	}
+
+	// 查询数据
+	if err := tx.Find(&rows).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+
+	return total, rows
+}
+
+// SelectList 根据实体查询
+func (r UDMVolteIMSUser) SelectList(u model.UDMVolteIMSUser) []model.UDMVolteIMSUser {
+	tx := db.DB("").Model(&model.UDMVolteIMSUser{})
+	// 查询条件拼接
+	if u.IMSI != "" {
+		tx = tx.Where("imsi = ?", u.IMSI)
+	}
+	if u.NeId != "" {
+		tx = tx.Where("ne_id = ?", u.NeId)
+	}
+	if u.Tag != "" {
+		tx = tx.Where("tag = ?", u.Tag)
+	}
+
+	// 查询数据
+	arr := []model.UDMVolteIMSUser{}
+	if err := tx.Order("user_name asc").Find(&arr).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+	return arr
+}
+
+// SelectByIMSIAndMSISDNAndNeID 通过imsi,msisdn,ne_id查询
+func (r UDMVolteIMSUser) SelectByIMSIAndMSISDNAndNeID(imsi, msisdn, neId string) model.UDMVolteIMSUser {
+	tx := db.DB("").Model(&model.UDMVolteIMSUser{})
+	item := model.UDMVolteIMSUser{}
+	// 查询条件拼接
+	tx = tx.Where("imsi = ? and msisdn = ? and ne_id = ?", imsi, msisdn, neId)
+	// 查询数据
+	if err := tx.Order("imsi asc").Limit(1).Find(&item).Error; err != nil {
+		logger.Errorf("query err => %v", err)
+	}
+	return item
+}
+
+// Insert 批量添加
+func (r UDMVolteIMSUser) Inserts(uArr []model.UDMVolteIMSUser) int64 {
+	tx := db.DB("").CreateInBatches(uArr, 500)
+	if err := tx.Error; err != nil {
+		logger.Errorf("CreateInBatches err => %v", err)
+	}
+	return tx.RowsAffected
+}
+
+// Delete 删除实体
+func (r UDMVolteIMSUser) Delete(imsi, neId string) int64 {
+	tx := db.DB("").Where("imsi = ? and ne_id = ?", imsi, neId).Delete(&model.UDMVolteIMSUser{})
+	if err := tx.Error; err != nil {
+		logger.Errorf("Delete err => %v", err)
+	}
+	return tx.RowsAffected
+}
+
+// DeletePrefixByIMSI 删除前缀匹配的实体
+func (r UDMVolteIMSUser) DeletePrefixByIMSI(imsi, neId string) int64 {
+	tx := db.DB("").Where("imsi like ? and ne_id = ?", fmt.Sprintf("%s%%", imsi), neId).Delete(&model.UDMVolteIMSUser{})
+	if err := tx.Error; err != nil {
+		logger.Errorf("DeletePrefixByIMSI err => %v", err)
+	}
+	return tx.RowsAffected
+}