lsm/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2025-11-08 20:14:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nil checking

Browse Source
This commit is contained in:
chrislu
2025-10-13 21:07:48 -07:00
parent fba4fc3a7d
commit f15eaaf8b9
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
weed/mq/kafka/integration/broker_client_subscribe.go
View File

@@ -442,6 +442,15 @@ func (bc *BrokerClient) ReadRecords(ctx context.Context, session *BrokerSubscrib
// Try to receive first record // Try to receive first record
go func() { go func() {
// Check if stream is nil (can happen during session recreation race condition)
if session.Stream == nil {
select {
case recvChan <- recvResult{resp: nil, err: fmt.Errorf("stream is nil")}:
case <-ctx.Done():
// Context cancelled, don't send (avoid blocking)
}
return
}
resp, err := session.Stream.Recv() resp, err := session.Stream.Recv()
select { select {
case recvChan <- recvResult{resp: resp, err: err}: case recvChan <- recvResult{resp: resp, err: err}:
@@ -501,6 +510,15 @@ func (bc *BrokerClient) ReadRecords(ctx context.Context, session *BrokerSubscrib
recvChan2 := make(chan recvResult, 1) recvChan2 := make(chan recvResult, 1)
go func() { go func() {
// Check if stream is nil (can happen during session recreation race condition)
if session.Stream == nil {
select {
case recvChan2 <- recvResult{resp: nil, err: fmt.Errorf("stream is nil")}:
case <-ctx2.Done():
// Context cancelled
}
return
}
resp, err := session.Stream.Recv() resp, err := session.Stream.Recv()
select { select {
case recvChan2 <- recvResult{resp: resp, err: err}: case recvChan2 <- recvResult{resp: resp, err: err}: