Dmitriy Matrenichev 8e9fc13d7c
feat: implement enum generator for proto files
`structprotogen` now supports generating enums directly instead of using predeclared file and hardcoded types. To use this functionality, simply put `structprotogen:gen_enum` in the comment above const block, you want to have the proto definitions for.

Closes #6215

Signed-off-by: Dmitriy Matrenichev <dmitry.matrenichev@siderolabs.com>
2023-01-11 16:02:21 +03:00

36 lines
855 B
Go

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
package nethelpers
import "fmt"
//go:generate enumer -type=ADSelect -linecomment -text
// ADSelect is ADSelect.
type ADSelect uint8
// ADSelect constants.
//
//structprotogen:gen_enum
const (
ADSelectStable ADSelect = iota // stable
ADSelectBandwidth // bandwidth
ADSelectCount // count
)
// ADSelectByName parses ADSelect.
func ADSelectByName(sel string) (ADSelect, error) {
switch sel {
case "", "stable":
return ADSelectStable, nil
case "bandwidth":
return ADSelectBandwidth, nil
case "count":
return ADSelectCount, nil
default:
return 0, fmt.Errorf("invalid ad_select mode %v", sel)
}
}