simonzhangsz/omc_api
1
0
Fork 0
Code Activity

feat: 删除不需要文件夹

Browse Source
This commit is contained in:
TsMask
2023-10-10 10:56:44 +08:00
parent ce7c3cae68
commit d173205528
154 changed files with 32276 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
features/udm_user/service/service_redis_data.go Normal file
View File

@@ -0,0 +1,108 @@
package service
import (
"strings"
"ems.agt/features/udm_user/model"
"ems.agt/lib/core/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
}
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: key[5:],
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: 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 {
a.EpsDat = v
arr := strings.Split(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]
a.StaticIp = arr[8]
}
user = append(user, a)
}
return user
}

143
features/udm_user/service/service_udm_auth_user.go Normal file
View File

@@ -0,0 +1,143 @@
package service
import (
"fmt"
"strconv"
"ems.agt/features/udm_user/model"
"ems.agt/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 = "1"
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 = "1"
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 = "1"
if s, ok := v["imsi"]; ok {
authUser.Imsi = s
}
if s, ok := v["ki"]; ok {
authUser.Ki = s
}
if s, ok := v["amf"]; ok {
authUser.Amf = s
}
if s, ok := v["algo"]; ok {
authUser.AlgoIndex = s
}
if s, ok := v["opc"]; ok {
authUser.Opc = s
}
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 = "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]
}
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)
}

201
features/udm_user/service/service_udm_sub_user.go Normal file
View File

@@ -0,0 +1,201 @@
package service
import (
"fmt"
"strconv"
"strings"
"ems.agt/features/udm_user/model"
"ems.agt/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 subUser.StaticIp != "" {
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
}
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_data"]; ok {
subUser.SmData = s
}
if s, ok := v["eps_dat"]; ok {
subUser.EpsDat = s
}
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) == 11 {
subUser.EpsDat = v[10]
}
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)
}