feat: 删除不需要文件夹

features/udm_user/model/udm_auth_user.go

@@ -0,0 +1,13 @@
+package model
+
+// UdmAuthUser UDM鉴权用户
+type UdmAuthUser struct {
+	ID        string `json:"id" xorm:"pk 'id' autoincr"`
+	Imsi      string `json:"imsi" xorm:"imsi"`            // SIM卡号
+	Amf       string `json:"amf" xorm:"amf"`              // ANF
+	Status    string `json:"status" xorm:"status"`        // 状态
+	Ki        string `json:"ki" xorm:"ki"`                // ki
+	AlgoIndex string `json:"algoIndex" xorm:"algo_index"` //
+	Opc       string `json:"opc" xorm:"opc"`
+	NeID      string `json:"neId" xorm:"ne_id"` // UDM网元标识-子系统
+}

features/udm_user/model/udm_sub_user.go

@@ -0,0 +1,29 @@
+package model
+
+// UdmSubUser UDM签约用户
+type UdmSubUser struct {
+	ID     string `json:"id"  xorm:"pk 'id' autoincr"`
+	Msisdn string `json:"msisdn" xorm:"msisdn"` // 相当手机号
+	Imsi   string `json:"imsi" xorm:"imsi"`     // SIM卡号
+	Ambr   string `json:"ambr" xorm:"ambr"`
+	Nssai  string `json:"nssai" xorm:"nssai"`
+	Rat    string `json:"rat" xorm:"rat"`
+	Arfb   string `json:"arfb" xorm:"arfb"`
+	Sar    string `json:"sar" xorm:"sar"`
+	Cn     string `json:"cn" xorm:"cn"`
+	SmData string `json:"smData" xorm:"sm_data"`
+	SmfSel string `json:"smfSel" xorm:"smf_sel"`
+	EpsDat string `json:"epsDat" xorm:"eps_dat"`
+	NeID   string `json:"neId" xorm:"ne_id"` // UDM网元标识-子系统
+
+	EpsFlag    string `json:"epsFlag" xorm:"eps_flag"`
+	EpsOdb     string `json:"epsOdb" xorm:"eps_odb"`
+	HplmnOdb   string `json:"hplmnOdb" xorm:"hplmn_odb"`
+	Ard        string `json:"ard" xorm:"ard"`
+	Epstpl     string `json:"epstpl" xorm:"epstpl"`
+	ContextId  string `json:"contextId" xorm:"context_id"`
+	ApnContext string `json:"apnContext" xorm:"apn_context"`
+	StaticIp   string `json:"staticIp" xorm:"static_ip"`
+
+	SubNum string `json:"subNum,omitempty" xorm:"-"` // 批量数
+}

features/udm_user/repo/repo_udm_auth_user.go

@@ -0,0 +1,274 @@
+package repo
+
+import (
+	"strconv"
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/lib/core/datasource"
+	"ems.agt/lib/core/utils/parse"
+	"ems.agt/lib/log"
+)
+
+// 实例化数据层 RepoUdmAuthUser 结构体
+var NewRepoUdmAuthUser = &RepoUdmAuthUser{
+	selectSql: `select 
+	id, imsi, amf, status, ki, algo_index, opc, ne_id 
+	from u_auth_user`,
+
+	resultMap: map[string]string{
+		"id":         "ID",
+		"imsi":       "Imsi",
+		"amf":        "Amf",
+		"status":     "Status",
+		"ki":         "Ki",
+		"algo_index": "AlgoIndex",
+		"opc":        "Opc",
+		"ne_id":      "NeID",
+	},
+}
+
+// RepoUdmAuthUser UDM鉴权用户 数据层处理
+type RepoUdmAuthUser struct {
+	// 查询视图对象SQL
+	selectSql string
+	// 结果字段与实体映射
+	resultMap map[string]string
+}
+
+// convertResultRows 将结果记录转实体结果组
+func (r *RepoUdmAuthUser) convertResultRows(rows []map[string]any) []model.UdmAuthUser {
+	arr := make([]model.UdmAuthUser, 0)
+	for _, row := range rows {
+		UdmUser := model.UdmAuthUser{}
+		for key, value := range row {
+			if keyMapper, ok := r.resultMap[key]; ok {
+				datasource.SetFieldValue(&UdmUser, keyMapper, value)
+			}
+		}
+		arr = append(arr, UdmUser)
+	}
+	return arr
+}
+
+// SelectPage 根据条件分页查询
+func (r *RepoUdmAuthUser) 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(concat('%', ?), '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["neId"]; ok && v != "" {
+		conditions = append(conditions, "ne_id = ?")
+		params = append(params, v)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	result := map[string]any{
+		"total": 0,
+		"rows":  []model.UdmAuthUser{},
+	}
+
+	// 查询数量 长度为0直接返回
+	totalSql := "select count(1) as 'total' from u_auth_user"
+	totalRows, err := datasource.RawDB("", totalSql+whereSql, params)
+	if err != nil {
+		log.Errorf("total err => %v", err)
+		return result
+	}
+	total := parse.Number(totalRows[0]["total"])
+	if total == 0 {
+		return result
+	} else {
+		result["total"] = total
+	}
+
+	// 分页
+	pageNum, pageSize := datasource.PageNumSize(query["pageNum"], query["pageSize"])
+	pageSql := " limit ?,? "
+	params = append(params, pageNum*pageSize)
+	params = append(params, pageSize)
+
+	// 排序
+	sortSql := ""
+	if v, ok := query["sortField"]; ok && v != "" {
+		if v == "imsi" {
+			sortSql += " order by imsi "
+		}
+		if v, ok := query["sortOrder"]; ok && v != nil {
+			if v == "desc" {
+				sortSql += " desc "
+			} else {
+				sortSql += " asc "
+			}
+		}
+	}
+
+	// 查询数据
+	querySql := r.selectSql + whereSql + sortSql + pageSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		log.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	result["rows"] = r.convertResultRows(results)
+	return result
+}
+
+// SelectList 根据实体查询
+func (r *RepoUdmAuthUser) SelectList(auth model.UdmAuthUser) []model.UdmAuthUser {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if auth.Imsi != "" {
+		conditions = append(conditions, "imsi = ?")
+		params = append(params, auth.Imsi)
+	}
+	if auth.NeID != "" {
+		conditions = append(conditions, "ne_id = ?")
+		params = append(params, auth.NeID)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	// 查询数据
+	querySql := r.selectSql + whereSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		log.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	return r.convertResultRows(results)
+}
+
+// ClearAndInsert 清空ne_id后新增实体
+func (r *RepoUdmAuthUser) ClearAndInsert(neID string, authArr []model.UdmAuthUser) int64 {
+	var num int64 = 0
+
+	// 清空指定ne_id
+	_, err := datasource.ExecDB("", "TRUNCATE TABLE u_auth_user", nil)
+	// _, err := datasource.ExecDB("", "DELETE FROM u_auth_user WHERE ne_id = ?", []any{neID})
+	if err != nil {
+		log.Errorf("TRUNCATE err => %v", err)
+	}
+
+	n := len(authArr)
+	batchSize := 2000
+	for i := 0; i < n; i += batchSize {
+		end := i + batchSize
+		if end > n {
+			end = n
+		}
+		batch := authArr[i:end]
+
+		// 调用 InsertMulti 函数将批量数据插入数据库
+		results, err := datasource.DefaultDB().Table("u_auth_user").InsertMulti(batch)
+		if err != nil {
+			continue
+		}
+		num += results
+	}
+	return num
+}
+
+// Insert 新增实体
+func (r *RepoUdmAuthUser) Insert(authUser model.UdmAuthUser) int64 {
+	results, err := datasource.DefaultDB().Table("u_auth_user").Insert(authUser)
+	if err != nil {
+		log.Errorf("Insert err => %v", err)
+		return results
+	}
+	return results
+}
+
+// Insert 批量添加
+func (r *RepoUdmAuthUser) Inserts(authUsers []model.UdmAuthUser) int64 {
+	var num int64
+	n := len(authUsers)
+	batchSize := 2000
+	for i := 0; i < n; i += batchSize {
+		end := i + batchSize
+		if end > n {
+			end = n
+		}
+		batch := authUsers[i:end]
+
+		// 调用 InsertMulti 函数将批量数据插入数据库
+		results, err := datasource.DefaultDB().Table("u_auth_user").InsertMulti(batch)
+		if err != nil {
+			continue
+		}
+		num += results
+	}
+	return num
+}
+
+// Update 修改更新
+func (r *RepoUdmAuthUser) Update(neID string, authUser model.UdmAuthUser) int64 {
+	// 查询先
+	var user model.UdmAuthUser
+	has, err := datasource.DefaultDB().Table("u_auth_user").Where("imsi = ? and ne_id = ?", authUser.Imsi, neID).Get(&user)
+	if !has || err != nil {
+		return 0
+	}
+
+	if authUser.Ki != "" && authUser.Ki != user.Ki {
+		user.Ki = authUser.Ki
+	}
+	if authUser.Amf != "" && authUser.Amf != user.Amf {
+		user.Amf = authUser.Amf
+	}
+	if authUser.AlgoIndex != "" && authUser.AlgoIndex != user.AlgoIndex {
+		user.AlgoIndex = authUser.AlgoIndex
+	}
+	if authUser.Opc != "" && authUser.Opc != user.Opc {
+		user.Opc = authUser.Opc
+	}
+
+	results, err := datasource.DefaultDB().Table("u_auth_user").Where("imsi = ? and ne_id = ?", user.Imsi, user.NeID).Update(user)
+	if err != nil {
+		return 0
+	}
+	return results
+}
+
+// Delete 删除实体
+func (r *RepoUdmAuthUser) Delete(neID, imsi string) int64 {
+	results, err := datasource.DefaultDB().Table("u_auth_user").Where("imsi = ? and ne_id = ?", imsi, neID).Delete()
+	if err != nil {
+		return results
+	}
+	return results
+}
+
+// Delete 删除范围实体
+func (r *RepoUdmAuthUser) Deletes(neID, imsi, num string) int64 {
+	imsiV, err := strconv.Atoi(imsi)
+	if err != nil {
+		return 0
+	}
+
+	numV, err := strconv.Atoi(num)
+	if err != nil {
+		return 0
+	}
+
+	results, err := datasource.DefaultDB().Table("u_auth_user").Where("imsi >= ? and imsi <= ? and ne_id = ?", imsiV, imsiV+numV, neID).Delete()
+	if err != nil {
+		return results
+	}
+	return results
+}

features/udm_user/repo/repo_udm_sub_user.go

@@ -0,0 +1,438 @@
+package repo
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/lib/core/datasource"
+	"ems.agt/lib/core/utils/parse"
+	"ems.agt/lib/log"
+)
+
+// 实例化数据层 RepoUdmSubUser 结构体
+var NewRepoUdmSubUser = &RepoUdmSubUser{
+	selectSql: `select 
+	id, msisdn, imsi, ambr, nssai, rat, arfb, sar, cn, sm_data, smf_sel, eps_dat, ne_id, eps_flag, eps_odb, hplmn_odb, ard, epstpl, context_id, apn_context, static_ip
+	from u_sub_user`,
+
+	resultMap: map[string]string{
+		"id":          "ID",
+		"msisdn":      "Msisdn",
+		"imsi":        "Imsi",
+		"ambr":        "Ambr",
+		"nssai":       "Nssai",
+		"rat":         "Rat",
+		"arfb":        "Arfb",
+		"sar":         "Sar",
+		"cn":          "Cn",
+		"sm_data":     "SmData",
+		"smf_sel":     "SmfSel",
+		"eps_dat":     "EpsDat",
+		"ne_id":       "NeID",
+		"eps_flag":    "EpsFlag",
+		"eps_odb":     "EpsOdb",
+		"hplmn_odb":   "HplmnOdb",
+		"ard":         "Ard",
+		"epstpl":      "Epstpl",
+		"context_id":  "ContextId",
+		"apn_context": "ApnContext",
+		"static_ip":   "StaticIp",
+	},
+}
+
+// RepoUdmSubUser UDM签约用户 数据层处理
+type RepoUdmSubUser struct {
+	// 查询视图对象SQL
+	selectSql string
+	// 结果字段与实体映射
+	resultMap map[string]string
+}
+
+// convertResultRows 将结果记录转实体结果组
+func (r *RepoUdmSubUser) convertResultRows(rows []map[string]any) []model.UdmSubUser {
+	arr := make([]model.UdmSubUser, 0)
+	for _, row := range rows {
+		UdmUser := model.UdmSubUser{}
+		for key, value := range row {
+			if keyMapper, ok := r.resultMap[key]; ok {
+				datasource.SetFieldValue(&UdmUser, keyMapper, value)
+			}
+		}
+		arr = append(arr, UdmUser)
+	}
+	return arr
+}
+
+// SelectPage 根据条件分页查询字典类型
+func (r *RepoUdmSubUser) SelectPage(query map[string]any) map[string]any {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if v, ok := query["msisdn"]; ok && v != "" {
+		conditions = append(conditions, "msisdn like concat(concat('%', ?), '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["imsi"]; ok && v != "" {
+		conditions = append(conditions, "imsi like concat(concat('%', ?), '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["neId"]; ok && v != "" {
+		conditions = append(conditions, "ne_id = ?")
+		params = append(params, v)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	result := map[string]any{
+		"total": 0,
+		"rows":  []model.UdmAuthUser{},
+	}
+
+	// 查询数量 长度为0直接返回
+	totalSql := "select count(1) as 'total' from u_sub_user"
+	totalRows, err := datasource.RawDB("", totalSql+whereSql, params)
+	if err != nil {
+		log.Errorf("total err => %v", err)
+		return result
+	}
+	total := parse.Number(totalRows[0]["total"])
+	if total == 0 {
+		return result
+	} else {
+		result["total"] = total
+	}
+
+	// 分页
+	pageNum, pageSize := datasource.PageNumSize(query["pageNum"], query["pageSize"])
+	pageSql := " limit ?,? "
+	params = append(params, pageNum*pageSize)
+	params = append(params, pageSize)
+
+	// 排序
+	sortSql := ""
+	if v, ok := query["sortField"]; ok && v != "" {
+		if v == "imsi" {
+			sortSql += " order by imsi "
+		}
+		if v, ok := query["sortOrder"]; ok && v != nil {
+			if v == "desc" {
+				sortSql += " desc "
+			} else {
+				sortSql += " asc "
+			}
+		}
+	}
+
+	// 查询数据
+	querySql := r.selectSql + whereSql + sortSql + pageSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		log.Errorf("query err => %v", err)
+		return result
+	}
+
+	// 转换实体
+	result["rows"] = r.convertResultRows(results)
+	return result
+}
+
+// SelectList 根据实体查询
+func (r *RepoUdmSubUser) SelectList(auth model.UdmSubUser) []model.UdmSubUser {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if auth.Imsi != "" {
+		conditions = append(conditions, "imsi = ?")
+		params = append(params, auth.Imsi)
+	}
+	if auth.NeID != "" {
+		conditions = append(conditions, "ne_id = ?")
+		params = append(params, auth.NeID)
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	// 查询数据
+	querySql := r.selectSql + whereSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		log.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	return r.convertResultRows(results)
+}
+
+// ClearAndInsert 清空ne_id后新增实体
+func (r *RepoUdmSubUser) ClearAndInsert(neID string, subArr []model.UdmSubUser) int64 {
+	var num int64 = 0
+
+	// 清空指定ne_id
+	_, err := datasource.ExecDB("", "TRUNCATE TABLE u_sub_user", nil)
+	// _, err := datasource.ExecDB("", "DELETE FROM u_sub_user WHERE ne_id = ?", []any{neID})
+	if err != nil {
+		log.Errorf("TRUNCATE err => %v", err)
+	}
+
+	n := len(subArr)
+	batchSize := 2000
+	for i := 0; i < n; i += batchSize {
+		end := i + batchSize
+		if end > n {
+			end = n
+		}
+		batch := subArr[i:end]
+
+		// 调用 InsertMulti 函数将批量数据插入数据库
+		results, err := datasource.DefaultDB().Table("u_sub_user").InsertMulti(batch)
+		if err != nil {
+			continue
+		}
+		num += results
+	}
+
+	// for _, u := range subArr {
+	// 	u.NeID = neID
+	// 	results, err := datasource.DefaultDB().Table("u_sub_user").Insert(u)
+	// 	if err != nil {
+	// 		return num
+	// 	}
+	// 	num += results
+	// }
+	return num
+}
+
+// Insert 新增实体
+func (r *RepoUdmSubUser) Insert(subUser model.UdmSubUser) int64 {
+	results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
+	if err != nil {
+		return results
+	}
+	return results
+}
+
+// Insert 批量添加
+func (r *RepoUdmSubUser) Inserts(subUser []model.UdmSubUser) int64 {
+	var num int64
+	n := len(subUser)
+	batchSize := 2000
+	for i := 0; i < n; i += batchSize {
+		end := i + batchSize
+		if end > n {
+			end = n
+		}
+		batch := subUser[i:end]
+
+		// 调用 InsertMulti 函数将批量数据插入数据库
+		results, err := datasource.DefaultDB().Table("u_sub_user").InsertMulti(batch)
+		if err != nil {
+			continue
+		}
+		num += results
+	}
+	return num
+}
+
+// Insert4G 批量添加4G用户
+func (r *RepoUdmSubUser) Insert4G(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil {
+		return 0
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return 0
+	}
+
+	subUser.NeID = neID
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
+		results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
+		if err == nil {
+			insertNum += results
+		}
+	}
+
+	return insertNum
+}
+
+// Update 修改更新
+func (r *RepoUdmSubUser) Update(neID string, authUser model.UdmSubUser) int64 {
+	// 查询先
+	var user model.UdmSubUser
+	has, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", authUser.Imsi, neID).Get(&user)
+	if !has || err != nil {
+		return 0
+	}
+
+	if authUser.Msisdn != "" && authUser.Msisdn != user.Msisdn {
+		user.Msisdn = authUser.Msisdn
+	}
+	if authUser.Ambr != "" && authUser.Ambr != user.Ambr {
+		user.Ambr = authUser.Ambr
+	}
+	if authUser.Arfb != "" && authUser.Arfb != user.Arfb {
+		user.Arfb = authUser.Arfb
+	}
+	if authUser.Sar != "" && authUser.Sar != user.Sar {
+		user.Sar = authUser.Sar
+	}
+	if authUser.Rat != "" && authUser.Rat != user.Rat {
+		user.Rat = authUser.Rat
+	}
+	if authUser.Cn != "" && authUser.Cn != user.Cn {
+		user.Cn = authUser.Cn
+	}
+	if authUser.SmfSel != "" && authUser.SmfSel != user.SmfSel {
+		user.SmfSel = authUser.SmfSel
+	}
+	if authUser.SmData != "" && authUser.SmData != user.SmData {
+		user.SmData = authUser.SmData
+	}
+	if authUser.EpsDat != "" && authUser.EpsDat != user.EpsDat {
+		user.EpsDat = authUser.EpsDat
+	}
+	if authUser.EpsFlag != "" && authUser.EpsFlag != user.EpsFlag {
+		user.EpsFlag = authUser.EpsFlag
+	}
+	if authUser.EpsOdb != "" && authUser.EpsDat != user.EpsDat {
+		user.EpsOdb = authUser.EpsOdb
+	}
+	if authUser.HplmnOdb != "" && authUser.HplmnOdb != user.HplmnOdb {
+		user.HplmnOdb = authUser.HplmnOdb
+	}
+	if authUser.Epstpl != "" && authUser.Epstpl != user.Epstpl {
+		user.Epstpl = authUser.Epstpl
+	}
+	if authUser.Ard != "" && authUser.Ard != user.Ard {
+		user.Ard = authUser.Ard
+	}
+	if authUser.ContextId != "" && authUser.ContextId != user.ContextId {
+		user.ContextId = authUser.ContextId
+	}
+	if authUser.ApnContext != "" && authUser.ApnContext != user.ApnContext {
+		user.ApnContext = authUser.ApnContext
+	}
+	if authUser.StaticIp != "" && authUser.StaticIp != user.StaticIp {
+		user.StaticIp = authUser.StaticIp
+	}
+
+	results, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", user.Imsi, user.NeID).Update(user)
+	if err != nil {
+		return 0
+	}
+	return results
+}
+
+// Update4GIP 批量修改4G IP
+func (r *RepoUdmSubUser) Update4GIP(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil || subUser.StaticIp == "" {
+		return insertNum
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return insertNum
+	}
+
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
+		// 查询先
+		var user model.UdmSubUser
+		has, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", subUser.Imsi, neID).Get(&user)
+		if has && err == nil {
+			// IP会自动递增
+			parts := strings.Split(subUser.StaticIp, ".")
+			lastPart := parts[3]
+			lastNum, _ := strconv.Atoi(lastPart)
+			lastNum += i
+			newLastPart := strconv.Itoa(lastNum)
+			parts[3] = newLastPart
+			newIP := strings.Join(parts, ".")
+			user.StaticIp = newIP
+			// 更新
+			results, err := datasource.DefaultDB().Table("u_sub_user").Update(user)
+			if err == nil {
+				insertNum += results
+			}
+		}
+	}
+	return insertNum
+}
+
+// UpdateSmData 批量修改sm-data
+func (r *RepoUdmSubUser) UpdateSmData(neID string, subUser model.UdmSubUser) int64 {
+	var insertNum int64
+
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil || subUser.StaticIp == "" {
+		return insertNum
+	}
+	numV, err := strconv.Atoi(subUser.SubNum)
+	if err != nil {
+		return insertNum
+	}
+
+	for i := 0; i < numV; i++ {
+		subUser.Imsi = fmt.Sprint(imsiV + i)
+
+		// 查询先
+		var user model.UdmSubUser
+		has, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", subUser.Imsi, neID).Get(&user)
+		if has && err == nil {
+			// IP会自动递增，需提前规划好DNN对应的IP；如dnn不需要绑定IP则不带此字段名
+			// parts := strings.Split(subUser.SmData, "&")
+			user.SmData = subUser.SmData
+			// 更新
+			results, err := datasource.DefaultDB().Table("u_sub_user").Update(user)
+			if err == nil {
+				insertNum += results
+			}
+		}
+	}
+	return insertNum
+}
+
+// Delete 删除实体
+func (r *RepoUdmSubUser) Delete(neID, imsi string) int64 {
+	results, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", imsi, neID).Delete()
+	if err != nil {
+		return results
+	}
+	return results
+}
+
+// Delete 删除范围实体
+func (r *RepoUdmSubUser) Deletes(neID, imsi, num string) int64 {
+	imsiV, err := strconv.Atoi(imsi)
+	if err != nil {
+		return 0
+	}
+
+	numV, err := strconv.Atoi(num)
+	if err != nil {
+		return 0
+	}
+
+	results, err := datasource.DefaultDB().Table("u_sub_user").Where("imsi >= ? and imsi <= ? and ne_id = ?", imsiV, imsiV+numV, neID).Delete()
+	if err != nil {
+		return results
+	}
+	return results
+}

features/udm_user/service/service_redis_data.go

@@ -0,0 +1,108 @@
+package service
+
+import (
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/lib/core/redis"
+)
+
+// phoneImsiList 获取所有imsi
+// func phoneImsiList() map[string]string {
+// 	phoneAndImsiArr := make(map[string]string, 0)
+// 	phoneKeys, err := redis.GetKeys("udmuser", "1*********")
+// 	if err != nil {
+// 		return phoneAndImsiArr
+// 	}
+// 	for _, phone := range phoneKeys {
+// 		imsi, err := redis.Get("udmuser", phone)
+// 		if err != nil {
+// 			continue
+// 		}
+// 		phoneAndImsiArr[phone] = imsi
+// 	}
+// 	return phoneAndImsiArr
+// }
+
+// redisUdmAuthUserList UDM鉴权用户
+func redisUdmAuthUserList() []model.UdmAuthUser {
+	user := []model.UdmAuthUser{}
+	ausfArr, err := redis.GetKeys("udmuser", "ausf:*")
+	if err != nil {
+		return user
+	}
+	for _, key := range ausfArr {
+		m, err := redis.GetHash("udmuser", key)
+		if err != nil {
+			continue
+		}
+
+		status := "0"
+		if _, ok := m["auth_success"]; ok {
+			status = "1"
+		}
+		amf := ""
+		if v, ok := m["amf"]; ok {
+			amf = strings.Replace(v, "\r\n", "", 1)
+		}
+		a := model.UdmAuthUser{
+			Imsi:      key[5:],
+			Amf:       amf,
+			Status:    status,
+			Ki:        m["ki"],
+			AlgoIndex: m["algo"],
+			Opc:       m["opc"],
+		}
+		user = append(user, a)
+	}
+	return user
+}
+
+// redisUdmSubUserList UDM签约用户
+func redisUdmSubUserList() []model.UdmSubUser {
+	user := []model.UdmSubUser{}
+	udmsdArr, err := redis.GetKeys("udmuser", "udm-sd:*")
+	if err != nil {
+		return user
+	}
+	for _, key := range udmsdArr {
+		m, err := redis.GetHash("udmuser", key)
+		if err != nil {
+			continue
+		}
+
+		a := model.UdmSubUser{
+			Imsi:   key[7:],
+			Msisdn: strings.TrimPrefix(m["gpsi"], "86"),
+			SmfSel: m["smf-sel"],
+			SmData: m["sm-dat"], // 1-000001&cmnet&ims&3gnet
+		}
+
+		// def_ambr,def_nssai,0,def_arfb,def_sar,3,1,12000,1,1000,0,1,-
+		if v, ok := m["am-dat"]; ok {
+			arr := strings.Split(v, ",")
+			a.Ambr = arr[0]
+			a.Nssai = arr[1]
+			a.Rat = arr[2]
+			a.Arfb = arr[3]
+			a.Sar = arr[4]
+			a.Cn = arr[5]
+		}
+		// 1,64,24,65,def_eps,1,2,010200000000,-
+		if v, ok := m["eps-dat"]; ok {
+			a.EpsDat = v
+			arr := strings.Split(v, ",")
+			a.EpsFlag = arr[0]
+			a.EpsOdb = arr[1]
+			a.HplmnOdb = arr[2]
+			a.Ard = arr[3]
+			a.Epstpl = arr[4]
+			a.ContextId = arr[5]
+			a.ApnContext = arr[7]
+			a.StaticIp = arr[8]
+		}
+
+		user = append(user, a)
+	}
+	return user
+}

features/udm_user/service/service_udm_auth_user.go

@@ -0,0 +1,143 @@
+package service
+
+import (
+	"fmt"
+	"strconv"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/features/udm_user/repo"
+)
+
+// 实例化服务层 ServiceUdmAuthUser 结构体
+var NewServiceUdmAuthUser = &ServiceUdmAuthUser{
+	repoAuthUser: *repo.NewRepoUdmAuthUser,
+}
+
+// ServiceUdmAuthUser UDM鉴权用户 服务层处理
+type ServiceUdmAuthUser struct {
+	repoAuthUser repo.RepoUdmAuthUser
+}
+
+// Save UDM鉴权用户-获取全部保存数据库
+func (r *ServiceUdmAuthUser) Save(neID string) int64 {
+	var num int64 = 0
+	authArr := redisUdmAuthUserList()
+	// 有数据才清空
+	if len(authArr) == 0 {
+		return num
+	}
+	go r.repoAuthUser.ClearAndInsert(neID, authArr)
+	return int64(len(authArr))
+}
+
+// Page UDM签约用户-分页查询数据库
+func (r *ServiceUdmAuthUser) Page(query map[string]any) map[string]any {
+	return r.repoAuthUser.SelectPage(query)
+}
+
+// List UDM签约用户-查询数据库
+func (r *ServiceUdmAuthUser) List(authUser model.UdmAuthUser) []model.UdmAuthUser {
+	return r.repoAuthUser.SelectList(authUser)
+}
+
+// Insert UDM鉴权用户-新增单个
+// imsi长度15，ki长度32，opc长度0或者32
+func (r *ServiceUdmAuthUser) Insert(neID string, authUser model.UdmAuthUser) int64 {
+	authUser.NeID = neID
+	authUser.Status = "1"
+	return r.repoAuthUser.Insert(authUser)
+}
+
+// Insert UDM鉴权用户-批量添加
+func (r *ServiceUdmAuthUser) Inserts(neID string, authUser model.UdmAuthUser, num string) int64 {
+	var arr []model.UdmAuthUser
+
+	imsiVlen := len(authUser.Imsi)
+	imsiV, err := strconv.Atoi(authUser.Imsi)
+	if err != nil {
+		return 0
+	}
+
+	numV, err := strconv.Atoi(num)
+	if err != nil {
+		return 0
+	}
+
+	authUser.NeID = neID
+	authUser.Status = "1"
+	for i := 0; i < numV; i++ {
+		imsi := fmt.Sprint(imsiV + i)
+		if len(imsi) < imsiVlen {
+			imsi = fmt.Sprintf("%0*s", imsiVlen, imsi)
+		}
+		authUser.Imsi = imsi
+		arr = append(arr, authUser)
+	}
+
+	return r.repoAuthUser.Inserts(arr)
+}
+
+// InsertCSV UDM鉴权用户-批量添加
+func (r *ServiceUdmAuthUser) InsertCSV(neID string, data []map[string]string) int64 {
+	var arr []model.UdmAuthUser
+	for _, v := range data {
+		var authUser model.UdmAuthUser
+		authUser.NeID = neID
+		authUser.Status = "1"
+		if s, ok := v["imsi"]; ok {
+			authUser.Imsi = s
+		}
+		if s, ok := v["ki"]; ok {
+			authUser.Ki = s
+		}
+		if s, ok := v["amf"]; ok {
+			authUser.Amf = s
+		}
+		if s, ok := v["algo"]; ok {
+			authUser.AlgoIndex = s
+		}
+		if s, ok := v["opc"]; ok {
+			authUser.Opc = s
+		}
+		arr = append(arr, authUser)
+	}
+	return r.repoAuthUser.Inserts(arr)
+}
+
+// InsertTxt UDM鉴权用户-批量添加
+func (r *ServiceUdmAuthUser) InsertTxt(neID string, data [][]string) int64 {
+	var arr []model.UdmAuthUser
+	for _, v := range data {
+		if len(v) < 4 {
+			continue
+		}
+		var authUser model.UdmAuthUser
+		authUser.NeID = neID
+		authUser.Status = "1"
+		authUser.Imsi = v[0]
+		authUser.Ki = v[1]
+		authUser.AlgoIndex = v[2]
+		authUser.Amf = v[3]
+		if len(v) == 5 {
+			authUser.Opc = v[4]
+		}
+		arr = append(arr, authUser)
+	}
+
+	return r.repoAuthUser.Inserts(arr)
+}
+
+// Insert UDM鉴权用户-修改更新
+func (r *ServiceUdmAuthUser) Update(neID string, authUser model.UdmAuthUser) int64 {
+	return r.repoAuthUser.Update(neID, authUser)
+}
+
+// Insert UDM鉴权用户-删除单个
+func (r *ServiceUdmAuthUser) Delete(neID, imsi string) int64 {
+	return r.repoAuthUser.Delete(neID, imsi)
+}
+
+// Insert UDM鉴权用户-删除范围
+func (r *ServiceUdmAuthUser) Deletes(neID, imsi, num string) int64 {
+	return r.repoAuthUser.Deletes(neID, imsi, num)
+}

features/udm_user/service/service_udm_sub_user.go

@@ -0,0 +1,201 @@
+package service
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/features/udm_user/repo"
+)
+
+// 实例化服务层 ServiceUdmSubUser 结构体
+var NewServiceUdmSubUser = &ServiceUdmSubUser{
+	repoSunUser: *repo.NewRepoUdmSubUser,
+}
+
+// ServiceUdmSubUser UDM签约用户 服务层处理
+type ServiceUdmSubUser struct {
+	repoSunUser repo.RepoUdmSubUser
+}
+
+// Save UDM签约用户-获取全部保存数据库
+func (r *ServiceUdmSubUser) Save(neID string) int64 {
+	var num int64 = 0
+	subArr := redisUdmSubUserList()
+	// 有数据才清空
+	if len(subArr) == 0 {
+		return num
+	}
+	go r.repoSunUser.ClearAndInsert(neID, subArr)
+	return int64(len(subArr))
+}
+
+// Page UDM签约用户-分页查询数据库
+func (r *ServiceUdmSubUser) Page(query map[string]any) map[string]any {
+	return r.repoSunUser.SelectPage(query)
+}
+
+// List UDM签约用户-查询数据库
+func (r *ServiceUdmSubUser) List(subUser model.UdmSubUser) []model.UdmSubUser {
+	return r.repoSunUser.SelectList(subUser)
+}
+
+// Insert UDM签约用户-新增单个
+// imsi长度15，ki长度32，opc长度0或者32
+func (r *ServiceUdmSubUser) Insert(neID string, subUser model.UdmSubUser) int64 {
+	subUser.NeID = neID
+	return r.repoSunUser.Insert(subUser)
+}
+
+// Insert UDM签约用户-批量添加
+func (r *ServiceUdmSubUser) Inserts(neID string, subUser model.UdmSubUser, num string) int64 {
+	var arr []model.UdmSubUser
+
+	imsiVlen := len(subUser.Imsi)
+	imsiV, err := strconv.Atoi(subUser.Imsi)
+	if err != nil {
+		return 0
+	}
+	msisdnVlen := len(subUser.Msisdn)
+	msisdnV, err := strconv.Atoi(subUser.Msisdn)
+	if err != nil {
+		return 0
+	}
+
+	numV, err := strconv.Atoi(num)
+	if err != nil {
+		return 0
+	}
+
+	subUser.NeID = neID
+	for i := 0; i < numV; i++ {
+		msisdn := fmt.Sprint(msisdnV + i)
+		if len(msisdn) < msisdnVlen {
+			msisdn = fmt.Sprintf("%0*s", msisdnVlen, msisdn)
+		}
+		subUser.Msisdn = msisdn
+
+		imsi := fmt.Sprint(imsiV + i)
+		if len(imsi) < imsiVlen {
+			imsi = fmt.Sprintf("%0*s", imsiVlen, imsi)
+		}
+		subUser.Imsi = imsi
+
+		// IP会自动递增
+		if subUser.StaticIp != "" {
+			parts := strings.Split(subUser.StaticIp, ".")
+			lastPart := parts[3]
+			lastNum, _ := strconv.Atoi(lastPart)
+			lastNum += i
+			newLastPart := strconv.Itoa(lastNum)
+			parts[3] = newLastPart
+			newIP := strings.Join(parts, ".")
+			subUser.StaticIp = newIP
+		}
+		arr = append(arr, subUser)
+	}
+
+	return r.repoSunUser.Inserts(arr)
+}
+
+// InsertCSV UDM签约用户-批量添加
+func (r *ServiceUdmSubUser) InsertCSV(neID string, data []map[string]string) int64 {
+	var arr []model.UdmSubUser
+	for _, v := range data {
+		var subUser model.UdmSubUser
+		subUser.NeID = neID
+		if s, ok := v["imsi"]; ok {
+			subUser.Imsi = s
+		}
+		if s, ok := v["msisdn"]; ok {
+			subUser.Msisdn = s
+		}
+		if s, ok := v["ambr"]; ok {
+			subUser.Ambr = s
+		}
+		if s, ok := v["nssai"]; ok {
+			subUser.Nssai = s
+		}
+		if s, ok := v["arfb"]; ok {
+			subUser.Arfb = s
+		}
+		if s, ok := v["sar"]; ok {
+			subUser.Sar = s
+		}
+		if s, ok := v["rat"]; ok {
+			subUser.Rat = s
+		}
+		if s, ok := v["cn"]; ok {
+			subUser.Cn = s
+		}
+		if s, ok := v["smf_sel"]; ok {
+			subUser.SmfSel = s
+		}
+		if s, ok := v["sm_data"]; ok {
+			subUser.SmData = s
+		}
+		if s, ok := v["eps_dat"]; ok {
+			subUser.EpsDat = s
+		}
+		arr = append(arr, subUser)
+	}
+	return r.repoSunUser.Inserts(arr)
+}
+
+// InsertTxt UDM签约用户-批量添加
+func (r *ServiceUdmSubUser) InsertTxt(neID string, data [][]string) int64 {
+	var arr []model.UdmSubUser
+	for _, v := range data {
+		if len(v) < 10 {
+			continue
+		}
+		var subUser model.UdmSubUser
+		subUser.NeID = neID
+		subUser.Imsi = v[0]
+		subUser.Msisdn = v[1]
+		subUser.Ambr = v[2]
+		subUser.Nssai = v[3]
+		subUser.Arfb = v[4]
+		subUser.Sar = v[5]
+		subUser.Rat = v[6]
+		subUser.Cn = v[7]
+		subUser.SmfSel = v[8]
+		subUser.SmData = v[9]
+		if len(v) == 11 {
+			subUser.EpsDat = v[10]
+		}
+		arr = append(arr, subUser)
+	}
+	return r.repoSunUser.Inserts(arr)
+}
+
+// Insert UDM签约用户-批量添加4G用户
+func (r *ServiceUdmSubUser) Insert4G(neID string, subUser model.UdmSubUser) int64 {
+	return r.repoSunUser.Insert4G(neID, subUser)
+}
+
+// Insert UDM签约用户-修改更新
+func (r *ServiceUdmSubUser) Update(neID string, subUser model.UdmSubUser) int64 {
+	return r.repoSunUser.Update(neID, subUser)
+}
+
+// Update4GIP UDM签约用户-批量修改4G IP
+func (r *ServiceUdmSubUser) Update4GIP(neID string, subUser model.UdmSubUser) int64 {
+	return r.repoSunUser.Update4GIP(neID, subUser)
+}
+
+// Update4GIP UDM签约用户-批量修改sm-data
+func (r *ServiceUdmSubUser) UpdateSmData(neID string, subUser model.UdmSubUser) int64 {
+	return r.repoSunUser.UpdateSmData(neID, subUser)
+}
+
+// Insert UDM签约用户-删除单个
+func (r *ServiceUdmSubUser) Delete(neID, imsi string) int64 {
+	return r.repoSunUser.Delete(neID, imsi)
+}
+
+// Insert UDM签约用户-删除范围
+func (r *ServiceUdmSubUser) Deletes(neID, imsi, num string) int64 {
+	return r.repoSunUser.Deletes(neID, imsi, num)
+}