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 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) { 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 logger := *c.logger
for { for {
if stream == nil { if stream == nil {
@ -135,9 +136,7 @@ func (c *GRPCClient) startEventStream() {
continue continue
} }
c.setStream(stream)
c.handleEventStream(stream) c.handleEventStream(stream)
c.setStream(nil)
// Wait before retrying // Wait before retrying
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)