We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0d419af commit 2c5d78dCopy full SHA for 2c5d78d
1 file changed
crates/core/src/client/messages.rs
@@ -297,10 +297,19 @@ fn num_rows_in(rows: &SubscriptionRows) -> usize {
297
}
298
299
300
+fn subscription_data_rows(rows: &SubscriptionData) -> usize {
301
+ match &rows.data {
302
+ FormatSwitch::Bsatn(x) => x.num_rows(),
303
+ FormatSwitch::Json(x) => x.num_rows(),
304
+ }
305
+}
306
+
307
impl SubscriptionMessage {
308
fn num_rows(&self) -> usize {
309
match &self.result {
310
SubscriptionResult::Subscribe(x) => num_rows_in(x),
311
+ SubscriptionResult::SubscribeMulti(x) => subscription_data_rows(x),
312
+ SubscriptionResult::UnsubscribeMulti(x) => subscription_data_rows(x),
313
SubscriptionResult::Unsubscribe(x) => num_rows_in(x),
314
_ => 0,
315
0 commit comments