Skip to content

Commit 21969ac

Browse files
committed
* MDF [mqtt_quic] count tx msg in prior_send as well
Signed-off-by: Jaylin <jaylin@emqx.io>
1 parent 6b0a34e commit 21969ac

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

src/mqtt/protocol/mqtt/mqtt_quic_client.c

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1027,11 +1027,20 @@ static void mqtt_quic_sock_init(void *arg, nni_sock *sock)
10271027
.si_atomic = true,
10281028
};
10291029
nni_stat_init(&s->msg_bytes_cached, &msg_bytes_cached);
1030+
static const nni_stat_info msg_sqlite_cached = {
1031+
.si_name = "mqtt_msg_sqlite_cached",
1032+
.si_desc = "count all cached msg in sqlite",
1033+
.si_type = NNG_STAT_COUNTER,
1034+
.si_unit = NNG_UNIT_BYTES,
1035+
.si_atomic = true,
1036+
};
1037+
nni_stat_init(&s->msg_sqlite_cached, &msg_sqlite_cached);
10301038
nni_sock_add_stat(s->nsock, &s->mqtt_reconnect);
10311039
nni_sock_add_stat(s->nsock, &s->msg_resend);
10321040
nni_sock_add_stat(s->nsock, &s->msg_send_drop);
10331041
nni_sock_add_stat(s->nsock, &s->msg_recv_drop);
10341042
nni_sock_add_stat(s->nsock, &s->msg_bytes_cached);
1043+
nni_sock_add_stat(s->nsock, &s->msg_sqlite_cached);
10351044
#endif
10361045
}
10371046

src/mqtt/transport/quic/mqtt_quic.c

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1162,6 +1162,10 @@ mqtt_quictran_pipe_send_prior(mqtt_quictran_pipe *p, nni_aio *aio)
11621162
// *flag |= p->substreams[nni_random()%2].id;
11631163
// nni_aio_set_prov_data(aio, flag);
11641164
nng_stream_send(p->conn, aio);
1165+
#ifdef NNG_ENABLE_STATS
1166+
// assume it is already sent, since we cannot access user cb.
1167+
nni_sock_bump_tx(p->ep->nsock, nni_msg_len(msg));
1168+
#endif
11651169
}
11661170

11671171
static void

0 commit comments

Comments
 (0)