From 95a2437dcc33858b14f265b5591f152565735300 Mon Sep 17 00:00:00 2001
From: Paul Tyng <paul@paultyng.net>
Date: Thu, 6 Feb 2020 11:11:41 -0500
Subject: [PATCH] Fix maxsta unmarshaling

---
 unifi/wlan_group.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/unifi/wlan_group.go b/unifi/wlan_group.go
index 996b190..4a9a372 100644
--- a/unifi/wlan_group.go
+++ b/unifi/wlan_group.go
@@ -1,5 +1,30 @@
 package unifi
 
+import "encoding/json"
+
+func (n *WLANGroup) UnmarshalJSON(b []byte) error {
+	type Alias WLANGroup
+	aux := &struct {
+		Maxsta json.Number `json:"maxsta"`
+		*Alias
+	}{
+		Alias: (*Alias)(n),
+	}
+	err := json.Unmarshal(b, &aux)
+	if err != nil {
+		return err
+	}
+	n.Maxsta = 0
+	if aux.Maxsta.String() != "" {
+		maxsta, err := aux.Maxsta.Int64()
+		if err != nil {
+			return err
+		}
+		n.Maxsta = int(maxsta)
+	}
+	return nil
+}
+
 func (c *Client) ListWLANGroup(site string) ([]WLANGroup, error) {
 	return c.listWLANGroup(site)
 }