mirror of
https://github.com/dolthub/dolt.git
synced 2026-01-26 10:37:04 -06:00
2769353a55
Due to limitations in Go we cannot create a Def for a Map or Set that has a key that is a Map, Set or a List. This is because the key if a Go map needs to be a comparable and maps and slices are not comparable.
118 lines
3.5 KiB
Cheetah
118 lines
3.5 KiB
Cheetah
// {{.Name}}
|
|
|
|
type {{.Name}} struct {
|
|
m types.Map
|
|
}
|
|
|
|
func New{{.Name}}() {{.Name}} {
|
|
return {{.Name}}{types.NewMap(
|
|
types.NewString("$name"), types.NewString("{{.Name}}"),
|
|
{{range .Fields}}types.NewString("{{title .Name}}"), {{valueZero .T}},
|
|
{{end}}{{if .HasUnion}}types.NewString("$unionIndex"), types.UInt32(0),
|
|
types.NewString("$unionValue"), {{valueZero .UnionZeroType}},{{end}}
|
|
)}
|
|
}
|
|
|
|
{{if .CanUseDef}}
|
|
type {{.Name}}Def struct {
|
|
{{range .Fields}}{{title .Name}} {{defType .T}}
|
|
{{end}}{{if .HasUnion}}__unionIndex uint32
|
|
__unionValue interface{}
|
|
{{end}}}
|
|
|
|
|
|
func (def {{.Name}}Def) New() {{.Name}} {
|
|
return {{.Name}}{
|
|
types.NewMap(
|
|
types.NewString("$name"), types.NewString("{{.Name}}"),
|
|
{{range .Fields}}types.NewString("{{title .Name}}"), {{defToValue (print "def." (title .Name)) .T}},
|
|
{{end}}{{if .HasUnion}}types.NewString("$unionIndex"), types.UInt32(def.__unionIndex),
|
|
types.NewString("$unionValue"), def.__unionDefToValue(),
|
|
{{end}}
|
|
)}
|
|
}
|
|
|
|
func (self {{.Name}}) Def() {{.Name}}Def {
|
|
return {{.Name}}Def{
|
|
{{range .Fields}}{{valueToDef (printf `self.m.Get(types.NewString("%s"))` (title .Name)) .T}},
|
|
{{end}}{{if .HasUnion}}uint32(self.m.Get(types.NewString("$unionIndex")).(types.UInt32)),
|
|
self.__unionValueToDef(),{{end}}
|
|
}
|
|
}
|
|
|
|
{{if .HasUnion}}
|
|
func (def {{.Name}}Def) __unionDefToValue() types.Value {
|
|
switch def.__unionIndex {
|
|
{{range $index, $field := .Choices}}case {{$index}}:
|
|
return {{defToValue (printf "def.__unionValue.(%s)" (defType .T)) .T}}
|
|
{{end}}}
|
|
panic("unreachable")
|
|
}
|
|
|
|
func (self {{.Name}}) __unionValueToDef() interface{} {
|
|
switch uint32(self.m.Get(types.NewString("$unionIndex")).(types.UInt32)) {
|
|
{{range $index, $field := .Choices}}case {{$index}}:
|
|
return {{valueToDef `self.m.Get(types.NewString("$unionValue"))` .T}}
|
|
{{end}}}
|
|
panic("unreachable")
|
|
}
|
|
{{end}}
|
|
{{end}}
|
|
|
|
func {{.Name}}FromVal(val types.Value) {{.Name}} {
|
|
// TODO: Validate here
|
|
return {{.Name}}{val.(types.Map)}
|
|
}
|
|
|
|
func (self {{.Name}}) NomsValue() types.Value {
|
|
return self.m
|
|
}
|
|
|
|
func (self {{.Name}}) Equals(other {{.Name}}) bool {
|
|
return self.m.Equals(other.m)
|
|
}
|
|
|
|
func (self {{.Name}}) Ref() ref.Ref {
|
|
return self.m.Ref()
|
|
}
|
|
|
|
{{$name := .Name}}
|
|
{{range $index, $field := .Fields}}
|
|
func (self {{$name}}) {{title .Name}}() {{userType .T}} {
|
|
return {{valueToUser (printf `self.m.Get(types.NewString("%s"))` (title .Name)) .T}}
|
|
}
|
|
|
|
func (self {{$name}}) Set{{title .Name}}(val {{userType .T}}) {{$name}} {
|
|
return {{$name}}{self.m.Set(types.NewString("{{title .Name}}"), {{userToValue "val" .T}})}
|
|
}
|
|
{{end}}
|
|
|
|
{{$canUseDef := .CanUseDef}}
|
|
{{range $index, $field := .Choices}}
|
|
func (self {{$name}}) {{title .Name}}() (val {{userType .T}}, ok bool) {
|
|
if self.m.Get(types.NewString("$unionIndex")).(types.UInt32) != {{$index}} {
|
|
return
|
|
}
|
|
return {{valueToUser `self.m.Get(types.NewString("$unionValue"))` .T}}, true
|
|
}
|
|
|
|
func (self {{$name}}) Set{{title .Name}}(val {{userType .T}}) {{$name}} {
|
|
return {{$name}}{self.m.Set(types.NewString("$unionIndex"), types.UInt32({{$index}})).Set(types.NewString("$unionValue"), {{userToValue "val" .T}})}
|
|
}
|
|
|
|
{{if $canUseDef}}
|
|
func (def {{$name}}Def) {{title .Name}}() (val {{defType .T}}, ok bool) {
|
|
if def.__unionIndex != {{$index}} {
|
|
return
|
|
}
|
|
return def.__unionValue.({{defType .T}}), true
|
|
}
|
|
|
|
func (def {{$name}}Def) Set{{title .Name}}(val {{defType .T}}) {{$name}}Def {
|
|
def.__unionIndex = {{$index}}
|
|
def.__unionValue = val
|
|
return def
|
|
}
|
|
{{end}}
|
|
{{end}}
|