feat: 工具模块ping功能

src/modules/tool/controller/ping.go

@@ -0,0 +1,158 @@
+package controller
+
+import (
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"be.ems/src/framework/i18n"
+	"be.ems/src/framework/logger"
+	"be.ems/src/framework/utils/ctx"
+	"be.ems/src/framework/vo/result"
+	neService "be.ems/src/modules/network_element/service"
+	"be.ems/src/modules/tool/model"
+	"be.ems/src/modules/tool/service"
+	wsService "be.ems/src/modules/ws/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+// 实例化控制层 PingController 结构体
+var NewPing = &PingController{
+	pingService: service.NewPing,
+	wsService:   wsService.NewWS,
+}
+
+// ping ICMP网络探测工具 https://github.com/prometheus-community/pro-bing
+//
+// PATH /tool/ping
+type PingController struct {
+	pingService *service.Ping // ping ICMP网络探测工具
+	wsService   *wsService.WS // WebSocket 服务
+}
+
+// ping 基本信息运行
+//
+// POST /
+func (s *PingController) Statistics(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var body model.Ping
+	if err := c.ShouldBindBodyWithJSON(&body); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	info, err := s.pingService.Statistics(body)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(i18n.TKey(language, err.Error())))
+		return
+	}
+	c.JSON(200, result.OkData(info))
+}
+
+// ping 传统UNIX运行
+//
+// GET /
+func (s *PingController) StatisticsOn(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	// 登录用户信息
+	loginUser, err := ctx.LoginUser(c)
+	if err != nil {
+		c.JSON(401, result.CodeMsg(401, i18n.TKey(language, err.Error())))
+		return
+	}
+
+	// 将 HTTP 连接升级为 WebSocket 连接
+	wsConn := s.wsService.UpgraderWs(c.Writer, c.Request)
+	if wsConn == nil {
+		return
+	}
+	defer wsConn.Close()
+
+	wsClient := s.wsService.ClientCreate(loginUser.UserID, nil, wsConn, nil)
+	go s.wsService.ClientWriteListen(wsClient)
+	go s.wsService.ClientReadListen(wsClient, s.pingService.StatisticsOn)
+
+	// 等待停止信号
+	for value := range wsClient.StopChan {
+		s.wsService.ClientClose(wsClient.ID)
+		logger.Infof("ws Stop Client UID %s %s", wsClient.BindUid, value)
+		return
+	}
+}
+
+// ping 网元端UNIX运行
+//
+// GET /run
+func (s *PingController) Run(c *gin.Context) {
+	language := ctx.AcceptLanguage(c)
+	var query struct {
+		NeType string `form:"neType" binding:"required"` // 网元类型
+		NeId   string `form:"neId" binding:"required"`   // 网元标识id
+		Cols   int    `form:"cols"`                      // 终端单行字符数
+		Rows   int    `form:"rows"`                      // 终端显示行数
+	}
+	if err := c.ShouldBindQuery(&query); err != nil {
+		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
+		return
+	}
+
+	// 登录用户信息
+	loginUser, err := ctx.LoginUser(c)
+	if err != nil {
+		c.JSON(401, result.CodeMsg(401, i18n.TKey(language, err.Error())))
+		return
+	}
+
+	// 网元主机的SSH客户端
+	sshClient, err := neService.NewNeInfoImpl.NeRunSSHClient(query.NeType, query.NeId)
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+	defer sshClient.Close()
+	// ssh连接会话
+	clientSession, err := sshClient.NewClientSession(query.Cols, query.Rows)
+	if err != nil {
+		c.JSON(200, result.ErrMsg("neinfo ssh client session new err"))
+		return
+	}
+	defer clientSession.Close()
+
+	// 将 HTTP 连接升级为 WebSocket 连接
+	wsConn := s.wsService.UpgraderWs(c.Writer, c.Request)
+	if wsConn == nil {
+		return
+	}
+	defer wsConn.Close()
+
+	wsClient := s.wsService.ClientCreate(loginUser.UserID, nil, wsConn, clientSession)
+	go s.wsService.ClientWriteListen(wsClient)
+	go s.wsService.ClientReadListen(wsClient, s.pingService.RunNE)
+
+	// 等待1秒，排空首次消息
+	time.Sleep(1 * time.Second)
+	_ = clientSession.Read()
+
+	// 实时读取Run消息直接输出
+	msTicker := time.NewTicker(100 * time.Millisecond)
+	defer msTicker.Stop()
+	for {
+		select {
+		case ms := <-msTicker.C:
+			outputByte := clientSession.Read()
+			if len(outputByte) > 0 {
+				outputStr := string(outputByte)
+				msgByte, _ := json.Marshal(result.Ok(map[string]any{
+					"requestId": fmt.Sprintf("ping_%d", ms.UnixMilli()),
+					"data":      outputStr,
+				}))
+				wsClient.MsgChan <- msgByte
+			}
+		case <-wsClient.StopChan: // 等待停止信号
+			s.wsService.ClientClose(wsClient.ID)
+			logger.Infof("ws Stop Client UID %s", wsClient.BindUid)
+			return
+		}
+	}
+}