github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-07-06 21:09:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update socket_data_streamer.go and socket_server_handlers.go

Browse Source
This commit is contained in:
Rami Berman 2022-04-03 14:51:49 +03:00
parent b568f42c59
commit 08212ff752
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
agent/pkg/api/socket_data_streamer.go
View File

@ -14,7 +14,7 @@ import (
)
type EntryStreamer interface {
Get(socketId int, params *WebSocketParams) (context.CancelFunc, error)
Get(ctx context.Context, socketId int, params *WebSocketParams) error
}
type EntryStreamerSocketConnector interface {
@ -66,7 +66,7 @@ func (e *DefaultEntryStreamerSocketConnector) CleanupSocket(socketId int) {
type BasenineEntryStreamer struct{}
func (e *BasenineEntryStreamer) Get(socketId int, params *WebSocketParams) (context.CancelFunc, error) {
func (e *BasenineEntryStreamer) Get(ctx context.Context, socketId int, params *WebSocketParams) error {
var connection *basenine.Connection
entryStreamerSocketConnector := dependency.GetInstance(dependency.EntryStreamerSocketConnector).(EntryStreamerSocketConnector)
@ -75,14 +75,12 @@ func (e *BasenineEntryStreamer) Get(socketId int, params *WebSocketParams) (cont
if err != nil {
logger.Log.Errorf("failed to establish a connection to Basenine: %v", err)
entryStreamerSocketConnector.CleanupSocket(socketId)
return nil, err
return err
}
data := make(chan []byte)
meta := make(chan []byte)
ctx, cancel := context.WithCancel(context.Background())
query := params.Query
err = basenine.Validate(shared.BasenineHost, shared.BaseninePort, query)
if err != nil {
@ -139,5 +137,5 @@ func (e *BasenineEntryStreamer) Get(socketId int, params *WebSocketParams) (cont
connection.Close()
}()
return cancel, nil
return nil
}

4
agent/pkg/api/socket_server_handlers.go
View File

@ -109,10 +109,12 @@ func (h *RoutesEventHandlers) WebSocketMessage(socketId int, isTapper bool, mess
}
entriesStreamer := dependency.GetInstance(dependency.EntriesSocketStreamer).(EntryStreamer)
cancelFunc, err := entriesStreamer.Get(socketId, &params)
ctx, cancelFunc := context.WithCancel(context.Background())
err := entriesStreamer.Get(ctx, socketId, &params)
if err != nil {
logger.Log.Errorf("error initializing basenine stream for browser socket %d %+v", socketId, err)
cancelFunc()
} else {
browserClients[socketId].dataStreamCancelFunc = cancelFunc
}