Skip to content

Instantly share code, notes, and snippets.

@lindsayzhou

lindsayzhou/attrsplitor.go

Created February 20, 2024 15:07
Download ZIP

Revisions

  1. lindsayzhou renamed this gist Feb 20, 2024. 1 changed file with 0 additions and 0 deletions.
    0 gistfile1.txt → attrsplitor.go
    View file
    File renamed without changes.
  2. lindsayzhou created this gist Feb 20, 2024.
    77 changes: 77 additions & 0 deletions gistfile1.txt
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,77 @@

    type AttrSplitor struct {
    attrMapList []map[string]interface{}
    }

    func NewSplitor(attrList ...map[string]interface{}) *AttrSplitor {
    return &AttrSplitor{attrMapList: attrList}
    }

    func (l *AttrSplitor) Split(src map[string]json.RawMessage) (result []map[string]json.RawMessage, err error) {
    for _, attrMap := range l.attrMapList {
    part := make(map[string]json.RawMessage)
    for key, obj := range attrMap {
    rawValue, ok := src[key]
    if !ok {
    continue
    }
    val := reflect.New(reflect.TypeOf(obj))
    err = json.Unmarshal(rawValue, val.Interface())
    if err != nil {
    return
    }
    part[key], err = json.Marshal(val.Interface())
    if err != nil {
    return
    }
    }
    result = append(result, part)
    }
    return
    }

    type AttachmentType string

    const (
    ATTACHMENT_TYPE_IPLINK_STATIC = "ip_link_static"
    ATTACHMENT_TYPE_IPLINK_PPPOE = "ip_link_pppoe"
    ATTACHMENT_TYPE_IPLINK_VPN = "ip_link_vpn"
    ATTACHMENT_TYPE_UNKNOWN = "unknown"
    )

    var AttachmentAttrSplitor map[AttachmentType]*AttrSplitor

    func init() {
    AttachmentAttrSplitor = map[AttachmentType]*AttrSplitor{
    ATTACHMENT_TYPE_IPLINK_STATIC: NewSplitor(
    map[string]interface{}{
    "vlan": uint32(0),
    "mac": HardwareAddr{},
    "ip_mask": IPMask{},
    "gateway": "",
    "max_tx": uint32(0),
    "max_rx": uint32(0),
    "ac_name": "",
    "server_name": "",
    },
    map[string]interface{}{
    "bridge": "",
    },
    ),
    ATTACHMENT_TYPE_IPLINK_PPPOE: NewSplitor(
    map[string]interface{}{
    "vlan": uint32(0),
    "mac": HardwareAddr{},
    "username": "",
    "password": "",
    "max_tx": uint32(0),
    "max_rx": uint32(0),
    "ac_name": "",
    "server_name": "",
    },
    map[string]interface{}{
    "bridge": "",
    },
    ),
    }
    }