mirror of
https://github.com/dolthub/dolt.git
synced 2026-04-25 19:50:32 -05:00
Erik Arvidsson
bbeb504943
* Make Type Describe use the Human Readable Serialization This reduces code duplication. * Remove Choices in favor of []Field
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package csv
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/attic-labs/noms/d"
|
|
"github.com/attic-labs/noms/types"
|
|
)
|
|
|
|
// ValueToListAndElemDesc ensures that v is a types.List of structs, pulls the types.StructDesc that describes the elements of v out of vr, and returns the List and related StructDesc.
|
|
func ValueToListAndElemDesc(v types.Value, vr types.ValueReader) (types.List, types.StructDesc) {
|
|
d.Exp.Equal(types.ListKind, v.Type().Kind(),
|
|
"Dataset must be List<>, found: %s", v.Type().Describe())
|
|
|
|
u := v.Type().Desc.(types.CompoundDesc).ElemTypes[0]
|
|
d.Exp.Equal(types.UnresolvedKind, u.Kind(),
|
|
"List<> must be UnresolvedKind, found: %s", u.Describe())
|
|
|
|
pkg := types.ReadPackage(u.PackageRef(), vr)
|
|
d.Exp.Equal(types.PackageKind, pkg.Type().Kind(),
|
|
"Failed to read package: %s", pkg.Type().Describe())
|
|
|
|
t := pkg.Types()[u.Ordinal()]
|
|
d.Exp.Equal(types.StructKind, t.Kind(), "Did not find Struct: %s", t.Describe())
|
|
return v.(types.List), t.Desc.(types.StructDesc)
|
|
}
|
|
|
|
// Write takes a types.List l of structs (described by sd) and writes it to output as comma-delineated values.
|
|
func Write(l types.List, sd types.StructDesc, comma rune, output io.Writer) {
|
|
fieldNames := getFieldNamesFromStruct(sd)
|
|
|
|
csvWriter := csv.NewWriter(output)
|
|
csvWriter.Comma = comma
|
|
|
|
d.Exp.NoError(csvWriter.Write(fieldNames), "Failed to write header %v", fieldNames)
|
|
record := make([]string, len(fieldNames))
|
|
l.IterAll(func(v types.Value, index uint64) {
|
|
for i, f := range fieldNames {
|
|
record[i] = fmt.Sprintf("%v", v.(types.Struct).Get(f))
|
|
}
|
|
d.Exp.NoError(csvWriter.Write(record), "Failed to write record %v", record)
|
|
})
|
|
|
|
csvWriter.Flush()
|
|
d.Exp.NoError(csvWriter.Error(), "error flushing csv")
|
|
}
|
|
|
|
func getFieldNamesFromStruct(structDesc types.StructDesc) (fieldNames []string) {
|
|
for _, f := range structDesc.Fields {
|
|
d.Exp.Equal(true, types.IsPrimitiveKind(f.T.Kind()),
|
|
"Non-primitive CSV export not supported:", f.T.Describe())
|
|
fieldNames = append(fieldNames, f.Name)
|
|
}
|
|
return
|
|
}
|