re
/
enumer
forked from mirror/enumer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added comment to generated ...String function. Removed usage of _ in identifiers.

This commit is contained in:
Michael Wolber 2018-01-03 15:35:09 +01:00
parent fbb114e61f
commit aa50e29ee8
No known key found for this signature in database
GPG Key ID: 1920FAA6357E7650
3 changed files with 243 additions and 236 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
enumer.go
View File

@ -4,8 +4,10 @@ import "fmt"
// Arguments to format are: // Arguments to format are:
// [1]: type name // [1]: type name
const stringValueToNameMap = `func %[1]sString(s string) (%[1]s, error) { const stringValueToNameMap = `// %[1]sString retrieves an enum value from the enum constants string name.
if val, ok := _%[1]sNameToValue_map[s]; ok { // Throws an error if the param is not part of the enum.
func %[1]sString(s string) (%[1]s, error) {
if val, ok := _%[1]sNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%%s does not belong to %[1]s values", s) return 0, fmt.Errorf("%%s does not belong to %[1]s values", s)
@ -14,7 +16,7 @@ const stringValueToNameMap = `func %[1]sString(s string) (%[1]s, error) {
func (g *Generator) buildValueToNameMap(runs [][]Value, typeName string, runsThreshold int) { func (g *Generator) buildValueToNameMap(runs [][]Value, typeName string, runsThreshold int) {
// At this moment, either "g.declareIndexAndNameVars()" or "g.declareNameVars()" has been called // At this moment, either "g.declareIndexAndNameVars()" or "g.declareNameVars()" has been called
g.Printf("\nvar _%sNameToValue_map = map[string]%s{\n", typeName, typeName) g.Printf("\nvar _%sNameToValueMap = map[string]%s{\n", typeName, typeName)
thereAreRuns := len(runs) > 1 && len(runs) <= runsThreshold thereAreRuns := len(runs) > 1 && len(runs) <= runsThreshold
var n int var n int
var runID string var runID string
@ -27,7 +29,7 @@ func (g *Generator) buildValueToNameMap(runs [][]Value, typeName string, runsThr
} }
for _, value := range values { for _, value := range values {
g.Printf("\t_%s_name%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), &value) g.Printf("\t_%sName%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), &value)
n += len(value.name) n += len(value.name)
} }
} }

440
golden_test.go
View File

@ -62,29 +62,29 @@ const (
` `
const day_out = ` const day_out = `
const _Day_name = "MondayTuesdayWednesdayThursdayFridaySaturdaySunday" const _DayName = "MondayTuesdayWednesdayThursdayFridaySaturdaySunday"
var _Day_index = [...]uint8{0, 6, 13, 22, 30, 36, 44, 50} var _DayIndex = [...]uint8{0, 6, 13, 22, 30, 36, 44, 50}
func (i Day) String() string { func (i Day) String() string {
if i < 0 || i >= Day(len(_Day_index)-1) { if i < 0 || i >= Day(len(_DayIndex)-1) {
return fmt.Sprintf("Day(%d)", i) return fmt.Sprintf("Day(%d)", i)
} }
return _Day_name[_Day_index[i]:_Day_index[i+1]] return _DayName[_DayIndex[i]:_DayIndex[i+1]]
} }
var _DayNameToValue_map = map[string]Day{ var _DayNameToValueMap = map[string]Day{
_Day_name[0:6]: 0, _DayName[0:6]: 0,
_Day_name[6:13]: 1, _DayName[6:13]: 1,
_Day_name[13:22]: 2, _DayName[13:22]: 2,
_Day_name[22:30]: 3, _DayName[22:30]: 3,
_Day_name[30:36]: 4, _DayName[30:36]: 4,
_Day_name[36:44]: 5, _DayName[36:44]: 5,
_Day_name[44:50]: 6, _DayName[44:50]: 6,
} }
func DayString(s string) (Day, error) { func DayString(s string) (Day, error) {
if val, ok := _DayNameToValue_map[s]; ok { if val, ok := _DayNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Day values", s) return 0, fmt.Errorf("%s does not belong to Day values", s)
@ -104,26 +104,26 @@ const (
` `
const offset_out = ` const offset_out = `
const _Number_name = "OneTwoThree" const _NumberName = "OneTwoThree"
var _Number_index = [...]uint8{0, 3, 6, 11} var _NumberIndex = [...]uint8{0, 3, 6, 11}
func (i Number) String() string { func (i Number) String() string {
i -= 1 i -= 1
if i < 0 || i >= Number(len(_Number_index)-1) { if i < 0 || i >= Number(len(_NumberIndex)-1) {
return fmt.Sprintf("Number(%d)", i+1) return fmt.Sprintf("Number(%d)", i+1)
} }
return _Number_name[_Number_index[i]:_Number_index[i+1]] return _NumberName[_NumberIndex[i]:_NumberIndex[i+1]]
} }
var _NumberNameToValue_map = map[string]Number{ var _NumberNameToValueMap = map[string]Number{
_Number_name[0:3]: 1, _NumberName[0:3]: 1,
_Number_name[3:6]: 2, _NumberName[3:6]: 2,
_Number_name[6:11]: 3, _NumberName[6:11]: 3,
} }
func NumberString(s string) (Number, error) { func NumberString(s string) (Number, error) {
if val, ok := _NumberNameToValue_map[s]; ok { if val, ok := _NumberNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Number values", s) return 0, fmt.Errorf("%s does not belong to Number values", s)
@ -146,45 +146,45 @@ const (
const gap_out = ` const gap_out = `
const ( const (
_Gap_name_0 = "TwoThree" _GapName_0 = "TwoThree"
_Gap_name_1 = "FiveSixSevenEightNine" _GapName_1 = "FiveSixSevenEightNine"
_Gap_name_2 = "Eleven" _GapName_2 = "Eleven"
) )
var ( var (
_Gap_index_0 = [...]uint8{0, 3, 8} _GapIndex_0 = [...]uint8{0, 3, 8}
_Gap_index_1 = [...]uint8{0, 4, 7, 12, 17, 21} _GapIndex_1 = [...]uint8{0, 4, 7, 12, 17, 21}
_Gap_index_2 = [...]uint8{0, 6} _GapIndex_2 = [...]uint8{0, 6}
) )
func (i Gap) String() string { func (i Gap) String() string {
switch { switch {
case 2 <= i && i <= 3: case 2 <= i && i <= 3:
i -= 2 i -= 2
return _Gap_name_0[_Gap_index_0[i]:_Gap_index_0[i+1]] return _GapName_0[_GapIndex_0[i]:_GapIndex_0[i+1]]
case 5 <= i && i <= 9: case 5 <= i && i <= 9:
i -= 5 i -= 5
return _Gap_name_1[_Gap_index_1[i]:_Gap_index_1[i+1]] return _GapName_1[_GapIndex_1[i]:_GapIndex_1[i+1]]
case i == 11: case i == 11:
return _Gap_name_2 return _GapName_2
default: default:
return fmt.Sprintf("Gap(%d)", i) return fmt.Sprintf("Gap(%d)", i)
} }
} }
var _GapNameToValue_map = map[string]Gap{ var _GapNameToValueMap = map[string]Gap{
_Gap_name_0[0:3]: 2, _GapName_0[0:3]: 2,
_Gap_name_0[3:8]: 3, _GapName_0[3:8]: 3,
_Gap_name_1[0:4]: 5, _GapName_1[0:4]: 5,
_Gap_name_1[4:7]: 6, _GapName_1[4:7]: 6,
_Gap_name_1[7:12]: 7, _GapName_1[7:12]: 7,
_Gap_name_1[12:17]: 8, _GapName_1[12:17]: 8,
_Gap_name_1[17:21]: 9, _GapName_1[17:21]: 9,
_Gap_name_2[0:6]: 11, _GapName_2[0:6]: 11,
} }
func GapString(s string) (Gap, error) { func GapString(s string) (Gap, error) {
if val, ok := _GapNameToValue_map[s]; ok { if val, ok := _GapNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Gap values", s) return 0, fmt.Errorf("%s does not belong to Gap values", s)
@ -203,28 +203,28 @@ const (
` `
const num_out = ` const num_out = `
const _Num_name = "m_2m_1m0m1m2" const _NumName = "m_2m_1m0m1m2"
var _Num_index = [...]uint8{0, 3, 6, 8, 10, 12} var _NumIndex = [...]uint8{0, 3, 6, 8, 10, 12}
func (i Num) String() string { func (i Num) String() string {
i -= -2 i -= -2
if i < 0 || i >= Num(len(_Num_index)-1) { if i < 0 || i >= Num(len(_NumIndex)-1) {
return fmt.Sprintf("Num(%d)", i+-2) return fmt.Sprintf("Num(%d)", i+-2)
} }
return _Num_name[_Num_index[i]:_Num_index[i+1]] return _NumName[_NumIndex[i]:_NumIndex[i+1]]
} }
var _NumNameToValue_map = map[string]Num{ var _NumNameToValueMap = map[string]Num{
_Num_name[0:3]: -2, _NumName[0:3]: -2,
_Num_name[3:6]: -1, _NumName[3:6]: -1,
_Num_name[6:8]: 0, _NumName[6:8]: 0,
_Num_name[8:10]: 1, _NumName[8:10]: 1,
_Num_name[10:12]: 2, _NumName[10:12]: 2,
} }
func NumString(s string) (Num, error) { func NumString(s string) (Num, error) {
if val, ok := _NumNameToValue_map[s]; ok { if val, ok := _NumNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Num values", s) return 0, fmt.Errorf("%s does not belong to Num values", s)
@ -247,37 +247,37 @@ const (
const unum_out = ` const unum_out = `
const ( const (
_Unum_name_0 = "m0m1m2" _UnumName_0 = "m0m1m2"
_Unum_name_1 = "m_2m_1" _UnumName_1 = "m_2m_1"
) )
var ( var (
_Unum_index_0 = [...]uint8{0, 2, 4, 6} _UnumIndex_0 = [...]uint8{0, 2, 4, 6}
_Unum_index_1 = [...]uint8{0, 3, 6} _UnumIndex_1 = [...]uint8{0, 3, 6}
) )
func (i Unum) String() string { func (i Unum) String() string {
switch { switch {
case 0 <= i && i <= 2: case 0 <= i && i <= 2:
return _Unum_name_0[_Unum_index_0[i]:_Unum_index_0[i+1]] return _UnumName_0[_UnumIndex_0[i]:_UnumIndex_0[i+1]]
case 253 <= i && i <= 254: case 253 <= i && i <= 254:
i -= 253 i -= 253
return _Unum_name_1[_Unum_index_1[i]:_Unum_index_1[i+1]] return _UnumName_1[_UnumIndex_1[i]:_UnumIndex_1[i+1]]
default: default:
return fmt.Sprintf("Unum(%d)", i) return fmt.Sprintf("Unum(%d)", i)
} }
} }
var _UnumNameToValue_map = map[string]Unum{ var _UnumNameToValueMap = map[string]Unum{
_Unum_name_0[0:2]: 0, _UnumName_0[0:2]: 0,
_Unum_name_0[2:4]: 1, _UnumName_0[2:4]: 1,
_Unum_name_0[4:6]: 2, _UnumName_0[4:6]: 2,
_Unum_name_1[0:3]: 253, _UnumName_1[0:3]: 253,
_Unum_name_1[3:6]: 254, _UnumName_1[3:6]: 254,
} }
func UnumString(s string) (Unum, error) { func UnumString(s string) (Unum, error) {
if val, ok := _UnumNameToValue_map[s]; ok { if val, ok := _UnumNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Unum values", s) return 0, fmt.Errorf("%s does not belong to Unum values", s)
@ -306,49 +306,49 @@ const (
` `
const prime_out = ` const prime_out = `
const _Prime_name = "p2p3p5p7p11p13p17p19p23p29p37p41p43" const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
var _Prime_map = map[Prime]string{ var _PrimeMap = map[Prime]string{
2: _Prime_name[0:2], 2: _PrimeName[0:2],
3: _Prime_name[2:4], 3: _PrimeName[2:4],
5: _Prime_name[4:6], 5: _PrimeName[4:6],
7: _Prime_name[6:8], 7: _PrimeName[6:8],
11: _Prime_name[8:11], 11: _PrimeName[8:11],
13: _Prime_name[11:14], 13: _PrimeName[11:14],
17: _Prime_name[14:17], 17: _PrimeName[14:17],
19: _Prime_name[17:20], 19: _PrimeName[17:20],
23: _Prime_name[20:23], 23: _PrimeName[20:23],
29: _Prime_name[23:26], 29: _PrimeName[23:26],
31: _Prime_name[26:29], 31: _PrimeName[26:29],
41: _Prime_name[29:32], 41: _PrimeName[29:32],
43: _Prime_name[32:35], 43: _PrimeName[32:35],
} }
func (i Prime) String() string { func (i Prime) String() string {
if str, ok := _Prime_map[i]; ok { if str, ok := _PrimeMap[i]; ok {
return str return str
} }
return fmt.Sprintf("Prime(%d)", i) return fmt.Sprintf("Prime(%d)", i)
} }
var _PrimeNameToValue_map = map[string]Prime{ var _PrimeNameToValueMap = map[string]Prime{
_Prime_name[0:2]: 2, _PrimeName[0:2]: 2,
_Prime_name[2:4]: 3, _PrimeName[2:4]: 3,
_Prime_name[4:6]: 5, _PrimeName[4:6]: 5,
_Prime_name[6:8]: 7, _PrimeName[6:8]: 7,
_Prime_name[8:11]: 11, _PrimeName[8:11]: 11,
_Prime_name[11:14]: 13, _PrimeName[11:14]: 13,
_Prime_name[14:17]: 17, _PrimeName[14:17]: 17,
_Prime_name[17:20]: 19, _PrimeName[17:20]: 19,
_Prime_name[20:23]: 23, _PrimeName[20:23]: 23,
_Prime_name[23:26]: 29, _PrimeName[23:26]: 29,
_Prime_name[26:29]: 31, _PrimeName[26:29]: 31,
_Prime_name[29:32]: 41, _PrimeName[29:32]: 41,
_Prime_name[32:35]: 43, _PrimeName[32:35]: 43,
} }
func PrimeString(s string) (Prime, error) { func PrimeString(s string) (Prime, error) {
if val, ok := _PrimeNameToValue_map[s]; ok { if val, ok := _PrimeNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Prime values", s) return 0, fmt.Errorf("%s does not belong to Prime values", s)
@ -374,49 +374,49 @@ const (
` `
const prime_json_out = ` const prime_json_out = `
const _Prime_name = "p2p3p5p7p11p13p17p19p23p29p37p41p43" const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
var _Prime_map = map[Prime]string{ var _PrimeMap = map[Prime]string{
2: _Prime_name[0:2], 2: _PrimeName[0:2],
3: _Prime_name[2:4], 3: _PrimeName[2:4],
5: _Prime_name[4:6], 5: _PrimeName[4:6],
7: _Prime_name[6:8], 7: _PrimeName[6:8],
11: _Prime_name[8:11], 11: _PrimeName[8:11],
13: _Prime_name[11:14], 13: _PrimeName[11:14],
17: _Prime_name[14:17], 17: _PrimeName[14:17],
19: _Prime_name[17:20], 19: _PrimeName[17:20],
23: _Prime_name[20:23], 23: _PrimeName[20:23],
29: _Prime_name[23:26], 29: _PrimeName[23:26],
31: _Prime_name[26:29], 31: _PrimeName[26:29],
41: _Prime_name[29:32], 41: _PrimeName[29:32],
43: _Prime_name[32:35], 43: _PrimeName[32:35],
} }
func (i Prime) String() string { func (i Prime) String() string {
if str, ok := _Prime_map[i]; ok { if str, ok := _PrimeMap[i]; ok {
return str return str
} }
return fmt.Sprintf("Prime(%d)", i) return fmt.Sprintf("Prime(%d)", i)
} }
var _PrimeNameToValue_map = map[string]Prime{ var _PrimeNameToValueMap = map[string]Prime{
_Prime_name[0:2]: 2, _PrimeName[0:2]: 2,
_Prime_name[2:4]: 3, _PrimeName[2:4]: 3,
_Prime_name[4:6]: 5, _PrimeName[4:6]: 5,
_Prime_name[6:8]: 7, _PrimeName[6:8]: 7,
_Prime_name[8:11]: 11, _PrimeName[8:11]: 11,
_Prime_name[11:14]: 13, _PrimeName[11:14]: 13,
_Prime_name[14:17]: 17, _PrimeName[14:17]: 17,
_Prime_name[17:20]: 19, _PrimeName[17:20]: 19,
_Prime_name[20:23]: 23, _PrimeName[20:23]: 23,
_Prime_name[23:26]: 29, _PrimeName[23:26]: 29,
_Prime_name[26:29]: 31, _PrimeName[26:29]: 31,
_Prime_name[29:32]: 41, _PrimeName[29:32]: 41,
_Prime_name[32:35]: 43, _PrimeName[32:35]: 43,
} }
func PrimeString(s string) (Prime, error) { func PrimeString(s string) (Prime, error) {
if val, ok := _PrimeNameToValue_map[s]; ok { if val, ok := _PrimeNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Prime values", s) return 0, fmt.Errorf("%s does not belong to Prime values", s)
@ -458,49 +458,49 @@ const (
` `
const prime_yaml_out = ` const prime_yaml_out = `
const _Prime_name = "p2p3p5p7p11p13p17p19p23p29p37p41p43" const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
var _Prime_map = map[Prime]string{ var _PrimeMap = map[Prime]string{
2: _Prime_name[0:2], 2: _PrimeName[0:2],
3: _Prime_name[2:4], 3: _PrimeName[2:4],
5: _Prime_name[4:6], 5: _PrimeName[4:6],
7: _Prime_name[6:8], 7: _PrimeName[6:8],
11: _Prime_name[8:11], 11: _PrimeName[8:11],
13: _Prime_name[11:14], 13: _PrimeName[11:14],
17: _Prime_name[14:17], 17: _PrimeName[14:17],
19: _Prime_name[17:20], 19: _PrimeName[17:20],
23: _Prime_name[20:23], 23: _PrimeName[20:23],
29: _Prime_name[23:26], 29: _PrimeName[23:26],
31: _Prime_name[26:29], 31: _PrimeName[26:29],
41: _Prime_name[29:32], 41: _PrimeName[29:32],
43: _Prime_name[32:35], 43: _PrimeName[32:35],
} }
func (i Prime) String() string { func (i Prime) String() string {
if str, ok := _Prime_map[i]; ok { if str, ok := _PrimeMap[i]; ok {
return str return str
} }
return fmt.Sprintf("Prime(%d)", i) return fmt.Sprintf("Prime(%d)", i)
} }
var _PrimeNameToValue_map = map[string]Prime{ var _PrimeNameToValueMap = map[string]Prime{
_Prime_name[0:2]: 2, _PrimeName[0:2]: 2,
_Prime_name[2:4]: 3, _PrimeName[2:4]: 3,
_Prime_name[4:6]: 5, _PrimeName[4:6]: 5,
_Prime_name[6:8]: 7, _PrimeName[6:8]: 7,
_Prime_name[8:11]: 11, _PrimeName[8:11]: 11,
_Prime_name[11:14]: 13, _PrimeName[11:14]: 13,
_Prime_name[14:17]: 17, _PrimeName[14:17]: 17,
_Prime_name[17:20]: 19, _PrimeName[17:20]: 19,
_Prime_name[20:23]: 23, _PrimeName[20:23]: 23,
_Prime_name[23:26]: 29, _PrimeName[23:26]: 29,
_Prime_name[26:29]: 31, _PrimeName[26:29]: 31,
_Prime_name[29:32]: 41, _PrimeName[29:32]: 41,
_Prime_name[32:35]: 43, _PrimeName[32:35]: 43,
} }
func PrimeString(s string) (Prime, error) { func PrimeString(s string) (Prime, error) {
if val, ok := _PrimeNameToValue_map[s]; ok { if val, ok := _PrimeNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Prime values", s) return 0, fmt.Errorf("%s does not belong to Prime values", s)
@ -542,49 +542,49 @@ const (
` `
const prime_sql_out = ` const prime_sql_out = `
const _Prime_name = "p2p3p5p7p11p13p17p19p23p29p37p41p43" const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
var _Prime_map = map[Prime]string{ var _PrimeMap = map[Prime]string{
2: _Prime_name[0:2], 2: _PrimeName[0:2],
3: _Prime_name[2:4], 3: _PrimeName[2:4],
5: _Prime_name[4:6], 5: _PrimeName[4:6],
7: _Prime_name[6:8], 7: _PrimeName[6:8],
11: _Prime_name[8:11], 11: _PrimeName[8:11],
13: _Prime_name[11:14], 13: _PrimeName[11:14],
17: _Prime_name[14:17], 17: _PrimeName[14:17],
19: _Prime_name[17:20], 19: _PrimeName[17:20],
23: _Prime_name[20:23], 23: _PrimeName[20:23],
29: _Prime_name[23:26], 29: _PrimeName[23:26],
31: _Prime_name[26:29], 31: _PrimeName[26:29],
41: _Prime_name[29:32], 41: _PrimeName[29:32],
43: _Prime_name[32:35], 43: _PrimeName[32:35],
} }
func (i Prime) String() string { func (i Prime) String() string {
if str, ok := _Prime_map[i]; ok { if str, ok := _PrimeMap[i]; ok {
return str return str
} }
return fmt.Sprintf("Prime(%d)", i) return fmt.Sprintf("Prime(%d)", i)
} }
var _PrimeNameToValue_map = map[string]Prime{ var _PrimeNameToValueMap = map[string]Prime{
_Prime_name[0:2]: 2, _PrimeName[0:2]: 2,
_Prime_name[2:4]: 3, _PrimeName[2:4]: 3,
_Prime_name[4:6]: 5, _PrimeName[4:6]: 5,
_Prime_name[6:8]: 7, _PrimeName[6:8]: 7,
_Prime_name[8:11]: 11, _PrimeName[8:11]: 11,
_Prime_name[11:14]: 13, _PrimeName[11:14]: 13,
_Prime_name[14:17]: 17, _PrimeName[14:17]: 17,
_Prime_name[17:20]: 19, _PrimeName[17:20]: 19,
_Prime_name[20:23]: 23, _PrimeName[20:23]: 23,
_Prime_name[23:26]: 29, _PrimeName[23:26]: 29,
_Prime_name[26:29]: 31, _PrimeName[26:29]: 31,
_Prime_name[29:32]: 41, _PrimeName[29:32]: 41,
_Prime_name[32:35]: 43, _PrimeName[32:35]: 43,
} }
func PrimeString(s string) (Prime, error) { func PrimeString(s string) (Prime, error) {
if val, ok := _PrimeNameToValue_map[s]; ok { if val, ok := _PrimeNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Prime values", s) return 0, fmt.Errorf("%s does not belong to Prime values", s)
@ -639,49 +639,49 @@ const (
` `
const prime_json_and_sql_out = ` const prime_json_and_sql_out = `
const _Prime_name = "p2p3p5p7p11p13p17p19p23p29p37p41p43" const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
var _Prime_map = map[Prime]string{ var _PrimeMap = map[Prime]string{
2: _Prime_name[0:2], 2: _PrimeName[0:2],
3: _Prime_name[2:4], 3: _PrimeName[2:4],
5: _Prime_name[4:6], 5: _PrimeName[4:6],
7: _Prime_name[6:8], 7: _PrimeName[6:8],
11: _Prime_name[8:11], 11: _PrimeName[8:11],
13: _Prime_name[11:14], 13: _PrimeName[11:14],
17: _Prime_name[14:17], 17: _PrimeName[14:17],
19: _Prime_name[17:20], 19: _PrimeName[17:20],
23: _Prime_name[20:23], 23: _PrimeName[20:23],
29: _Prime_name[23:26], 29: _PrimeName[23:26],
31: _Prime_name[26:29], 31: _PrimeName[26:29],
41: _Prime_name[29:32], 41: _PrimeName[29:32],
43: _Prime_name[32:35], 43: _PrimeName[32:35],
} }
func (i Prime) String() string { func (i Prime) String() string {
if str, ok := _Prime_map[i]; ok { if str, ok := _PrimeMap[i]; ok {
return str return str
} }
return fmt.Sprintf("Prime(%d)", i) return fmt.Sprintf("Prime(%d)", i)
} }
var _PrimeNameToValue_map = map[string]Prime{ var _PrimeNameToValueMap = map[string]Prime{
_Prime_name[0:2]: 2, _PrimeName[0:2]: 2,
_Prime_name[2:4]: 3, _PrimeName[2:4]: 3,
_Prime_name[4:6]: 5, _PrimeName[4:6]: 5,
_Prime_name[6:8]: 7, _PrimeName[6:8]: 7,
_Prime_name[8:11]: 11, _PrimeName[8:11]: 11,
_Prime_name[11:14]: 13, _PrimeName[11:14]: 13,
_Prime_name[14:17]: 17, _PrimeName[14:17]: 17,
_Prime_name[17:20]: 19, _PrimeName[17:20]: 19,
_Prime_name[20:23]: 23, _PrimeName[20:23]: 23,
_Prime_name[23:26]: 29, _PrimeName[23:26]: 29,
_Prime_name[26:29]: 31, _PrimeName[26:29]: 31,
_Prime_name[29:32]: 41, _PrimeName[29:32]: 41,
_Prime_name[32:35]: 43, _PrimeName[32:35]: 43,
} }
func PrimeString(s string) (Prime, error) { func PrimeString(s string) (Prime, error) {
if val, ok := _PrimeNameToValue_map[s]; ok { if val, ok := _PrimeNameToValueMap[s]; ok {
return val, nil return val, nil
} }
return 0, fmt.Errorf("%s does not belong to Prime values", s) return 0, fmt.Errorf("%s does not belong to Prime values", s)

29
stringer.go
View File

@ -58,6 +58,11 @@
// where t is the lower-cased name of the first type listed. It can be overridden // where t is the lower-cased name of the first type listed. It can be overridden
// with the -output flag. // with the -output flag.
// //
// ------
// This is a patched versin of the original stringer. The original generates source code
// using '_' (underscore) as part of generated golang identifiers. This creates a number of
// golint warnings. The patched version simply leaves the '_' out
package main package main
import ( import (
@ -559,10 +564,10 @@ func (g *Generator) createIndexAndNameDecl(run []Value, typeName string, suffix
b.WriteString(run[i].name) b.WriteString(run[i].name)
indexes[i] = b.Len() indexes[i] = b.Len()
} }
nameConst := fmt.Sprintf("_%s_name%s = %q", typeName, suffix, b.String()) nameConst := fmt.Sprintf("_%sName%s = %q", typeName, suffix, b.String())
nameLen := b.Len() nameLen := b.Len()
b.Reset() b.Reset()
fmt.Fprintf(b, "_%s_index%s = [...]uint%d{0, ", typeName, suffix, usize(nameLen)) fmt.Fprintf(b, "_%sIndex%s = [...]uint%d{0, ", typeName, suffix, usize(nameLen))
for i, v := range indexes { for i, v := range indexes {
if i > 0 { if i > 0 {
fmt.Fprintf(b, ", ") fmt.Fprintf(b, ", ")
@ -575,7 +580,7 @@ func (g *Generator) createIndexAndNameDecl(run []Value, typeName string, suffix
// declareNameVars declares the concatenated names string representing all the values in the runs. // declareNameVars declares the concatenated names string representing all the values in the runs.
func (g *Generator) declareNameVars(runs [][]Value, typeName string, suffix string) { func (g *Generator) declareNameVars(runs [][]Value, typeName string, suffix string) {
g.Printf("const _%s_name%s = \"", typeName, suffix) g.Printf("const _%sName%s = \"", typeName, suffix)
for _, run := range runs { for _, run := range runs {
for i := range run { for i := range run {
g.Printf("%s", run[i].name) g.Printf("%s", run[i].name)
@ -606,10 +611,10 @@ func (g *Generator) buildOneRun(runs [][]Value, typeName string) {
// [2]: size of index element (8 for uint8 etc.) // [2]: size of index element (8 for uint8 etc.)
// [3]: less than zero check (for signed types) // [3]: less than zero check (for signed types)
const stringOneRun = `func (i %[1]s) String() string { const stringOneRun = `func (i %[1]s) String() string {
if %[3]si >= %[1]s(len(_%[1]s_index)-1) { if %[3]si >= %[1]s(len(_%[1]sIndex)-1) {
return fmt.Sprintf("%[1]s(%%d)", i) return fmt.Sprintf("%[1]s(%%d)", i)
} }
return _%[1]s_name[_%[1]s_index[i]:_%[1]s_index[i+1]] return _%[1]sName[_%[1]sIndex[i]:_%[1]sIndex[i+1]]
} }
` `
@ -622,10 +627,10 @@ const stringOneRun = `func (i %[1]s) String() string {
*/ */
const stringOneRunWithOffset = `func (i %[1]s) String() string { const stringOneRunWithOffset = `func (i %[1]s) String() string {
i -= %[2]s i -= %[2]s
if %[4]si >= %[1]s(len(_%[1]s_index)-1) { if %[4]si >= %[1]s(len(_%[1]sIndex)-1) {
return fmt.Sprintf("%[1]s(%%d)", i + %[2]s) return fmt.Sprintf("%[1]s(%%d)", i + %[2]s)
} }
return _%[1]s_name[_%[1]s_index[i] : _%[1]s_index[i+1]] return _%[1]sName[_%[1]sIndex[i] : _%[1]sIndex[i+1]]
} }
` `
@ -639,14 +644,14 @@ func (g *Generator) buildMultipleRuns(runs [][]Value, typeName string) {
for i, values := range runs { for i, values := range runs {
if len(values) == 1 { if len(values) == 1 {
g.Printf("\tcase i == %s:\n", &values[0]) g.Printf("\tcase i == %s:\n", &values[0])
g.Printf("\t\treturn _%s_name_%d\n", typeName, i) g.Printf("\t\treturn _%sName_%d\n", typeName, i)
continue continue
} }
g.Printf("\tcase %s <= i && i <= %s:\n", &values[0], &values[len(values)-1]) g.Printf("\tcase %s <= i && i <= %s:\n", &values[0], &values[len(values)-1])
if values[0].value != 0 { if values[0].value != 0 {
g.Printf("\t\ti -= %s\n", &values[0]) g.Printf("\t\ti -= %s\n", &values[0])
} }
g.Printf("\t\treturn _%s_name_%d[_%s_index_%d[i]:_%s_index_%d[i+1]]\n", g.Printf("\t\treturn _%sName_%d[_%sIndex_%d[i]:_%sIndex_%d[i+1]]\n",
typeName, i, typeName, i, typeName, i) typeName, i, typeName, i, typeName, i)
} }
g.Printf("\tdefault:\n") g.Printf("\tdefault:\n")
@ -660,11 +665,11 @@ func (g *Generator) buildMultipleRuns(runs [][]Value, typeName string) {
func (g *Generator) buildMap(runs [][]Value, typeName string) { func (g *Generator) buildMap(runs [][]Value, typeName string) {
g.Printf("\n") g.Printf("\n")
g.declareNameVars(runs, typeName, "") g.declareNameVars(runs, typeName, "")
g.Printf("\nvar _%s_map = map[%s]string{\n", typeName, typeName) g.Printf("\nvar _%sMap = map[%s]string{\n", typeName, typeName)
n := 0 n := 0
for _, values := range runs { for _, values := range runs {
for _, value := range values { for _, value := range values {
g.Printf("\t%s: _%s_name[%d:%d],\n", &value, typeName, n, n+len(value.name)) g.Printf("\t%s: _%sName[%d:%d],\n", &value, typeName, n, n+len(value.name))
n += len(value.name) n += len(value.name)
} }
} }
@ -674,7 +679,7 @@ func (g *Generator) buildMap(runs [][]Value, typeName string) {
// Argument to format is the type name. // Argument to format is the type name.
const stringMap = `func (i %[1]s) String() string { const stringMap = `func (i %[1]s) String() string {
if str, ok := _%[1]s_map[i]; ok { if str, ok := _%[1]sMap[i]; ok {
return str return str
} }
return fmt.Sprintf("%[1]s(%%d)", i) return fmt.Sprintf("%[1]s(%%d)", i)