From: Stephen Hemminger <sthem...@microsoft.com>

Add a helper to print rate and size in numeric or pretty format
based on JSON flag.

Signed-off-by: Stephen Hemminger <step...@networkplumber.org>
---
 tc/tc_util.c | 59 ++++++++++++++++++++++++++--------------------------
 tc/tc_util.h |  2 ++
 2 files changed, 31 insertions(+), 30 deletions(-)

diff --git a/tc/tc_util.c b/tc/tc_util.c
index 95cb49b98612..787c3fc8a57a 100644
--- a/tc/tc_util.c
+++ b/tc/tc_util.c
@@ -330,6 +330,19 @@ char *sprint_rate(__u64 rate, char *buf)
        return buf;
 }
 
+/*
+ * print rate as numeric in JSON
+ * or in human format otherwise.
+ */
+void print_rate(const char *key, const char *fmt, __u64 rate)
+{
+       SPRINT_BUF(b1);
+
+       print_u64(PRINT_JSON, key, NULL, rate);
+       print_string(PRINT_FP, NULL, fmt,
+                    sprint_rate(rate, b1));
+}
+
 int get_time(unsigned int *time, const char *str)
 {
        double t;
@@ -462,6 +475,15 @@ char *sprint_size(__u32 sz, char *buf)
        return buf;
 }
 
+void print_size(const char *key, const char *fmt, __u32 sz)
+{
+       SPRINT_BUF(b1);
+
+       print_uint(PRINT_JSON, key, NULL, sz);
+       print_string(PRINT_FP, NULL, fmt,
+                    sprint_size(sz, b1));
+}
+
 void print_qdisc_handle(char *buf, int len, __u32 h)
 {
        snprintf(buf, len, "%x:", TC_H_MAJ(h)>>16);
@@ -792,7 +814,6 @@ void print_tm(FILE *f, const struct tcf_t *tm)
 void print_tcstats2_attr(FILE *fp, struct rtattr *rta,
                         const char *prefix, struct rtattr **xstats)
 {
-       SPRINT_BUF(b1);
        struct rtattr *tbs[TCA_STATS_MAX + 1];
 
        parse_rtattr_nested(tbs, TCA_STATS_MAX, rta);
@@ -823,21 +844,15 @@ void print_tcstats2_attr(FILE *fp, struct rtattr *rta,
                       MIN(RTA_PAYLOAD(tbs[TCA_STATS_RATE_EST64]),
                           sizeof(re)));
                print_string(PRINT_FP, NULL, "\n%s", prefix);
-               print_lluint(PRINT_JSON, "rate", NULL, re.bps);
-               print_string(PRINT_FP, NULL, "rate %s",
-                            sprint_rate(re.bps, b1));
+               print_rate("rate", "rate %s", re.bps);
                print_lluint(PRINT_ANY, "pps", " %llupps", re.pps);
        } else if (tbs[TCA_STATS_RATE_EST]) {
                struct gnet_stats_rate_est re = {0};
 
                memcpy(&re, RTA_DATA(tbs[TCA_STATS_RATE_EST]),
                       MIN(RTA_PAYLOAD(tbs[TCA_STATS_RATE_EST]), sizeof(re)));
-               fprintf(fp, "\n%srate %s %upps ",
-                       prefix, sprint_rate(re.bps, b1), re.pps);
                print_string(PRINT_FP, NULL, "\n%s", prefix);
-               print_uint(PRINT_JSON, "rate", NULL, re.bps);
-               print_string(PRINT_FP, NULL, "rate %s",
-                            sprint_rate(re.bps, b1));
+               print_rate("rate", "rate %s", re.bps);
                print_uint(PRINT_ANY, "pps", " %upps", re.pps);
        }
 
@@ -847,9 +862,7 @@ void print_tcstats2_attr(FILE *fp, struct rtattr *rta,
                memcpy(&q, RTA_DATA(tbs[TCA_STATS_QUEUE]), 
MIN(RTA_PAYLOAD(tbs[TCA_STATS_QUEUE]), sizeof(q)));
                if (!tbs[TCA_STATS_RATE_EST])
                        print_string(PRINT_FP, NULL, "\n%s", prefix);
-               print_uint(PRINT_JSON, "backlog", NULL, q.backlog);
-               print_string(PRINT_FP, NULL, "backlog %s",
-                            sprint_size(q.backlog, b1));
+               print_size("backlog",  "backlog %s", q.backlog);
                print_uint(PRINT_ANY, "qlen", " %up", q.qlen);
                print_uint(PRINT_FP, NULL, " requeues %u", q.requeues);
        }
@@ -861,8 +874,6 @@ void print_tcstats2_attr(FILE *fp, struct rtattr *rta,
 void print_tcstats_attr(FILE *fp, struct rtattr *tb[],
                        const char *prefix, struct rtattr **xstats)
 {
-       SPRINT_BUF(b1);
-
        if (tb[TCA_STATS2]) {
                print_tcstats2_attr(fp, tb[TCA_STATS2], prefix, xstats);
                if (xstats && NULL == *xstats)
@@ -887,26 +898,14 @@ void print_tcstats_attr(FILE *fp, struct rtattr *tb[],
 
                        if (st.bps || st.pps) {
                                print_string(PRINT_FP, NULL, "%s", "rate ");
-                               print_uint(PRINT_JSON, "rate", NULL, st.bps);
-                               if (st.bps)
-                                       print_string(PRINT_FP, NULL, "%s ",
-                                                    sprint_rate(st.bps, b1));
-
-                               print_uint(PRINT_JSON, "pps", NULL, st.pps);
-                               if (st.pps)
-                                       print_uint(PRINT_FP, NULL, "%upps ", 
st.pps);
+                               print_rate("rate", "%s ", st.bps);
+                               print_uint(PRINT_JSON, "pps", "%upps ", st.pps);
                        }
                        if (st.qlen || st.backlog) {
                                print_string(PRINT_FP, NULL, "%s", "backlog ");
 
-                               print_uint(PRINT_JSON, "backlog", NULL, 
st.backlog);
-                               print_uint(PRINT_JSON, "qlen", NULL, st.qlen);
-                               if (st.backlog)
-                                       print_string(PRINT_FP, NULL,
-                                                    "%s ", 
sprint_size(st.backlog, b1));
-                               if (st.qlen)
-                                       print_uint(PRINT_FP, NULL,
-                                                    "%up ", st.qlen);
+                               print_size("backlog", "%s ", st.backlog);
+                               print_uint(PRINT_ANY, "qlen", "%up ", st.qlen);
                        }
                }
        }
diff --git a/tc/tc_util.h b/tc/tc_util.h
index 16babd21b473..d6bdfe035050 100644
--- a/tc/tc_util.h
+++ b/tc/tc_util.h
@@ -88,6 +88,8 @@ void print_qdisc_handle(char *buf, int len, __u32 h);
 void print_linklayer(char *buf, int len, unsigned int linklayer);
 void print_devname(enum output_type type, int ifindex);
 
+void print_rate(const char *key, const char *fmt, __u64 rate);
+void print_size(const char *key, const char *fmt, __u32 sz);
 char *sprint_rate(__u64 rate, char *buf);
 char *sprint_size(__u32 size, char *buf);
 char *sprint_qdisc_handle(__u32 h, char *buf);
-- 
2.18.0

Reply via email to