Superminaren
/
kvm
mirror of https://github.com/jetkvm/kvm.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: move setStream to handleEventStream

This commit is contained in:
Siyuan 2025-11-18 09:39:02 +00:00
parent 470fcf4d64
commit 9c9c085690
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
internal/native/grpc_client.go
View File

@ -75,13 +75,14 @@ func NewGRPCClient(opts grpcClientOptions) (*GRPCClient, error) {
return grpcClient, nil
}
func (c *GRPCClient) setStream(stream pb.NativeService_StreamEventsClient) {
c.eventM.Lock()
defer c.eventM.Unlock()
c.eventStream = stream
}
func (c *GRPCClient) handleEventStream(stream pb.NativeService_StreamEventsClient) {
c.eventM.Lock()
c.eventStream = stream
defer func() {
c.eventStream = nil
c.eventM.Unlock()
}()
logger := *c.logger
for {
if stream == nil {
@ -135,9 +136,7 @@ func (c *GRPCClient) startEventStream() {
continue
}
c.setStream(stream)
c.handleEventStream(stream)
c.setStream(nil)
// Wait before retrying
time.Sleep(1 * time.Second)