Fix unmarshaling of dhcpd_leasetime

Issue #2
paultyng · Feb 10, 2020 · 3fb1d0b · 3fb1d0b
1 parent 95a2437
commit 3fb1d0b
Showing 1 changed file with 19 additions and 6 deletions.
diff --git a/unifi/network.go b/unifi/network.go
@@ -5,26 +5,39 @@ import (
 	"fmt"
 )
 
-func (n *Network) UnmarshalJSON(b []byte) error {
+func (dst *Network) UnmarshalJSON(b []byte) error {
 	type Alias Network
 	aux := &struct {
-		VLAN json.Number `json:"vlan"`
+		VLAN           json.Number `json:"vlan"`
+		DHCPDLeaseTime json.Number `json:"dhcpd_leasetime"`
+
 		*Alias
 	}{
-		Alias: (*Alias)(n),
+		Alias: (*Alias)(dst),
 	}
 	err := json.Unmarshal(b, &aux)
 	if err != nil {
 		return err
 	}
-	n.VLAN = 0
+
+	dst.VLAN = 0
 	if aux.VLAN.String() != "" {
-		vlan, err := aux.VLAN.Int64()
+		n, err := aux.VLAN.Int64()
+		if err != nil {
+			return err
+		}
+		dst.VLAN = int(n)
+	}
+
+	dst.DHCPDLeaseTime = 0
+	if aux.DHCPDLeaseTime.String() != "" {
+		n, err := aux.DHCPDLeaseTime.Int64()
 		if err != nil {
 			return err
 		}
-		n.VLAN = int(vlan)
+		dst.DHCPDLeaseTime = int(n)
 	}
+
 	return nil
 }