From 1ab2d977567108737b89be37abae0384ed69ab4f Mon Sep 17 00:00:00 2001 From: TsMask <340112800@qq.com> Date: Tue, 23 Sep 2025 10:47:34 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20UDM=E6=95=B0=E6=8D=AE=E6=89=B9=E9=87=8F?= =?UTF-8?q?=E5=AF=BC=E5=85=A5=E6=8B=BC=E5=87=8F=E5=B0=91=E9=98=B2=E6=AD=A2?= =?UTF-8?q?=E6=BA=A2=E5=87=BA=E9=99=90=E5=88=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- features/ue/repository/ims_user.go | 4 +++- features/ue/repository/voip_auth.go | 4 +++- src/modules/network_data/repository/udm_auth.go | 5 +++-- src/modules/network_data/repository/udm_sub.go | 6 ++++-- 4 files changed, 13 insertions(+), 6 deletions(-) diff --git a/features/ue/repository/ims_user.go b/features/ue/repository/ims_user.go index 2d02f9a6..690a71d6 100644 --- a/features/ue/repository/ims_user.go +++ b/features/ue/repository/ims_user.go @@ -16,7 +16,7 @@ import ( // 实例化数据层 IMSUserRepository 结构体 var NewIMSUserRepository = &IMSUserRepository{ selectSql: `select - s.id, s.ne_id, s.imsi, s.msisdn, s.tag, s.vni, + s.id, s.ne_id, s.imsi, s.msisdn, s.tag, s.vni, s.create_time, s.created_at, t.tenant_id, t.tenant_name from u_ims_user s left join sys_tenant t on t.tenant_id = s.tenant_id and t.status = 1`, @@ -29,6 +29,8 @@ var NewIMSUserRepository = &IMSUserRepository{ "tag": "Tag", "vni": "VNI", + "create_time": "CreateTime", + "created_at": "CreatedAt", "tenant_id": "TenantID", "tenant_name": "TenantName", // Tenant name for multi-tenancy }, diff --git a/features/ue/repository/voip_auth.go b/features/ue/repository/voip_auth.go index 44f1e051..27b2b277 100644 --- a/features/ue/repository/voip_auth.go +++ b/features/ue/repository/voip_auth.go @@ -16,7 +16,7 @@ import ( // 实例化数据层 VoIPAuthRepository 结构体 var NewVoIPAuthRepository = &VoIPAuthRepository{ selectSql: `select - s.id, s.ne_id, s.user_name, s.password, + s.id, s.ne_id, s.user_name, s.password, s.create_time, s.created_at, t.tenant_id, t.tenant_name from u_voip_auth s left join sys_tenant t on t.tenant_id = s.tenant_id and t.status = 1`, @@ -27,6 +27,8 @@ var NewVoIPAuthRepository = &VoIPAuthRepository{ "user_name": "UserName", "password": "Password", + "create_time": "CreateTime", + "created_at": "CreatedAt", "tenant_id": "TenantID", "tenant_name": "TenantName", // Tenant name for multi-tenancy }, diff --git a/src/modules/network_data/repository/udm_auth.go b/src/modules/network_data/repository/udm_auth.go index a44c842a..4aa39016 100644 --- a/src/modules/network_data/repository/udm_auth.go +++ b/src/modules/network_data/repository/udm_auth.go @@ -13,7 +13,7 @@ import ( // 实例化数据层 UDMAuthUser 结构体 var NewUDMAuthUser = &UDMAuthUser{ - selectSql: `select id, imsi, ne_id, amf, status, ki, algo_index, opc, create_time from u_auth_user`, + selectSql: `select id, imsi, ne_id, amf, status, ki, algo_index, opc, create_time, created_at from u_auth_user`, resultMap: map[string]string{ "id": "ID", @@ -25,6 +25,7 @@ var NewUDMAuthUser = &UDMAuthUser{ "algo_index": "AlgoIndex", "opc": "Opc", "create_time": "CreateTime", + "created_at": "CreatedAt", }, } @@ -189,7 +190,7 @@ func (r *UDMAuthUser) SelectByIMSIAndNeID(imsi, neId string) model.UDMAuthUser { // Insert 批量添加 func (r *UDMAuthUser) Inserts(uArr []model.UDMAuthUser) int64 { - tx := datasource.DefaultDB().CreateInBatches(uArr, 3000) + tx := datasource.DefaultDB().CreateInBatches(uArr, 2000) if err := tx.Error; err != nil { logger.Errorf("CreateInBatches err => %v", err) } diff --git a/src/modules/network_data/repository/udm_sub.go b/src/modules/network_data/repository/udm_sub.go index bee472e6..e2cebd9c 100644 --- a/src/modules/network_data/repository/udm_sub.go +++ b/src/modules/network_data/repository/udm_sub.go @@ -19,7 +19,7 @@ var NewUDMSub = &UDMSubUser{ s.id, s.imsi, s.msisdn, s.ne_id, s.am_dat, s.ambr, s.nssai, s.rat, s.arfb, s.sar, s.cn_type, s.rfsp_index, s.reg_timer, s.ue_usage_type, s.active_time, s.mico, s.odb_ps, s.group_id, s.eps_dat, s.eps_flag, s.eps_odb, s.hplmn_odb, s.ard, s.epstpl, s.context_id, s.apn_mum, s.apn_context, s.static_ip, - s.sm_data, s.smf_sel, s.cag, + s.sm_data, s.smf_sel, s.cag, s.create_time, s.created_at, t.tenant_id, t.tenant_name from u_sub_user s left join sys_tenant t on t.tenant_id = s.tenant_id and t.status = 1`, @@ -60,6 +60,8 @@ var NewUDMSub = &UDMSubUser{ "smf_sel": "SmfSel", "cag": "Cag", + "create_time": "CreateTime", + "created_at": "CreatedAt", "tenant_id": "TenantID", "tenant_name": "TenantName", // Tenant name for multi-tenancy }, @@ -262,7 +264,7 @@ func (r *UDMSubUser) Inserts(uArr []model.UDMSubUser) int64 { // multi-tenancy r.SetTenantID(&uArr) - tx := datasource.DefaultDB().CreateInBatches(uArr, 2000) + tx := datasource.DefaultDB().CreateInBatches(uArr, 1000) if err := tx.Error; err != nil { logger.Errorf("CreateInBatches err => %v", err) }