Golang gRPC流式请求异常处理

for {
    select {
    case <-ctx.Done():
        log.Println("stream context canceled:", ctx.Err())
        return ctx.Err()
    default:
        req, err := stream.Recv()
        if err != nil {
            // 进入统一错误处理
            break
        }
        // 处理请求
    }
}

req, err := stream.Recv()
if err != nil {
    if statusErr, ok := status.FromError(err); ok {
        switch statusErr.Code() {
        case codes.Canceled:
            log.Println("client canceled the stream")
        case codes.DeadlineExceeded:
            log.Println("stream deadline exceeded")
        default:
            log.Printf("stream error: %v", statusErr.Message())
        }
    } else {
        log.Printf("network or serialization error: %v", err)
    }
    return err
}

for item := range dataChan {
    if err := stream.Send(item); err != nil {
        log.Printf("failed to send item: %v", err)
        return err // 结束当前流处理
    }
}

defer func() {
    // 清理逻辑
    cancel() // 如果有 context.WithCancel
    close(someChannel)
}()