string - 将工作日字符串解析为 time.Weekday

Question

我遇到了将星期几字符串转换为 time.Weekday 值的问题。我找不到 time 包中内置的任何内容。

然后我编写了这个简单的函数(满足我的需求):

var daysOfWeek = [...]string{
    "Sunday",
    "Monday",
    "Tuesday",
    "Wednesday",
    "Thursday",
    "Friday",
    "Saturday",
}

func parseWeekday(v string) (time.Weekday, error) {
    for i := range daysOfWeek {
        if daysOfWeek[i] == v {
            return time.Weekday(i), nil
        }
    }

    return time.Sunday, fmt.Errorf("invalid weekday '%s'", v)
}

在 Golang 中是否有任何其他推荐或更惯用的方式来执行此转换？

谢谢!

Answer 1

前言:我在 github.com/icza/gox 中发布了这个实用程序, 请参阅 timex.ParseWeekday() .

---

是的，使用映射而不是数组，因此查找更快更直接:

var daysOfWeek = map[string]time.Weekday{
    "Sunday":    time.Sunday,
    "Monday":    time.Monday,
    "Tuesday":   time.Tuesday,
    "Wednesday": time.Wednesday,
    "Thursday":  time.Thursday,
    "Friday":    time.Friday,
    "Saturday":  time.Saturday,
}

func parseWeekday(v string) (time.Weekday, error) {
    if d, ok := daysOfWeek[v]; ok {
        return d, nil
    }

    return time.Sunday, fmt.Errorf("invalid weekday '%s'", v)
}

测试它:

fmt.Println(parseWeekday("Monday"))
fmt.Println(parseWeekday("Friday"))
fmt.Println(parseWeekday("invalid"))

输出(在 Go Playgorund 上尝试):

Monday <nil>
Friday <nil>
Sunday invalid weekday 'invalid'

提示:

您甚至可以使用 for 循环来安全地初始化 daysOfWeek 映射，如下所示:

var daysOfWeek = map[string]time.Weekday{}

func init() {
    for d := time.Sunday; d <= time.Saturday; d++ {
        daysOfWeek[d.String()] = d
    }
}

测试和输出是一样的。在 Go Playground 上试试这个.

这个 map 解决方案的另一个不错的特性(与您的数组解决方案相比)是您可以在同一个 map 中列出额外的有效值，这些值可以被解析为 time.Weekday 而无需额外的解析代码.

例如，让我们也将 3 个字母的短工作日名称解析为它们的 time.Weekday 等价物，例如"Mon" 到 time.Monday。

这个扩展可以用一个简单的循环来添加:

var daysOfWeek = map[string]time.Weekday{}

func init() {
    for d := time.Sunday; d <= time.Saturday; d++ {
        name := d.String()
        daysOfWeek[name] = d
        daysOfWeek[name[:3]] = d
    }
}

测试它:

fmt.Println(parseWeekday("Monday"))
fmt.Println(parseWeekday("Friday"))
fmt.Println(parseWeekday("Mon"))
fmt.Println(parseWeekday("Fri"))
fmt.Println(parseWeekday("invalid"))

输出(在 Go Playground 上尝试):

Monday <nil>
Friday <nil>
Monday <nil>
Friday <nil>
Sunday invalid weekday 'invalid'

查看类似问题:Get integer month value from string