feat: omc原始代码
This commit is contained in:
TsMask
119
features/udm_user/service/service_redis_data.go
Normal file
119
features/udm_user/service/service_redis_data.go
Normal file
@@ -0,0 +1,119 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"nms_nbi/features/udm_user/model"
|
||||
"nms_nbi/src/framework/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
|
||||
}
|
||||
|
||||
// 跳过-号数据
|
||||
imsi := key[5:]
|
||||
if strings.Contains(imsi, "-") {
|
||||
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: imsi,
|
||||
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: m["gpsi"], // 46003550072 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 {
|
||||
arr := strings.Split(v, ",")
|
||||
// 跳过非常规数据
|
||||
if len(arr) > 9 {
|
||||
continue
|
||||
}
|
||||
a.EpsDat = 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]
|
||||
// [6] 是不要的，导入和导出不用
|
||||
a.StaticIp = arr[8]
|
||||
}
|
||||
|
||||
user = append(user, a)
|
||||
}
|
||||
return user
|
||||
}
|
||||
151
features/udm_user/service/service_udm_auth_user.go
Normal file
151
features/udm_user/service/service_udm_auth_user.go
Normal file
@@ -0,0 +1,151 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
|
||||
"nms_nbi/features/udm_user/model"
|
||||
"nms_nbi/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 = "0"
|
||||
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 = "0"
|
||||
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 = "0"
|
||||
if s, ok := v["imsi"]; ok {
|
||||
authUser.Imsi = s
|
||||
}
|
||||
if s, ok := v["ki"]; ok {
|
||||
authUser.Ki = s
|
||||
}
|
||||
if s, ok := v["algo"]; ok {
|
||||
authUser.AlgoIndex = s
|
||||
}
|
||||
if s, ok := v["amf"]; ok {
|
||||
authUser.Amf = s
|
||||
}
|
||||
if s, ok := v["opc"]; ok {
|
||||
authUser.Opc = s
|
||||
}
|
||||
|
||||
if authUser.Imsi == "" {
|
||||
continue
|
||||
}
|
||||
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 = "0"
|
||||
authUser.Imsi = v[0]
|
||||
authUser.Ki = v[1]
|
||||
authUser.AlgoIndex = v[2]
|
||||
authUser.Amf = v[3]
|
||||
if len(v) == 5 {
|
||||
authUser.Opc = v[4]
|
||||
}
|
||||
|
||||
if authUser.Imsi == "" {
|
||||
continue
|
||||
}
|
||||
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)
|
||||
}
|
||||
231
features/udm_user/service/service_udm_sub_user.go
Normal file
231
features/udm_user/service/service_udm_sub_user.go
Normal file
@@ -0,0 +1,231 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"nms_nbi/features/udm_user/model"
|
||||
"nms_nbi/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 len(subUser.StaticIp) > 5 {
|
||||
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
|
||||
}
|
||||
|
||||
subUser.EpsDat = fmt.Sprintf("%s,%s,%s,%s,%s,%s,2,%s,%s", subUser.EpsFlag, subUser.EpsOdb, subUser.HplmnOdb, subUser.Ard, subUser.Epstpl, subUser.ContextId, subUser.ApnContext, subUser.StaticIp)
|
||||
|
||||
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_dat"]; ok {
|
||||
subUser.SmData = s
|
||||
}
|
||||
if s, ok := v["eps_dat"]; ok {
|
||||
subUser.EpsDat = s
|
||||
arr := strings.Split(s, ",")
|
||||
if len(arr) == 8 {
|
||||
subUser.EpsFlag = arr[0]
|
||||
subUser.EpsOdb = arr[1]
|
||||
subUser.HplmnOdb = arr[2]
|
||||
subUser.Ard = arr[3]
|
||||
subUser.Epstpl = arr[4]
|
||||
subUser.ContextId = arr[5]
|
||||
subUser.ApnContext = arr[6]
|
||||
subUser.StaticIp = arr[7]
|
||||
}
|
||||
}
|
||||
|
||||
if subUser.Imsi == "" {
|
||||
continue
|
||||
}
|
||||
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) > 17 {
|
||||
subUser.EpsFlag = v[10]
|
||||
subUser.EpsOdb = v[11]
|
||||
subUser.HplmnOdb = v[12]
|
||||
subUser.Ard = v[13]
|
||||
subUser.Epstpl = v[14]
|
||||
subUser.ContextId = v[15]
|
||||
subUser.ApnContext = v[16]
|
||||
subUser.StaticIp = v[17]
|
||||
subUser.EpsDat = strings.Join(v[10:], ",")
|
||||
}
|
||||
|
||||
if subUser.Imsi == "" {
|
||||
continue
|
||||
}
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user