// NewDatasetWithData creates a new Dataset.
func NewDatasetWithData(headers []string, data interface{}) (*tablib.Dataset, error) {
if data == nil {
return tablib.NewDatasetWithData(headers, nil), nil
}
n := len(headers)
switch data.(type) {
case [][]interface{}:
return tablib.NewDatasetWithData(headers, data.([][]interface{})), nil
case []map[string]interface{}:
dataSlice := data.([]map[string]interface{})
if len(dataSlice) > 0 {
if len(dataSlice[0]) == 0 {
return tablib.NewDatasetWithData(headers, make([][]interface{}, len(dataSlice))), nil
} else {
if n != len(dataSlice[0]) {
return nil, ErrParamsType
}
mapHeaders := make(map[string]int, n)
for i := 0; i < n; i++ {
mapHeaders[headers[i]] = i
}
for k, _ := range dataSlice[0] {
if _, ok := mapHeaders[k]; !ok {
return nil, ErrParamsType
}
}
d := tablib.NewDataset(headers)
var row []interface{}
for i, _ := range dataSlice {
row = nil
for j := 0; j < n; j++ {
row = append(row, dataSlice[i][headers[j]])
}
d.Append(row)
}
return d, nil
}
} else {
return tablib.NewDatasetWithData(headers, nil), nil
}
default:
return nil, ErrParamsType
}
}
// NewDataset creates a new Dataset.
func NewDataset(headers []string) *tablib.Dataset {
return tablib.NewDataset(headers)
}
