golang json 反序列化多态记录

最新推荐文章于 2024-09-19 06:01:52 发布

gitxuzan_

最新推荐文章于 2024-09-19 06:01:52 发布

阅读量134

点赞数

文章标签： golang json 开发语言

本文链接：https://blog.csdn.net/qq_36517296/article/details/134142461

版权

func Test_polymorphic(t *testing.T) {
	body := `{
  "vehicles": [
    {
      "type": "car",
      "make": "BMW",
      "model": "M3",
      "seatingCapacity": 4,
      "topSpeed": 250
    },
    {
      "type": "truck",
      "make": "Volvo",
      "model": "FH",
      "payloadCapacity": 40000
    },
    {
      "type": "bike",
      "make": "Yamaha",
      "model": "YZF-R1",
      "topSpeed": 293
    }
  ]
}`
	var f *Fleet
	err := json.Unmarshal([]byte(body), &f)
	if err != nil {
		t.Fatal(err)
	}
	for _, vehicle := range f.Vehicles {
		switch v := vehicle.(type) {
		case *Car:
			fmt.Printf("%s %s has a seating capacity of %d and a top speed of %.2f km/h\n",
				v.Make,
				v.Model,
				v.SeatingCapacity,
				v.TopSpeed)
		case *Truck:
			fmt.Printf("%s %s has a payload capacity of %.2f kg\n",
				v.Make,
				v.Model,
				v.PayloadCapacity)
		case *Bike:
			fmt.Printf("%s %s has a top speed of %.2f km/h\n",
				v.Make,
				v.Model,
				v.TopSpeed)
		}
	}
}

type Fleet struct {
	Vehicles    []interface{}     `json:"-"`
	RawVehicles []json.RawMessage `json:"vehicles"`
}
type Vehicle struct {
	Type  string `json:"type"`
	Make  string `json:"make"`
	Model string `json:"model"`
}
type Car struct {
	SeatingCapacity int     `json:"seatingCapacity"`
	TopSpeed        float64 `json:"topSpeed"`
	Vehicle
}
type Truck struct {
	PayloadCapacity float64 `json:"payloadCapacity"`
	Vehicle
}
type Bike struct {
	TopSpeed float64 `json:"topSpeed"`
	Vehicle
}

func (f *Fleet) UnmarshalJSON(b []byte) error {
	type fleet Fleet
	err := json.Unmarshal(b, (*fleet)(f))
	if err != nil {
		return err
	}

	for _, raw := range f.RawVehicles {
		var v Vehicle
		err = json.Unmarshal(raw, &v)
		if err != nil {
			return err
		}
		var i interface{}
		switch v.Type {
		case "car":
			i = &Car{}
		case "truck":
			i = &Truck{}
		case "bike":
			i = &Bike{}
		default:
			return errors.New("unknown vehicle type")
		}
		err = json.Unmarshal(raw, i)
		if err != nil {
			return err
		}
		f.Vehicles = append(f.Vehicles, i)
	}
	return nil
}
func (f *Fleet) MarshalJSON() ([]byte, error) {

	type fleet Fleet
	if f.Vehicles != nil {
		for _, v := range f.Vehicles {
			b, err := json.Marshal(v)
			if err != nil {
				return nil, err
			}
			f.RawVehicles = append(f.RawVehicles, b)
		}
	}
	return json.Marshal((*fleet)(f))
}