golang动态实例化解析json

在做mock接口的时候，有几类数据，数据的类型不一样，每类数据是一个json的数组以文件形式保存，使用的时候最好能够根据需求把不同类型的数据混合在一起返回。为了尽量少写代码想着跟python之类的语言一样，使用字符串动态实例化。
例如，有点播数据、直播数据，结构体如下：

// vod 点播数据
type vod struct {
    ID string `json:"id"`
    Title string `json:"title"`
    CommentID string `json:"commentId"`
}
// live 直播数据
type live struct {
    ID string `json:"id"`
    Title string `json:"title"`
    ChatID string `json:"chatId"`
}

动态实例化需要先创建相应的实例的map，然后通过反射获取相应字符串对应结构的类型，然后在实例化对象

var regStruct = make(map[string]interface{})
regStruct["vod"] = []vod{}
regStruct["live"] = []live{}

var _vodstr = "vod"
t := reflect.ValueOf(regStruct[_vodstr]).Type() // 获取对象类型
_vod := reflect.New(t) // 创建实例类型为 *[]vod，使用json.Unmarshal无法直接修改_vod其实例指针的对应指针的值，需要获取到其值所对应的地址
_vodSlice := _vod.Interface() // 获取_vod的实例

json.Unmarshal([]byte(_json),&_vodSlice)
// 转化成统一的[]interface{}返回，[]vod直接转成[]interface{}会失败，无法统一转化，需要逐个元素进行转换
var res []interface{}
for i := 0; i < _vod.Elem().Len(); i++ {
    res = append(res, _vod.Elem().Index(i).Interface())
}

之后把解析好了的数据统一放到一个切片中，随机取其中的部分数据，完成随机生成数据的功能。

// 获取数据
func getData(length int, dataset ...[]interface{}) []interface{} {
	ds := make([]interface{}, 0)
	for i := 0; i < len(dataset); i++ {
		ds = append(ds, dataset[i]...)
	}

	res := make([]interface{}, length)
	rand.Seed(time.Now().UnixNano())  

	for i := 0; i < length; i++ {
		x := rand.Intn(len(ds))
		res[i] = ds[x]
	}

	return res
}

完整mock代码

package main

import (
  "fmt"
  "os"
  "io/ioutil"
  "encoding/json"
  "net/http"
  "math/rand"
  "time"
  "reflect"
)

type vod struct {
  ID string `json:"id"`
  Title string `json:"title"`
  CommentID string `json:"commentId"`
}
// live 直播数据
type live struct {
  ID string `json:"id"`
  Title string `json:"title"`
  ChatID string `json:"chatId"`
}
var (
  vs []interface{}
  ls []interface{}
  regStruct map[string]interface{}
)
// 获取数据
func getData(length int, dataset ...[]interface{}) []interface{} {
	ds := make([]interface{}, 0)
	for i := 0; i < len(dataset); i++ {
		ds = append(ds, dataset[i]...)
	}

	res := make([]interface{}, length)
	rand.Seed(time.Now().UnixNano())  

	for i := 0; i < length; i++ {
		x := rand.Intn(len(ds))
		res[i] = ds[x]
	}

	return res
}
// 根据名称预加载数据
func generaterData(name string, file string) []interface{} {
	t := reflect.ValueOf(regStruct[name]).Type()

	_json := reflect.New(t)
	_jsonSlice := _json.Interface()

	json.Unmarshal([]byte(read(file)),&_jsonSlice)
	 
	var res []interface{}
	for i := 0; i < _json.Elem().Len(); i++ {
    res = append(res, _json.Elem().Index(i).Interface())
	}

	return res
}
func main() {
	regStruct = make(map[string]interface{})
	regStruct["vod"] = []vod{}
	regStruct["live"] = []live{}

	vs = generaterData("vod", "./vod.json")
	ls = generaterData("live", "./live.json")
	// 从两个的json数据中随机获取6个数据
    res := getData(6, vs, ls)
    fmt.Printlf("Data: %+v", res)
}