OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

新增补充UDM用户数据接口

Browse Source
This commit is contained in:
TsMask
2023-09-11 21:02:56 +08:00
parent 449d81c950
commit 614e792670
5 changed files with 1087 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
features/udm_user/repo/repo_udm_auth_user.go
View File

@@ -1,6 +1,8 @@
package repo
import (
"fmt"
"strconv"
"strings"
"ems.agt/features/udm_user/model"
@@ -22,7 +24,7 @@ var NewRepoUdmAuthUser = &RepoUdmAuthUser{
"amf": "Amf",
"status": "Status",
"ki": "Ki",
"algo_index": "AlgoLndex",
"algo_index": "AlgoIndex",
"opc": "Opc",
"ne_id": "NeID",
},
@@ -60,6 +62,10 @@ func (r *RepoUdmAuthUser) SelectPage(query map[string]any) map[string]any {
conditions = append(conditions, "msisdn like concat(?, '%')")
params = append(params, v)
}
if v, ok := query["imsi"]; ok && v != "" {
conditions = append(conditions, "imsi like concat(?, '%')")
params = append(params, v)
}
if v, ok := query["neId"]; ok && v != "" {
conditions = append(conditions, "ne_id = ?")
params = append(params, v)
@@ -137,8 +143,8 @@ func (r *RepoUdmAuthUser) SelectList(auth model.UdmAuthUser) []model.UdmAuthUser
return r.convertResultRows(results)
}
// Insert 清空ne_id后新增实体
func (r *RepoUdmAuthUser) Inserts(neID string, authArr []model.UdmAuthUser) int64 {
// ClearAndInsert 清空ne_id后新增实体
func (r *RepoUdmAuthUser) ClearAndInsert(neID string, authArr []model.UdmAuthUser) int64 {
var num int64 = 0
// 清空指定ne_id
@@ -158,3 +164,98 @@ func (r *RepoUdmAuthUser) Inserts(neID string, authArr []model.UdmAuthUser) int6
}
return num
}
// Insert 新增实体
func (r *RepoUdmAuthUser) Insert(neID string, authUser model.UdmAuthUser) int64 {
authUser.NeID = neID
authUser.Status = "1"
results, err := datasource.DefaultDB().Table("u_auth_user").Insert(authUser)
if err != nil {
return results
}
return results
}
// Insert 批量添加
func (r *RepoUdmAuthUser) Inserts(neID string, authUser model.UdmAuthUser, num string) int64 {
var insertNum int64
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++ {
authUser.Imsi = fmt.Sprint(imsiV + i)
results, err := datasource.DefaultDB().Table("u_auth_user").Insert(authUser)
if err == nil {
insertNum += results
}
}
return insertNum
}
// Update 修改更新
func (r *RepoUdmAuthUser) Update(neID string, authUser model.UdmAuthUser) int64 {
// 查询先
var user model.UdmAuthUser
err := datasource.DefaultDB().Table("u_auth_user").Where("imsi = ? and ne_id = ?", authUser.Imsi, neID).Find(&user)
if 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").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
}

156
features/udm_user/repo/repo_udm_sub_user.go
View File

@@ -1,6 +1,8 @@
package repo
import (
"fmt"
"strconv"
"strings"
"ems.agt/features/udm_user/model"
@@ -12,22 +14,31 @@ import (
// 实例化数据层 RepoUdmSubUser 结构体
var NewRepoUdmSubUser = &RepoUdmSubUser{
selectSql: `select
id, msisdn, imsi, sub_ue_ambr_temp, sub_snssai_temp, rai, forbidden_areas_temp, service_area_restrict_temp, cn_type, sub_data, eps, ne_id
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",
"sub_ue_ambr_temp": "SubUeAMBRTemp",
"sub_snssai_temp": "subSNSSAITemp",
"rai": "Rai",
"forbidden_areas_temp": "ForbiddenAreasTemp",
"service_area_restrict_temp": "ServiceAreaRestrictTemp",
"cn_type": "CnType",
"sub_data": "SubData",
"eps": "Eps",
"ne_id": "NeID",
"id": "ID",
"msisdn": "Msisdn",
"imsi": "Imsi",
"ambr": "Ambr",
"nssai": "Assai",
"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",
},
}
@@ -144,8 +155,8 @@ func (r *RepoUdmSubUser) SelectList(auth model.UdmSubUser) []model.UdmSubUser {
return r.convertResultRows(results)
}
// Insert 清空ne_id后新增实体
func (r *RepoUdmSubUser) Inserts(neID string, authArr []model.UdmSubUser) int64 {
// ClearAndInsert 清空ne_id后新增实体
func (r *RepoUdmSubUser) ClearAndInsert(neID string, subArr []model.UdmSubUser) int64 {
var num int64 = 0
// 清空指定ne_id
@@ -155,7 +166,7 @@ func (r *RepoUdmSubUser) Inserts(neID string, authArr []model.UdmSubUser) int64
log.Errorf("TRUNCATE err => %v", err)
}
for _, u := range authArr {
for _, u := range subArr {
u.NeID = neID
results, err := datasource.DefaultDB().Table("u_sub_user").Insert(u)
if err != nil {
@@ -165,3 +176,116 @@ func (r *RepoUdmSubUser) Inserts(neID string, authArr []model.UdmSubUser) int64
}
return num
}
// Insert 新增实体
func (r *RepoUdmSubUser) Insert(neID string, subUser model.UdmSubUser) int64 {
subUser.NeID = neID
results, err := datasource.DefaultDB().Table("u_sub_user").Insert(subUser)
if err != nil {
return results
}
return results
}
// Insert 批量添加
func (r *RepoUdmSubUser) Inserts(neID string, subUser model.UdmSubUser, num string) int64 {
var insertNum int64
imsiV, err := strconv.Atoi(subUser.Imsi)
if err != nil {
return 0
}
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++ {
subUser.Imsi = fmt.Sprint(imsiV + i)
subUser.Msisdn = fmt.Sprint(msisdnV + 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
err := datasource.DefaultDB().Table("u_sub_user").Where("imsi = ? and ne_id = ?", authUser.Imsi, neID).Find(&user)
if 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
}
results, err := datasource.DefaultDB().Table("u_sub_user").Update(user)
if err != nil {
return 0
}
return results
}
// 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
}