fix: UDM导入txt支持

This commit is contained in:
TsMask
2023-09-15 18:01:43 +08:00
parent c6f277f0e9
commit 289e892254
5 changed files with 96 additions and 15 deletions

View File

@@ -47,7 +47,7 @@ func (r *ServiceUdmAuthUser) Inserts(neID string, authUser model.UdmAuthUser, nu
return r.repoAuthUser.Inserts(neID, authUser, num)
}
// Insert UDM鉴权用户-批量添加
// InsertCSV UDM鉴权用户-批量添加
func (r *ServiceUdmAuthUser) InsertCSV(neID string, data []map[string]string) int64 {
var num int64
for _, v := range data {
@@ -74,6 +74,28 @@ func (r *ServiceUdmAuthUser) InsertCSV(neID string, data []map[string]string) in
return num
}
// InsertTxt UDM鉴权用户-批量添加
func (r *ServiceUdmAuthUser) InsertTxt(neID string, data [][]string) int64 {
var num int64
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]
}
num += r.repoAuthUser.Insert(neID, authUser)
}
return num
}
// Insert UDM鉴权用户-修改更新
func (r *ServiceUdmAuthUser) Update(neID string, authUser model.UdmAuthUser) int64 {
return r.repoAuthUser.Update(neID, authUser)