UDM 用户信息接口添加到路由

features/udm_user/api_udm_user.go

@@ -0,0 +1,75 @@
+package udmuser
+
+import (
+	"net/http"
+
+	"ems.agt/features/udm_user/service"
+	"ems.agt/lib/core/utils/ctx"
+	"ems.agt/lib/core/vo/result"
+	"ems.agt/lib/services"
+	"ems.agt/restagent/config"
+)
+
+// UDM 用户信息接口添加到路由
+func Routers() []services.RouterItem {
+	// 实例化控制层 SysDictTypeApi 结构体
+	var apis = &UdmUserApi{
+		sysDictTypeService: *service.NewServiceUdmUser,
+	}
+
+	rs := [...]services.RouterItem{
+		{
+			Method:     "GET",
+			Pattern:    "/ausfList",
+			Handler:    apis.AusfList,
+			Middleware: nil, //midware.Authorize(nil),
+		},
+		{
+			Method:     "GET",
+			Pattern:    "/sdList",
+			Handler:    apis.SdList,
+			Middleware: nil, //midware.Authorize(nil),
+		},
+		// 添加更多的 Router 对象...
+	}
+
+	// 生成两组前缀路由
+	rsPrefix := []services.RouterItem{}
+	for _, v := range rs {
+		path := "/udmUserManage/{apiVersion}" + v.Pattern
+		// 固定前缀
+		v.Pattern = config.DefaultUriPrefix + path
+		rsPrefix = append(rsPrefix, v)
+		// 可配置
+		v.Pattern = config.UriPrefix + path
+		rsPrefix = append(rsPrefix, v)
+	}
+	return rsPrefix
+}
+
+// UDM 用户
+//
+// PATH /udmUserManage
+type UdmUserApi struct {
+	sysDictTypeService service.ServiceUdmUser
+}
+
+// UDM鉴权用户
+//
+// GET /ausfList
+func (s *UdmUserApi) AusfList(w http.ResponseWriter, r *http.Request) {
+	// querys := ctx.QueryMap(r)
+
+	data := s.sysDictTypeService.ReadAusfList()
+	ctx.JSON(w, 200, result.OkData(data))
+}
+
+// UDM签约用户
+//
+// GET /sdList
+func (s *UdmUserApi) SdList(w http.ResponseWriter, r *http.Request) {
+	// querys := ctx.QueryMap(r)
+
+	data := s.sysDictTypeService.ReadSdList()
+	ctx.JSON(w, 200, result.OkData(data))
+}

features/udm_user/model/udm_ausf_user.go

@@ -0,0 +1,12 @@
+package model
+
+// UdmAusfUser UDM鉴权用户
+type UdmAusfUser struct {
+	Phone     string `json:"phone"`
+	Imsi      string `json:"imssi"`
+	Amf       string `json:"amf" binding:"required"`
+	Status    string `json:"status" binding:"required"`
+	Ki        string `json:"ki"`
+	AlgoIndex string `json:"algo_index"`
+	Opc       string `json:"opc"`
+}

features/udm_user/model/udm_sd_user.go

@@ -0,0 +1,16 @@
+package model
+
+// UdmSdUser UDM签约用户
+type UdmSdUser struct {
+	Phone                   string `json:"phone"`
+	Imsi                    string `json:"imssi"`
+	Msisdn                  string `json:"msisdn"`
+	SubUeAMBRTemp           string `json:"subUeAMBRTemp" binding:"required"`
+	SubSNSSAITemp           string `json:"subSNSSAITemp" binding:"required"`
+	Rai                     string `json:"rai"`
+	ForbiddenAreasTemp      string `json:"forbiddenAreasTemp"`
+	ServiceAreaRestrictTemp string `json:"serviceAreaRestrictTemp"`
+	CnType                  string `json:"cnType"`
+	SubData                 string `json:"subData"`
+	Eps                     string `json:"eps"`
+}

features/udm_user/repo/repo_udm_user.go

@@ -0,0 +1,129 @@
+package repo
+
+import (
+	"strings"
+
+	"ems.agt/lib/core/datasource"
+	"ems.agt/lib/core/utils/date"
+	"ems.agt/lib/core/utils/parse"
+	"ems.agt/lib/log"
+)
+
+// 实例化数据层 RepoUdmUser 结构体
+var NewRepoUdmUser = &RepoUdmUser{
+	selectSql: `select 
+	dict_id, dict_name, dict_type, status, create_by, create_time, remark 
+	from sys_dict_type`,
+
+	resultMap: map[string]string{
+		"dict_id":     "DictID",
+		"dict_name":   "DictName",
+		"dict_type":   "DictType",
+		"remark":      "Remark",
+		"status":      "Status",
+		"create_by":   "CreateBy",
+		"create_time": "CreateTime",
+		"update_by":   "UpdateBy",
+		"update_time": "UpdateTime",
+	},
+}
+
+// RepoUdmUser 字典类型表 数据层处理
+type RepoUdmUser struct {
+	// 查询视图对象SQL
+	selectSql string
+	// 结果字段与实体映射
+	resultMap map[string]string
+}
+
+// convertResultRows 将结果记录转实体结果组
+func (r *RepoUdmUser) convertResultRows(rows []map[string]any) []any {
+	arr := make([]any, 0)
+	for _, row := range rows {
+		UdmUser := []any{}
+		for key, value := range row {
+			if keyMapper, ok := r.resultMap[key]; ok {
+				datasource.SetFieldValue(&UdmUser, keyMapper, value)
+			}
+		}
+		arr = append(arr, UdmUser)
+	}
+	return arr
+}
+
+// SelectDictTypePage 根据条件分页查询字典类型
+func (r *RepoUdmUser) SelectDictTypePage(query map[string]any) map[string]any {
+	// 查询条件拼接
+	var conditions []string
+	var params []any
+	if v, ok := query["dictName"]; ok && v != "" {
+		conditions = append(conditions, "dict_name like concat(?, '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["dictType"]; ok && v != "" {
+		conditions = append(conditions, "dict_type like concat(?, '%')")
+		params = append(params, v)
+	}
+	if v, ok := query["status"]; ok && v != "" {
+		conditions = append(conditions, "status = ?")
+		params = append(params, v)
+	}
+	beginTime, ok := query["beginTime"]
+	if !ok {
+		beginTime, ok = query["params[beginTime]"]
+	}
+	if ok && beginTime != "" {
+		conditions = append(conditions, "create_time >= ?")
+		beginDate := date.ParseStrToDate(beginTime.(string), date.YYYY_MM_DD)
+		params = append(params, beginDate.UnixMilli())
+	}
+	endTime, ok := query["endTime"]
+	if !ok {
+		endTime, ok = query["params[endTime]"]
+	}
+	if ok && endTime != "" {
+		conditions = append(conditions, "create_time <= ?")
+		endDate := date.ParseStrToDate(endTime.(string), date.YYYY_MM_DD)
+		params = append(params, endDate.UnixMilli())
+	}
+
+	// 构建查询条件语句
+	whereSql := ""
+	if len(conditions) > 0 {
+		whereSql += " where " + strings.Join(conditions, " and ")
+	}
+
+	// 查询数量 长度为0直接返回
+	totalSql := "select count(1) as 'total' from sys_dict_type"
+	totalRows, err := datasource.RawDB("", totalSql+whereSql, params)
+	if err != nil {
+		log.Errorf("total err => %v", err)
+	}
+	total := parse.Number(totalRows[0]["total"])
+	if total == 0 {
+		return map[string]any{
+			"total": total,
+			"rows":  nil,
+		}
+	}
+
+	// 分页
+	pageNum, pageSize := datasource.PageNumSize(query["pageNum"], query["pageSize"])
+	pageSql := " limit ?,? "
+	params = append(params, pageNum*pageSize)
+	params = append(params, pageSize)
+
+	// 查询数据
+	querySql := r.selectSql + whereSql + pageSql
+	results, err := datasource.RawDB("", querySql, params)
+	if err != nil {
+		log.Errorf("query err => %v", err)
+	}
+
+	// 转换实体
+	rows := r.convertResultRows(results)
+	return map[string]any{
+		"total": total,
+		"rows":  rows,
+	}
+}

features/udm_user/service/service_udm_user.go

@@ -0,0 +1,90 @@
+package service
+
+import (
+	"fmt"
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/lib/core/redis"
+)
+
+// 实例化服务层 ServiceUdmUser 结构体
+var NewServiceUdmUser = &ServiceUdmUser{}
+
+// ServiceUdmUser UDM 服务层处理
+type ServiceUdmUser struct{}
+
+// phoneImsiList 获取所有imsi
+func (r *ServiceUdmUser) phoneImsiList() map[string]string {
+	phoneKeys := redis.GetKeys("1*********")
+	phoneAndImsiArr := make(map[string]string, 0)
+	for _, phone := range phoneKeys {
+		imsi := redis.Get(phone)
+		phoneAndImsiArr[phone] = imsi
+	}
+	return phoneAndImsiArr
+}
+
+// ReadAusfList UDM鉴权用户
+func (r *ServiceUdmUser) ReadAusfList() []model.UdmAusfUser {
+	phoneAndImsiArr := r.phoneImsiList()
+	user := []model.UdmAusfUser{}
+	for phone, imsi := range phoneAndImsiArr {
+		key := fmt.Sprintf("ausf:%s", imsi)
+		m := redis.GetHash(key)
+
+		status := "0"
+		if _, ok := m["auth_success"]; ok {
+			status = "1"
+		}
+		a := model.UdmAusfUser{
+			Phone:     phone,
+			Imsi:      imsi,
+			Amf:       m["amf"],
+			Status:    status,
+			Ki:        m["ki"],
+			AlgoIndex: m["algo"],
+			Opc:       m["opc"],
+		}
+		user = append(user, a)
+	}
+
+	fmt.Println(len(user))
+	return user
+}
+
+// ReadSdList UDM签约用户
+func (r *ServiceUdmUser) ReadSdList() []model.UdmSdUser {
+	phoneAndImsiArr := r.phoneImsiList()
+	user := []model.UdmSdUser{}
+	for phone, imsi := range phoneAndImsiArr {
+		key := fmt.Sprintf("udm-sd:%s", imsi)
+		m := redis.GetHash(key)
+
+		a := model.UdmSdUser{
+			Phone:   phone,
+			Imsi:    imsi,
+			Msisdn:  m["gpsi"],
+			SubData: m["sm-dat"],
+		}
+
+		if v, ok := m["am-dat"]; ok {
+			arr := strings.Split(v, ",")
+			a.SubUeAMBRTemp = arr[0]
+			a.SubSNSSAITemp = arr[1]
+			a.Rai = arr[2]
+			a.ForbiddenAreasTemp = arr[3]
+			a.ServiceAreaRestrictTemp = arr[4]
+			a.CnType = arr[5]
+		}
+		if v, ok := m["eps-dat"]; ok {
+			arr := strings.Split(v, ",")
+			a.Eps = arr[0]
+		}
+
+		user = append(user, a)
+	}
+
+	fmt.Println(len(user))
+	return user
+}

lib/routes/routes.go

@@ -25,6 +25,7 @@ import (
 	sysrole "ems.agt/features/sys_role"
 	sysuser "ems.agt/features/sys_user"
 	"ems.agt/features/trace"
+	udmuser "ems.agt/features/udm_user"
 	"ems.agt/lib/midware"
 	"ems.agt/lib/services"
 
@@ -327,6 +328,11 @@ func init() {
 	for _, v := range sysuser.Routers() {
 		Register(v.Method, v.Pattern, v.Handler, v.Middleware)
 	}
+
+	// UDM 用户信息接口添加到路由
+	for _, v := range udmuser.Routers() {
+		Register(v.Method, v.Pattern, v.Handler, v.Middleware)
+	}
 }
 
 // To resolv rest POST/PUT/DELETE/PATCH cross domain