refactor: migrate to fiber http driver

app/http/controllers/ssh_controller.go

@@ -3,14 +3,15 @@ package controllers
 import (
 	"bytes"
 	"context"
-	nethttp "net/http"
 	"sync"
 	"time"
 
+	"github.com/fasthttp/websocket"
+	"github.com/goravel/fiber"
 	"github.com/goravel/framework/contracts/http"
 	"github.com/goravel/framework/facades"
-	"github.com/gorilla/websocket"
 	"github.com/spf13/cast"
+	"github.com/valyala/fasthttp"
 
 	"panel/app/models"
 	"panel/app/services"
@@ -92,71 +93,72 @@ func (r *SshController) UpdateInfo(ctx http.Context) http.Response {
 
 // Session SSH 会话
 func (r *SshController) Session(ctx http.Context) http.Response {
-	upGrader := websocket.Upgrader{
+	upGrader := websocket.FastHTTPUpgrader{
 		ReadBufferSize:  4096,
 		WriteBufferSize: 4096,
-		CheckOrigin: func(r *nethttp.Request) bool {
+		CheckOrigin: func(ctx *fasthttp.RequestCtx) bool {
 			return true
 		},
 		Subprotocols: []string{ctx.Request().Header("Sec-WebSocket-Protocol")},
 	}
 
-	ws, err := upGrader.Upgrade(ctx.Response().Writer(), ctx.Request().Origin(), nil)
-	if err != nil {
-		facades.Log().Error("[面板][SSH] 建立连接失败 ", err)
-		return ErrorSystem(ctx)
-	}
-	defer ws.Close()
-
 	config := ssh.SSHClientConfigPassword(
 		r.setting.Get(models.SettingKeySshHost)+":"+r.setting.Get(models.SettingKeySshPort),
 		r.setting.Get(models.SettingKeySshUser),
 		r.setting.Get(models.SettingKeySshPassword),
 	)
 	client, err := ssh.NewSSHClient(config)
+
+	err = upGrader.Upgrade(ctx.(*fiber.Context).Instance().Context(), func(conn *websocket.Conn) {
+		if err != nil {
+			_ = conn.WriteControl(websocket.CloseMessage,
+				[]byte(err.Error()), time.Now().Add(time.Second))
+			return
+		}
+
+		defer client.Close()
+
+		turn, err := ssh.NewTurn(conn, client)
+		if err != nil {
+			_ = conn.WriteControl(websocket.CloseMessage,
+				[]byte(err.Error()), time.Now().Add(time.Second))
+			return
+		}
+		defer turn.Close()
+
+		var bufPool = sync.Pool{
+			New: func() any {
+				return new(bytes.Buffer)
+			},
+		}
+		var logBuff = bufPool.Get().(*bytes.Buffer)
+		logBuff.Reset()
+		defer bufPool.Put(logBuff)
+
+		ctx2, cancel := context.WithCancel(context.Background())
+		wg := sync.WaitGroup{}
+		wg.Add(2)
+		go func() {
+			defer wg.Done()
+			err := turn.LoopRead(logBuff, ctx2)
+			if err != nil {
+				facades.Log().Error("[面板][SSH] 读取数据失败 ", err.Error())
+			}
+		}()
+		go func() {
+			defer wg.Done()
+			err := turn.SessionWait()
+			if err != nil {
+				facades.Log().Error("[面板][SSH] 会话失败 ", err.Error())
+			}
+			cancel()
+		}()
+		wg.Wait()
+	})
 	if err != nil {
-		_ = ws.WriteControl(websocket.CloseMessage,
-			[]byte(err.Error()), time.Now().Add(time.Second))
+		facades.Log().Error("[面板][SSH] 建立连接失败 ", err)
 		return ErrorSystem(ctx)
 	}
-	defer client.Close()
 
-	turn, err := ssh.NewTurn(ws, client)
-	if err != nil {
-		_ = ws.WriteControl(websocket.CloseMessage,
-			[]byte(err.Error()), time.Now().Add(time.Second))
-		return ErrorSystem(ctx)
-	}
-	defer turn.Close()
-
-	var bufPool = sync.Pool{
-		New: func() interface{} {
-			return new(bytes.Buffer)
-		},
-	}
-	var logBuff = bufPool.Get().(*bytes.Buffer)
-	logBuff.Reset()
-	defer bufPool.Put(logBuff)
-
-	ctx2, cancel := context.WithCancel(context.Background())
-	wg := sync.WaitGroup{}
-	wg.Add(2)
-	go func() {
-		defer wg.Done()
-		err := turn.LoopRead(logBuff, ctx2)
-		if err != nil {
-			facades.Log().Error("[面板][SSH] 读取数据失败 ", err.Error())
-		}
-	}()
-	go func() {
-		defer wg.Done()
-		err := turn.SessionWait()
-		if err != nil {
-			facades.Log().Error("[面板][SSH] 会话失败 ", err.Error())
-		}
-		cancel()
-	}()
-	wg.Wait()
-
-	return Success(ctx, logBuff.String())
+	return nil
 }