From: Maciej Gajdzica <maciejx.t.gajdz...@intel.com>

Added statistics for sched writer port.

Signed-off-by: Maciej Gajdzica <maciejx.t.gajdzica at intel.com>
---
 lib/librte_port/rte_port_sched.c |   57 ++++++++++++++++++++++++++++++++++----
 1 file changed, 52 insertions(+), 5 deletions(-)

diff --git a/lib/librte_port/rte_port_sched.c b/lib/librte_port/rte_port_sched.c
index a82e4fa..c5ff8ab 100644
--- a/lib/librte_port/rte_port_sched.c
+++ b/lib/librte_port/rte_port_sched.c
@@ -132,7 +132,23 @@ rte_port_sched_reader_stats_read(void *port,
 /*
  * Writer
  */
+#ifdef RTE_PORT_STATS_COLLECT
+
+#define RTE_PORT_SCHED_WRITER_STATS_PKTS_IN_ADD(port, val) \
+       port->stats.n_pkts_in += val
+#define RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(port, val) \
+       port->stats.n_pkts_drop += val
+
+#else
+
+#define RTE_PORT_SCHED_WRITER_STATS_PKTS_IN_ADD(port, val)
+#define RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(port, val)
+
+#endif
+
 struct rte_port_sched_writer {
+       struct rte_port_out_stats stats;
+
        struct rte_mbuf *tx_buf[2 * RTE_PORT_IN_BURST_SIZE_MAX];
        struct rte_sched_port *sched;
        uint32_t tx_burst_sz;
@@ -180,8 +196,12 @@ rte_port_sched_writer_tx(void *port, struct rte_mbuf *pkt)
        struct rte_port_sched_writer *p = (struct rte_port_sched_writer *) port;

        p->tx_buf[p->tx_buf_count++] = pkt;
+       RTE_PORT_SCHED_WRITER_STATS_PKTS_IN_ADD(p, 1);
        if (p->tx_buf_count >= p->tx_burst_sz) {
-               rte_sched_port_enqueue(p->sched, p->tx_buf, p->tx_buf_count);
+               __rte_unused uint32_t nb_tx;
+
+               nb_tx = rte_sched_port_enqueue(p->sched, p->tx_buf, 
p->tx_buf_count);
+               RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(p, p->tx_buf_count - 
nb_tx);
                p->tx_buf_count = 0;
        }

@@ -200,15 +220,18 @@ rte_port_sched_writer_tx_bulk(void *port,
                        ((pkts_mask & bsz_mask) ^ bsz_mask);

        if (expr == 0) {
+               __rte_unused uint32_t nb_tx;
                uint64_t n_pkts = __builtin_popcountll(pkts_mask);

                if (tx_buf_count) {
-                       rte_sched_port_enqueue(p->sched, p->tx_buf,
+                       nb_tx = rte_sched_port_enqueue(p->sched, p->tx_buf,
                                tx_buf_count);
+                       RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(p, 
tx_buf_count - nb_tx);
                        p->tx_buf_count = 0;
                }

-               rte_sched_port_enqueue(p->sched, pkts, n_pkts);
+               nb_tx = rte_sched_port_enqueue(p->sched, pkts, n_pkts);
+               RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(p, n_pkts - nb_tx);
        } else {
                for ( ; pkts_mask; ) {
                        uint32_t pkt_index = __builtin_ctzll(pkts_mask);
@@ -216,13 +239,17 @@ rte_port_sched_writer_tx_bulk(void *port,
                        struct rte_mbuf *pkt = pkts[pkt_index];

                        p->tx_buf[tx_buf_count++] = pkt;
+                       RTE_PORT_SCHED_WRITER_STATS_PKTS_IN_ADD(p, 1);
                        pkts_mask &= ~pkt_mask;
                }
                p->tx_buf_count = tx_buf_count;

                if (tx_buf_count >= p->tx_burst_sz) {
-                       rte_sched_port_enqueue(p->sched, p->tx_buf,
+                       __rte_unused uint32_t nb_tx;
+
+                       nb_tx = rte_sched_port_enqueue(p->sched, p->tx_buf,
                                tx_buf_count);
+                       RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(p, 
tx_buf_count - nb_tx);
                        p->tx_buf_count = 0;
                }
        }
@@ -236,7 +263,10 @@ rte_port_sched_writer_flush(void *port)
        struct rte_port_sched_writer *p = (struct rte_port_sched_writer *) port;

        if (p->tx_buf_count) {
-               rte_sched_port_enqueue(p->sched, p->tx_buf, p->tx_buf_count);
+               __rte_unused uint32_t nb_tx;
+
+               nb_tx = rte_sched_port_enqueue(p->sched, p->tx_buf, 
p->tx_buf_count);
+               RTE_PORT_SCHED_WRITER_STATS_PKTS_DROP_ADD(p, p->tx_buf_count - 
nb_tx);
                p->tx_buf_count = 0;
        }

@@ -257,6 +287,22 @@ rte_port_sched_writer_free(void *port)
        return 0;
 }

+static int
+rte_port_sched_writer_stats_read(void *port,
+               struct rte_port_out_stats *stats, int clear)
+{
+       struct rte_port_sched_writer *p =
+               (struct rte_port_sched_writer *) port;
+
+       if (stats != NULL)
+               memcpy(stats, &p->stats, sizeof(p->stats));
+
+       if (clear)
+               memset(&p->stats, 0, sizeof(p->stats));
+
+       return 0;
+}
+
 /*
  * Summary of port operations
  */
@@ -273,4 +319,5 @@ struct rte_port_out_ops rte_port_sched_writer_ops = {
        .f_tx = rte_port_sched_writer_tx,
        .f_tx_bulk = rte_port_sched_writer_tx_bulk,
        .f_flush = rte_port_sched_writer_flush,
+       .f_stats = rte_port_sched_writer_stats_read,
 };
-- 
1.7.9.5

Reply via email to