diff --git a/features/udm_user/api_udm_user.go b/features/udm_user/api_udm_user.go new file mode 100644 index 00000000..af3d55ef --- /dev/null +++ b/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)) +} diff --git a/features/udm_user/model/udm_ausf_user.go b/features/udm_user/model/udm_ausf_user.go new file mode 100644 index 00000000..896272ad --- /dev/null +++ b/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"` +} diff --git a/features/udm_user/model/udm_sd_user.go b/features/udm_user/model/udm_sd_user.go new file mode 100644 index 00000000..64000218 --- /dev/null +++ b/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"` +} diff --git a/features/udm_user/repo/repo_udm_user.go b/features/udm_user/repo/repo_udm_user.go new file mode 100644 index 00000000..cedf05b0 --- /dev/null +++ b/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, + } +} diff --git a/features/udm_user/service/service_udm_user.go b/features/udm_user/service/service_udm_user.go new file mode 100644 index 00000000..734e537f --- /dev/null +++ b/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 +} diff --git a/lib/routes/routes.go b/lib/routes/routes.go index d5b959b4..decf76b5 100644 --- a/lib/routes/routes.go +++ b/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