klauvsteinen/go-unifi
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Generate UnmarshalJSON to handle emptyStringInt types

Browse Source 
This commit changes the code generator to generate a `UnmarshalJSON` for each
struct, so that if unmarshalled it properly handles UniFis varying integer values
via the `emptyStringInt` type.

Structs not including a field of `int` type will still have the function generated,
but it will effectively do nothing.

Fixes #18
This commit is contained in:
Hendrik "T4cC0re" Meyer
2021-01-02 09:40:52 +01:00
committed by Paul Tyng
parent c5ff8c8593
commit 4aed7d703d
67 changed files with 2298 additions and 155 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
unifi/setting_usg.generated.go
View File

@@ -5,13 +5,15 @@ package unifi
import (
"context"
"encoding/json"
"fmt"
)
// just to fix compile issues with the import
var (
_ fmt.Formatter
_ context.Context
_ fmt.Formatter
_ json.Marshaler
)
type SettingUsg struct {
@@ -82,6 +84,57 @@ type SettingUsg struct {
UpnpWANInterface string `json:"upnp_wan_interface,omitempty"` // WAN|WAN2
}
func (dst *SettingUsg) UnmarshalJSON(b []byte) error {
type Alias SettingUsg
aux := &struct {
ArpCacheBaseReachable emptyStringInt `json:"arp_cache_base_reachable"`
DHCPRelayHopCount emptyStringInt `json:"dhcp_relay_hop_count"`
DHCPRelayMaxSize emptyStringInt `json:"dhcp_relay_max_size"`
DHCPRelayPort emptyStringInt `json:"dhcp_relay_port"`
ICMPTimeout emptyStringInt `json:"icmp_timeout"`
MssClampMss emptyStringInt `json:"mss_clamp_mss"`
OtherTimeout emptyStringInt `json:"other_timeout"`
TCPCloseTimeout emptyStringInt `json:"tcp_close_timeout"`
TCPCloseWaitTimeout emptyStringInt `json:"tcp_close_wait_timeout"`
TCPEstablishedTimeout emptyStringInt `json:"tcp_established_timeout"`
TCPFinWaitTimeout emptyStringInt `json:"tcp_fin_wait_timeout"`
TCPLastAckTimeout emptyStringInt `json:"tcp_last_ack_timeout"`
TCPSynRecvTimeout emptyStringInt `json:"tcp_syn_recv_timeout"`
TCPSynSentTimeout emptyStringInt `json:"tcp_syn_sent_timeout"`
TCPTimeWaitTimeout emptyStringInt `json:"tcp_time_wait_timeout"`
UDPOtherTimeout emptyStringInt `json:"udp_other_timeout"`
UDPStreamTimeout emptyStringInt `json:"udp_stream_timeout"`
*Alias
}{
Alias: (*Alias)(dst),
}
err := json.Unmarshal(b, &aux)
if err != nil {
return fmt.Errorf("unable to unmarshal alias: %w", err)
}
dst.ArpCacheBaseReachable = int(aux.ArpCacheBaseReachable)
dst.DHCPRelayHopCount = int(aux.DHCPRelayHopCount)
dst.DHCPRelayMaxSize = int(aux.DHCPRelayMaxSize)
dst.DHCPRelayPort = int(aux.DHCPRelayPort)
dst.ICMPTimeout = int(aux.ICMPTimeout)
dst.MssClampMss = int(aux.MssClampMss)
dst.OtherTimeout = int(aux.OtherTimeout)
dst.TCPCloseTimeout = int(aux.TCPCloseTimeout)
dst.TCPCloseWaitTimeout = int(aux.TCPCloseWaitTimeout)
dst.TCPEstablishedTimeout = int(aux.TCPEstablishedTimeout)
dst.TCPFinWaitTimeout = int(aux.TCPFinWaitTimeout)
dst.TCPLastAckTimeout = int(aux.TCPLastAckTimeout)
dst.TCPSynRecvTimeout = int(aux.TCPSynRecvTimeout)
dst.TCPSynSentTimeout = int(aux.TCPSynSentTimeout)
dst.TCPTimeWaitTimeout = int(aux.TCPTimeWaitTimeout)
dst.UDPOtherTimeout = int(aux.UDPOtherTimeout)
dst.UDPStreamTimeout = int(aux.UDPStreamTimeout)
return nil
}
func (c *Client) getSettingUsg(ctx context.Context, site string) (*SettingUsg, error) {
var respBody struct {
Meta meta `json:"meta"`