Add MarshalBinary() func to enums

asmyasnikov · Feb 26, 2021 · be15b97 · be15b97
1 parent d2b09c4
commit be15b97
Show file tree

Hide file tree

Showing 13 changed files with 5,485 additions and 0 deletions.
diff --git a/mavgen/mavgen.go b/mavgen/mavgen.go
@@ -561,6 +561,7 @@ func (d *Dialect) generateGo(dialectPath string, packageName string, commonPacka
 			if len(d.Enums) > 0 {
 				return []string{
 					"fmt",
+					"strconv",
 				}
 			}
 			return nil
@@ -644,6 +645,10 @@ func (d *Dialect) generateEnums(w io.Writer) error {
 // {{$enumName}} type{{if .Description}}. {{.Description}}{{end}}
 type {{.Name}} int
 
+func (e {{.Name}}) MarshalBinary() (data []byte, err error) {
+	return []byte(strconv.Itoa(int(e))), nil
+}
+
 const ({{range .Entries}}
 	// {{.Name}} enum{{if .Description}}. {{.Description}}{{end}}
 	{{.Name}} {{$enumName}} = {{.Value}} {{end}}