Немаршалинг вложенных структур и утверждения типа - PullRequest
Новая
       42

Немаршалинг вложенных структур и утверждения типа

0 голосов
/ 09 июля 2020

Я хочу маршалировать и демаршалировать бинарную древовидную структуру в Go. Каждый узел соответствует структуре типа Node. Узлы связаны между собой указателями (левый и правый дочерние узлы), как в связанном списке. Листья дерева несут контент, который реализован в виде интерфейса. Все листья дерева имеют один и тот же тип Контента, который заранее известен демаршаллеру. Я знаю, что при демаршалинге структуры с интерфейсом в одном поле (скажем, «Содержимое») я должен выполнить утверждение типа, например err = json.Unmarshal(byteSlice, &decodedStruct{Content: &MyStruct{}}) Однако, поскольку дерево имеет произвольный размер, мои структуры глубоко вложены. Есть ли простой / идиоматический c способ маршалинга / демаршалинга такого объекта, о котором я не знаю?

Ниже я публикую минимальный пример, который, как мне кажется, представляет две ключевые особенности первого, последовательность указатели и, во-вторых, интерфейс в «конце». (площадка: https://play.golang.org/p/t9C9Hn4ONlE) 

// LinkedList is a simple linked list defined by a root node
type LinkedList struct {
    Name string
    Root *Node
}

// Node is a list's node with Content
type Node struct {
    Child *Node
    C     Content
}

// Content is a dummy interface
type Content interface {
    CalculateSum() int
}

// MyStruct implements Content
type MyStruct struct {
    ID     int
    Values []int
}

// CalculateSum computes the sum of the slice in the field @Values
func (ms MyStruct) CalculateSum() (s int) {
    for _, i := range ms.Values {
        s += i
    }
    return
}

func main() {
    // Make a list of three nodes with content in the leaf
    ms := MyStruct{2, []int{2, 4, 7}}
    leaf := Node{nil, ms}
    node := Node{&leaf, nil}
    rootNode := Node{&node, nil}
    ll := LinkedList{"list1", &rootNode}

    // Encoding linked list works fine...
    llEncoded, err := json.Marshal(ll)

    // ...decoding doesn't:
    // error decoding:  json: cannot unmarshal object into Go struct field Node.Root.Child.Child.C of type main.Content
    llDecoded := LinkedList{}
    err = json.Unmarshal(llEncoded, &llDecoded)
    fmt.Println("error decoding: ", err)
}

1 Ответ

0 голосов
/ 09 июля 2020

Если вы заранее знаете конкретный тип Content, вы можете реализовать интерфейс json.Unmarshaler, демаршалируясь в жестко запрограммированный конкретный тип, а затем присвоить результат типу интерфейса. 

func (n *Node) UnmarshalJSON(data []byte) error {
    var node struct {
        Child *Node
        C     *MyStruct
    }
    if err := json.Unmarshal(data, &node); err != nil {
        return err
    }
    n.Child = node.Child
    n.C = node.C
    return nil
}

https://play.golang.org/p/QOJuiLpYrze

Если вам нужно сделать его более гибким, вам нужно как-то сообщить реализации json.Unmarshaler, какой конкретный тип представляет json. Один из способов сделать это - встроить информацию о типе в json содержимого, например (теперь с помощью интерфейса json.Marshaler): 

func (ms MyStruct) MarshalJSON() ([]byte, error) {
    type _MyStruct MyStruct

    var out = struct {
        Type string `json:"_type"`
        _MyStruct
    }{
        Type:      "MyStruct",
        _MyStruct: _MyStruct(ms),
    }
    return json.Marshal(out)
}

Обновить демаршалер Node реализация соответственно: 

func (n *Node) UnmarshalJSON(data []byte) error {
    var node struct {
        Child *Node
        C     json.RawMessage
    }
    if err := json.Unmarshal(data, &node); err != nil {
        return err
    }
    n.Child = node.Child

    if len(node.C) > 0 && string(node.C) != `null` {
        var _type struct {
            Type string `json:"_type"`
        }
        if err := json.Unmarshal([]byte(node.C), &_type); err != nil {
            return err
        }

        c := newContent[_type.Type]()
        if err := json.Unmarshal([]byte(node.C), c); err != nil {
            return err
        }
        n.C = c
    }
    return nil
}

и определите newContent как карту, значениями которой являются функции, возвращающие новые экземпляры конкретного типа: 

var newContent = map[string]func() Content{
    "MyStruct": func() Content { return new(MyStruct) },
    // ...
}

Попробуйте на игровой площадке: https://play.golang.org/p/u9L0VxEG4dT

...