将工作日字符串解析为 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 上试试这个。

此映射解决方案的另一个不错的属性（与您的数组解决方案相比）是您可以在同一个映射中列出其他有效值，这些值可以被解析为 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

Answer 2

这看起来可以做到：

package main
import "time"

func parseWeekday(v string) (time.Weekday, error) {
   t, e := time.Parse("Monday 2", v + " 2")
   if e != nil { return 0, e }
   return t.Weekday(), nil
}

func main() {
   n, e := parseWeekday("Sunday")
   if e != nil {
      panic(e)
   }
   println(n == time.Sunday)
}

https://golang.org/pkg/time#Parse