Skip to content

Commit

Permalink
add: custom enum
Browse files Browse the repository at this point in the history
  • Loading branch information
GreatDiscovery committed Sep 20, 2024
1 parent 72ebcd8 commit c80e447
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions test/basic/enum_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ package main

import (
"fmt"
"strings"
"testing"
)

Expand All @@ -26,8 +27,29 @@ func (d Day) String() string {
return names[d]
}

func DayOf(day string) Day {
switch strings.ToLower(day) {
case "sunday":
return Sunday
case "monday":
return Monday
case "tuesday":
return Tuesday
case "wednesday":
return Wednesday
case "thursday":
return Thursday
case "friday":
return Friday
case "saturday":
return Saturday
}
return Sunday
}

func TestEnum(t *testing.T) {
fmt.Println(Sunday) // 输出: Sunday
fmt.Println(Wednesday) // 输出: Wednesday
fmt.Println(Day(7)) // 输出: Unknown
fmt.Println(DayOf("friday"))
}

0 comments on commit c80e447

Please sign in to comment.