-
Notifications
You must be signed in to change notification settings - Fork 7
/
book.go
63 lines (48 loc) · 984 Bytes
/
book.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
package holidays
import "time"
var _ Queryer = (*book)(nil)
type book struct {
events []event
index map[string]event
}
func newBookfromEvents(events []event) (book, error) {
index := make(map[string]event)
for _, e := range events {
days, err := e.days()
if err != nil {
return book{}, err
}
for _, day := range days {
index[_key(day)] = e
}
}
return book{events, index}, nil
}
func (b *book) IsHoliday(d time.Time) (bool, error) {
e := b.findEvent(d)
if e == nil {
return isWeekend(d), nil
}
return e.isHoliday(), nil
}
func (b *book) IsWorkingday(d time.Time) (bool, error) {
e := b.findEvent(d)
if e == nil {
return !isWeekend(d), nil
}
return e.isWorkingday(), nil
}
func (b book) findEvent(d time.Time) *event {
e, ok := b.index[_key(d)]
if !ok {
return nil
}
return &e
}
func isWeekend(d time.Time) bool {
day := d.Weekday()
return day == 6 || day == 0
}
func _key(d time.Time) string {
return d.Format("2006-01-02")
}