speatzle/nfsense
1
0
Fork 0
mirror of https://github.com/speatzle/nfsense.git synced 2025-05-10 18:38:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

hookup websocket to api handler

Browse source
This commit is contained in:
Samuel Lorch 2023-03-05 20:53:16 +01:00
parent 1675882b84
commit 4b246f6c51
1 changed files with 32 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
pkg/server/websocket.go
View file

@ -1,16 +1,20 @@
package server
import (
"bytes"
"context"
"fmt"
"net/http"
"runtime/debug"
"time"
"golang.org/x/exp/slog"
"nhooyr.io/websocket"
"nhooyr.io/websocket/wsjson"
)
func HandleWebsocketAPI(w http.ResponseWriter, r *http.Request) {
ctx, cancel := context.WithCancel(r.Context())
defer cancel()
c, err := websocket.Accept(w, r, nil)
if err != nil {
slog.Error("Accepting Websocket Connection", err)
@ -18,17 +22,33 @@ func HandleWebsocketAPI(w http.ResponseWriter, r *http.Request) {
}
defer c.Close(websocket.StatusInternalError, "Unexpected Closing")
ctx, cancel := context.WithTimeout(r.Context(), time.Second*10)
defer cancel()
slog.Info("Accepted API Websocket Connection")
var v interface{}
err = wsjson.Read(ctx, c, &v)
if err != nil {
slog.Error("Reading Websocket Message", err)
return
for {
_, m, err := c.Read(ctx)
if websocket.CloseStatus(err) == websocket.StatusNormalClosure {
slog.Info("API Websocket Closed Normally")
cancel()
return
} else if err != nil {
slog.Error("API Websocket Closed Unexpectedly", err)
cancel()
}
go func() {
defer func() {
if r := recover(); r != nil {
slog.Error("Recovered Panic Handling Websocket API Request", fmt.Errorf("%v", r), "stack", debug.Stack())
return
}
}()
ctx, cancel := context.WithTimeout(ctx, time.Second*10)
defer cancel()
err := apiHandler.HandleRequest(ctx, bytes.NewReader(m), w)
if err != nil {
slog.Error("Handling Websocket API Request", err)
}
}()
}
slog.Info("received websocket msg", "msg", v)
c.Close(websocket.StatusNormalClosure, "")
}