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

fix: reload after batch add or del

Browse Source
This commit is contained in:
zhangsz
2025-04-15 19:59:23 +08:00
parent e1edf203c9
commit 82562de598
5 changed files with 31 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
features/ue/ims_user/repository.go
View File

@@ -42,10 +42,10 @@ type Repository struct {
}
// convertResultRows 将结果记录转实体结果组
func (r *Repository) convertResultRows(rows []map[string]any) []VoLTEUser {
arr := make([]VoLTEUser, 0)
func (r *Repository) convertResultRows(rows []map[string]any) []IMSUser {
arr := make([]IMSUser, 0)
for _, row := range rows {
item := VoLTEUser{}
item := IMSUser{}
for key, value := range row {
if keyMapper, ok := r.resultMap[key]; ok {
repo.SetFieldValue(&item, keyMapper, value)
@@ -57,7 +57,7 @@ func (r *Repository) convertResultRows(rows []map[string]any) []VoLTEUser {
}
// ClearAndInsert 清空ne_id后新增实体
func (r *Repository) ClearAndInsert(neId string, u []VoLTEUser) int64 {
func (r *Repository) ClearAndInsert(neId string, u []IMSUser) int64 {
// 不指定neID时，用 TRUNCATE 清空表快
// _, err := datasource.ExecDB("", "TRUNCATE TABLE u_ims_user", nil)
_, err := datasource.ExecDB("", "DELETE FROM u_ims_user WHERE ne_id = ?", []any{neId})
@@ -128,7 +128,7 @@ func (r *Repository) SelectPage(query map[string]any) map[string]any {
result := map[string]any{
"total": 0,
"rows": []VoLTEUser{},
"rows": []IMSUser{},
}
// 查询数量 长度为0直接返回
@@ -179,7 +179,7 @@ func (r *Repository) SelectPage(query map[string]any) map[string]any {
}
// SelectList 根据实体查询
func (r *Repository) SelectList(u VoLTEUser) []VoLTEUser {
func (r *Repository) SelectList(u IMSUser) []IMSUser {
// 查询条件拼接
var conditions []string
var params []any
@@ -210,23 +210,23 @@ func (r *Repository) SelectList(u VoLTEUser) []VoLTEUser {
}
// SelectByIMSIAndNeID 通过imsi和ne_id查询
func (r *Repository) SelectByIMSIAndNeID(imsi, neId string) VoLTEUser {
func (r *Repository) SelectByIMSIAndNeID(imsi, neId string) IMSUser {
querySql := r.selectSql + " where imsi = ? and ne_id = ?"
results, err := datasource.RawDB("", querySql, []any{imsi, neId})
if err != nil {
logger.Errorf("query err => %v", err)
return VoLTEUser{}
return IMSUser{}
}
// 转换实体
rows := r.convertResultRows(results)
if len(rows) > 0 {
return rows[0]
}
return VoLTEUser{}
return IMSUser{}
}
// Insert 批量添加
func (r *Repository) Inserts(uArr []VoLTEUser) int64 {
func (r *Repository) Inserts(uArr []IMSUser) int64 {
// multi-tenancy
r.SetTenantID(&uArr)
@@ -255,7 +255,7 @@ func (r *Repository) DeletePrefixByIMSI(imsiPrefix, neId string) int64 {
return tx.RowsAffected
}
func (r *Repository) SetTenantID(subArr *[]VoLTEUser) {
func (r *Repository) SetTenantID(subArr *[]IMSUser) {
for s := 0; s < len(*subArr); s++ {
var tenantID []string
err := dborm.DefaultDB().Table("sys_tenant").