Fuyu/jsonenums
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improved generated code by defining map outside of function.

Browse Source
This commit is contained in:
Francesc Campoy 2015-02-04 13:38:28 +00:00
parent 40291a96dd
commit dfe6af903e
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
template.go
View File

@ -20,32 +20,39 @@ import (
{{range $typename, $values := .TypesAndValues}} {{range $typename, $values := .TypesAndValues}}
func (r {{$typename}}) MarshalJSON() ([]byte, error) { var (
if s, ok := interface{}(r).(fmt.Stringer); ok { _{{$typename}}NameToValue = map[string]{{$typename}} {
return json.Marshal(s.String()) {{range $values}}"{{.}}": {{.}},
} {{end}}
s, ok := map[{{$typename}}]string {
{{range $values}}{{.}}: "{{.}}",{{end}}
}[r]
if !ok {
return nil, fmt.Errorf("invalid {{$typename}}: %d", r)
}
return json.Marshal(s)
} }
var _{{$typename}}NameToValue = map[string]{{$typename}} { _{{$typename}}ValueToName = map[{{$typename}}]string {
{{range $values}}"{{.}}": {{.}},{{end}} {{range $values}}{{.}}: "{{.}}",
{{end}}
} }
)
func init() { func init() {
var v {{$typename}} var v {{$typename}}
if _, ok := interface{}(v).(fmt.Stringer); ok { if _, ok := interface{}(v).(fmt.Stringer); ok {
_{{$typename}}NameToValue = map[string]{{$typename}} { _{{$typename}}NameToValue = map[string]{{$typename}} {
{{range $values}}interface{}({{.}}).(fmt.Stringer).String(): {{.}},{{end}} {{range $values}}interface{}({{.}}).(fmt.Stringer).String(): {{.}},
{{end}}
} }
} }
} }
func (r {{$typename}}) MarshalJSON() ([]byte, error) {
if s, ok := interface{}(r).(fmt.Stringer); ok {
return json.Marshal(s.String())
}
s, ok := _{{$typename}}ValueToName[r]
if !ok {
return nil, fmt.Errorf("invalid {{$typename}}: %d", r)
}
return json.Marshal(s)
}
func (r *{{$typename}}) UnmarshalJSON(data []byte) error { func (r *{{$typename}}) UnmarshalJSON(data []byte) error {
var s string var s string
if err := json.Unmarshal(data, &s); err != nil { if err := json.Unmarshal(data, &s); err != nil {