调整UDM用户函数导出CSV文件

features/udm_user/api_udm_user.go

@@ -4,7 +4,9 @@ import (
 	"fmt"
 	"net/http"
 
+	"ems.agt/features/udm_user/model"
 	"ems.agt/features/udm_user/service"
+	"ems.agt/lib/core/file"
 	mmlclient "ems.agt/lib/core/mml_client"
 	"ems.agt/lib/core/utils/ctx"
 	"ems.agt/lib/core/vo/result"
@@ -42,6 +44,12 @@ func Routers() []services.RouterItem {
 			Handler:    apis.UdmAuthUserInfo,
 			Middleware: nil, //midware.Authorize(nil),
 		},
+		{
+			Method:     "POST",
+			Pattern:    "/auth/export",
+			Handler:    apis.UdmAuthUserExport,
+			Middleware: nil, //midware.Authorize(nil),
+		},
 		// UDM签约用户
 		{
 			Method:     "GET",
@@ -101,7 +109,7 @@ type UdmUserApi struct {
 // GET /auths
 func (s *UdmUserApi) UdmAuthUserList(w http.ResponseWriter, r *http.Request) {
 	querys := ctx.QueryMap(r)
-	data := s.authUser.AuthUserList(querys)
+	data := s.authUser.Page(querys)
 	ctx.JSON(w, 200, result.Ok(data))
 }
 
@@ -109,7 +117,7 @@ func (s *UdmUserApi) UdmAuthUserList(w http.ResponseWriter, r *http.Request) {
 //
 // POST /auth/getSave
 func (s *UdmUserApi) UdmAuthUserSave(w http.ResponseWriter, r *http.Request) {
-	data := s.authUser.AuthUserSave("")
+	data := s.authUser.Save("")
 	ctx.JSON(w, 200, result.OkData(data))
 }
 
@@ -141,12 +149,35 @@ func (s *UdmUserApi) UdmAuthUserInfo(w http.ResponseWriter, r *http.Request) {
 	ctx.JSON(w, 200, result.OkData(data))
 }
 
+// UDM鉴权用户-导出
+//
+// POST /auth/export
+func (s *UdmUserApi) UdmAuthUserExport(w http.ResponseWriter, r *http.Request) {
+	list := s.authUser.List(model.UdmAuthUser{})
+	// 文件名
+	fileName := "OMC_AUTH_100.csv"
+	filePath := "C:/AMP/Probject/ems_backend/restagent/OMC_AUTH_100.csv"
+	// 转换数据
+	data := [][]string{}
+	data = append(data, []string{"ID", "Msisdn", "Imsi", "Amf", "Status", "Ki", "AlgoIndex", "Opc"})
+	for _, v := range list {
+		data = append(data, []string{v.ID, v.Msisdn, v.Imsi, v.Amf, v.Status, v.Ki, v.AlgoIndex, v.Opc})
+	}
+	// 输出到文件
+	err := file.WriterCSVFile(data, filePath)
+	if err != nil {
+		ctx.JSON(w, 200, result.ErrMsg(err.Error()))
+		return
+	}
+	ctx.FileAttachment(w, r, filePath, fileName)
+}
+
 // UDM签约用户
 //
 // GET /subs
 func (s *UdmUserApi) UdmSubUserList(w http.ResponseWriter, r *http.Request) {
 	querys := ctx.QueryMap(r)
-	data := s.subUser.SubUserList(querys)
+	data := s.subUser.Page(querys)
 	ctx.JSON(w, 200, result.Ok(data))
 }
 
@@ -154,7 +185,7 @@ func (s *UdmUserApi) UdmSubUserList(w http.ResponseWriter, r *http.Request) {
 //
 // POST /sub/getSave
 func (s *UdmUserApi) UdmSubUserSave(w http.ResponseWriter, r *http.Request) {
-	data := s.subUser.SubUserSave("")
+	data := s.subUser.Save("")
 	ctx.JSON(w, 200, result.OkData(data))
 }