Go Enums

role.go

type Role int

const (
	RoleUnknown Role = iota
	_
	_
	RoleUser
	_
	_
	RoleAdmin
)

var roleStrings = []string{"Unknown", "", "", "User", "", "", "Admin"}

func (r Role) String() string {
	return roleStrings[r]
}

func RoleFromString(r string) Role {
	for idx, elem := range roleStrings {
		if elem == r {
			return Role(idx)
		}
	}
	return RoleUnknown
}

func (r *Role) UnmarshalJSON(b []byte) error {
	roleStr := strings.Trim(string(b), `"`)
	*r = RoleFromString(roleStr)
	return nil
}

func (r Role) MarshalJSON() ([]byte, error) {
	return json.Marshal(r.String())
}