From 70783bb1b0ca6376ce49cdbd4f462f74ebe16ee2 Mon Sep 17 00:00:00 2001
From: David Newhall II <david@sleepers.pro>
Date: Tue, 18 Jun 2019 19:47:05 -0700
Subject: [PATCH] Add client DPI stats

---
 core/unifi/clients_influx.go |  6 ++++++
 core/unifi/clients_type.go   | 12 ++++++------
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/core/unifi/clients_influx.go b/core/unifi/clients_influx.go
index e59bbcf7..da58ea50 100644
--- a/core/unifi/clients_influx.go
+++ b/core/unifi/clients_influx.go
@@ -98,6 +98,12 @@ func (c Client) Points() ([]*influx.Point, error) {
 		"wired-tx_bytes":         c.WiredTxBytes,
 		"wired-tx_bytes-r":       c.WiredTxBytesR,
 		"wired-tx_packets":       c.WiredTxPackets,
+		"dpi_app":                c.DpiStats.App.Val,
+		"dpi_cat":                c.DpiStats.Cat.Val,
+		"dpi_rx_bytes":           c.DpiStats.RxBytes.Val,
+		"dpi_rx_packets":         c.DpiStats.RxPackets.Val,
+		"dpi_tx_bytes":           c.DpiStats.TxBytes.Val,
+		"dpi_tx_packets":         c.DpiStats.TxPackets.Val,
 	}
 	pt, err := influx.NewPoint("clients", tags, fields, time.Now())
 	if err != nil {
diff --git a/core/unifi/clients_type.go b/core/unifi/clients_type.go
index 662a4ca6..436b6018 100644
--- a/core/unifi/clients_type.go
+++ b/core/unifi/clients_type.go
@@ -26,12 +26,12 @@ type Client struct {
 	DevFamily     int      `json:"dev_family"`
 	DevID         int      `json:"dev_id"`
 	DpiStats      struct {
-		App       int64
-		Cat       int64
-		RxBytes   int64
-		RxPackets int64
-		TxBytes   int64
-		TxPackets int64
+		App       FlexInt
+		Cat       FlexInt
+		RxBytes   FlexInt
+		RxPackets FlexInt
+		TxBytes   FlexInt
+		TxPackets FlexInt
 	} `json:"dpi_stats"`
 	DpiStatsLastUpdated int64    `json:"dpi_stats_last_updated"`
 	Essid               string   `json:"essid"`