UDM 用户信息接口补充读取数据库

features/udm_user/service/service_redis_data.go

@@ -0,0 +1,83 @@
+package service
+
+import (
+	"fmt"
+	"strings"
+
+	"ems.agt/features/udm_user/model"
+	"ems.agt/lib/core/redis"
+)
+
+// phoneImsiList 获取所有imsi
+func 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
+}
+
+// redisUdmAuthUserList UDM鉴权用户
+func redisUdmAuthUserList() []model.UdmAuthUser {
+	phoneAndImsiArr := phoneImsiList()
+	user := []model.UdmAuthUser{}
+	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.UdmAuthUser{
+			Msisdn:    phone,
+			Imsi:      imsi,
+			Amf:       m["amf"],
+			Status:    status,
+			Ki:        m["ki"],
+			AlgoIndex: m["algo"],
+			Opc:       m["opc"],
+		}
+		user = append(user, a)
+	}
+	return user
+}
+
+// redisUdmSubUserList UDM签约用户
+func redisUdmSubUserList() []model.UdmSubUser {
+	phoneAndImsiArr := phoneImsiList()
+	user := []model.UdmSubUser{}
+	for phone, imsi := range phoneAndImsiArr {
+		key := fmt.Sprintf("udm-sd:%s", imsi)
+		m := redis.GetHash(key)
+
+		a := model.UdmSubUser{
+			Imsi:    imsi,
+			Msisdn:  m["gpsi"],
+			SubData: m["sm-dat"],
+		}
+
+		if a.Msisdn == "" {
+			a.Msisdn = phone
+		}
+
+		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)
+	}
+	return user
+}