Desempacotando dados JSON não homogêneos

Esta investigação gira em torno da desempacotamento eficiente de dados JSON, aproveitando um mecanismo que permite a desempacotamento seletivo de seções específicas ou segmentos. O objetivo é lidar com estruturas JSON onde a primeira parte serve como um "código" que determina o tipo de dados contidos na última parte.

Imagine um cenário onde você tem múltiplas estruturas de dados:

type Range struct {
    Start int
    End   int
}

type User struct {
    ID    int
    Pass  int
}

E sua mensagem JSON tem um campo "código" que sinaliza quais dados estão no campo "carga útil":

{
    "Code": 4,
    "Payload": {
        "Start": 1,
        "End": 10
    }
}
{
    "Code": 3,
    "Payload": {
        "ID": 1,
        "Pass": 1234
    }
}

Você precisa desempacotar o campo "payload" na estrutura de dados apropriada com base no campo "code".

Solução

A chave aqui é usar o tipo json.RawMessage para atrasar a desempacotamento do campo "carga útil" até que você saiba seu tipo. Por exemplo:

package main

import (
    "encoding/json"
    "fmt"
)

type Message struct {
    Code    int
    Payload json.RawMessage // Delay parsing until we know the code
}

// Unmarshall into appropriate structures based on Code
func unmarshal(m []byte) error {
    var message Message
    err := json.Unmarshal(m, &message)
    if err != nil {
        return err
    }
    switch message.Code {
    case 3:
        var user User
        if err := json.Unmarshal(message.Payload, &user); err != nil {
            return err
        }
        fmt.Printf("Unmarshalled a User: %#v\n", user)
    case 4:
        var range Range
        if err := json.Unmarshal(message.Payload, &range); err != nil {
            return err
        }
        fmt.Printf("Unmarshalled a Range: %#v\n", range)
    default:
        return fmt.Errorf("unknown code: %d", message.Code)
    }
    return nil
}

func main() {
    json1 := []byte(`{"Code": 3, "Payload": {"ID": 1, "Pass": 1234}}`)
    if err := unmarshal(json1); err != nil {
        fmt.Printf("error: %v\n", err)
    }

    json2 := []byte(`{"Code": 4, "Payload": {"Start": 1, "End": 10}}`)
    if err := unmarshal(json2); err != nil {
        fmt.Printf("error: %v\n", err)
    }
}