init
This commit is contained in:
commit
f3824d4e42
16
Readme.md
Normal file
16
Readme.md
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
This is little library to operate with bank holidays in Poland.
|
||||||
|
|
||||||
|
```go
|
||||||
|
|
||||||
|
import "git.ma-al.com/goc_marek/holidayspl"
|
||||||
|
|
||||||
|
holidays := holidayspl.New()
|
||||||
|
|
||||||
|
// to get list of all holidays
|
||||||
|
holidays.GetHolidaysList(d.Year())
|
||||||
|
|
||||||
|
// get holiday by date or false as second returned value
|
||||||
|
hh, ok := holidays.GetByDate(time.Date(2024, time.March, 31, 0, 0, 0, 0, time.UTC))
|
||||||
|
|
||||||
|
```
|
||||||
|
For more info just look at th tests file.
|
170
holidays.go
Normal file
170
holidays.go
Normal file
@ -0,0 +1,170 @@
|
|||||||
|
package holidayspl
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Holidays struct {
|
||||||
|
list []Holiday
|
||||||
|
cache map[int][]Holiday
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() *Holidays {
|
||||||
|
return &Holidays{
|
||||||
|
list: createList(),
|
||||||
|
cache: make(map[int][]Holiday),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (hol *Holidays) GetEasterDate(year int) time.Time {
|
||||||
|
a := year % 19
|
||||||
|
b := year / 100
|
||||||
|
c := year % 100
|
||||||
|
d := b / 4
|
||||||
|
e := b % 4
|
||||||
|
f := (b + 8) / 25
|
||||||
|
g := (b - f + 1) / 3
|
||||||
|
h := (19*a + b - d - g + 15) % 30
|
||||||
|
i := c / 4
|
||||||
|
k := c % 4
|
||||||
|
l := (32 + 2*e + 2*i - h - k) % 7
|
||||||
|
m := (a + 11*h + 22*l) / 451
|
||||||
|
n := (h + l - 7*m + 114) / 31
|
||||||
|
o := (h + l - 7*m + 114) % 31
|
||||||
|
return time.Date(year, time.Month(n), o+1, 0, 0, 0, 0, time.UTC)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (hol *Holidays) GetHolidaysList(year int) []Holiday {
|
||||||
|
// Check if the holiday list for this year is already cached
|
||||||
|
if holidays, ok := hol.cache[year]; ok {
|
||||||
|
return holidays
|
||||||
|
}
|
||||||
|
|
||||||
|
// Calculate and store holidays in the cache
|
||||||
|
easter := hol.GetEasterDate(year)
|
||||||
|
for i, h := range hol.list {
|
||||||
|
if h.HolidayType == FIXED {
|
||||||
|
hol.list[i].Date = time.Date(year, h.Month, h.Day, 0, 0, 0, 0, time.UTC)
|
||||||
|
} else if h.HolidayType == MOVABLE {
|
||||||
|
hol.list[i].Date = easter.AddDate(0, 0, h.PlusEaster)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Cache the computed holiday list
|
||||||
|
hol.cache[year] = hol.list
|
||||||
|
return hol.list
|
||||||
|
}
|
||||||
|
|
||||||
|
func (hol *Holidays) GetByDate(d time.Time) (Holiday, bool) {
|
||||||
|
for _, h := range hol.GetHolidaysList(d.Year()) {
|
||||||
|
if h.Date.Equal(d) {
|
||||||
|
return h, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Holiday{}, false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (hol *Holidays) GetCache() map[int][]Holiday {
|
||||||
|
return hol.cache
|
||||||
|
}
|
||||||
|
|
||||||
|
func createList() []Holiday {
|
||||||
|
|
||||||
|
return []Holiday{
|
||||||
|
{
|
||||||
|
Name: "New Year",
|
||||||
|
NamePL: "Nowy Rok",
|
||||||
|
Month: time.Month(1),
|
||||||
|
Day: 1,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Three Kings' Day",
|
||||||
|
NamePL: "Święto Trzech Króli",
|
||||||
|
Month: time.Month(1),
|
||||||
|
Day: 6,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Labour Day",
|
||||||
|
NamePL: "Święto Pracy",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 1,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "3 May Constitution Day",
|
||||||
|
NamePL: "Narodowe Święto Konstytucji Trzeciego Maja",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 3,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Assumption of Mary",
|
||||||
|
NamePL: "Wniebowzięcie Najświętszej Maryi Panny",
|
||||||
|
Month: time.Month(8),
|
||||||
|
Day: 15,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "All Saints' Day",
|
||||||
|
NamePL: "Wszystkich Świętych",
|
||||||
|
Month: time.Month(11),
|
||||||
|
Day: 1,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "National Independence Day",
|
||||||
|
NamePL: "Narodowe Święto Niepodległości",
|
||||||
|
Month: time.Month(11),
|
||||||
|
Day: 11,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Christmas",
|
||||||
|
NamePL: "Boże Narodzenie",
|
||||||
|
Month: time.Month(12),
|
||||||
|
Day: 25,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Second Day of Christmas",
|
||||||
|
NamePL: "Boże Narodzenie - drugi dzień",
|
||||||
|
Month: time.Month(12),
|
||||||
|
Day: 26,
|
||||||
|
HolidayType: FIXED,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Easter Sunday",
|
||||||
|
NamePL: "Niedziela Wielkanocna",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 3,
|
||||||
|
HolidayType: MOVABLE,
|
||||||
|
PlusEaster: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Easter Monday",
|
||||||
|
NamePL: "Poniedziałek Wielkanocny",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 3,
|
||||||
|
HolidayType: MOVABLE,
|
||||||
|
PlusEaster: 1,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Green Week",
|
||||||
|
NamePL: "Zielone Świątki",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 3,
|
||||||
|
HolidayType: MOVABLE,
|
||||||
|
PlusEaster: 49,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "Feast of Corpus Christi",
|
||||||
|
NamePL: "Boże Ciało",
|
||||||
|
Month: time.Month(5),
|
||||||
|
Day: 3,
|
||||||
|
HolidayType: MOVABLE,
|
||||||
|
PlusEaster: 60,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
69
holidays_test.go
Normal file
69
holidays_test.go
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
package holidayspl_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.ma-al.com/goc_marek/holidayspl"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestGetEasterDate(t *testing.T) {
|
||||||
|
holidays := holidayspl.New()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
year int
|
||||||
|
expected time.Time
|
||||||
|
}{
|
||||||
|
{2024, time.Date(2024, time.March, 31, 0, 0, 0, 0, time.UTC)},
|
||||||
|
{2025, time.Date(2025, time.April, 20, 0, 0, 0, 0, time.UTC)},
|
||||||
|
{2026, time.Date(2026, time.April, 5, 0, 0, 0, 0, time.UTC)},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
result := holidays.GetEasterDate(test.year)
|
||||||
|
if !result.Equal(test.expected) {
|
||||||
|
t.Errorf("GetEasterDate(%d) = %v; want %v", test.year, result, test.expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetByDate(t *testing.T) {
|
||||||
|
holidays := holidayspl.New()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
date time.Time
|
||||||
|
expected string
|
||||||
|
found bool
|
||||||
|
}{
|
||||||
|
{time.Date(2024, time.January, 1, 0, 0, 0, 0, time.UTC), "New Year", true},
|
||||||
|
{time.Date(2024, time.March, 31, 0, 0, 0, 0, time.UTC), "Easter Sunday", true},
|
||||||
|
{time.Date(2024, time.December, 25, 0, 0, 0, 0, time.UTC), "Christmas", true},
|
||||||
|
{time.Date(2024, time.August, 15, 0, 0, 0, 0, time.UTC), "Assumption of Mary", true},
|
||||||
|
{time.Date(2024, time.July, 4, 0, 0, 0, 0, time.UTC), "", false}, // No holiday on this date
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
result, found := holidays.GetByDate(test.date)
|
||||||
|
if found != test.found {
|
||||||
|
t.Errorf("GetByDate(%v) = %v; want %v", test.date, found, test.found)
|
||||||
|
}
|
||||||
|
if found && result.Name != test.expected {
|
||||||
|
t.Errorf("GetByDate(%v) = %v; want %v", test.date, result.Name, test.expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCaching(t *testing.T) {
|
||||||
|
holidays := holidayspl.New()
|
||||||
|
|
||||||
|
year := 2024
|
||||||
|
// First call should calculate and cache holidays for the year
|
||||||
|
holidays.GetHolidaysList(year)
|
||||||
|
// Modify the cache and see if the function returns the cached value
|
||||||
|
holidays.GetCache()[year][0].Name = "Modified New Year"
|
||||||
|
|
||||||
|
result := holidays.GetHolidaysList(year)
|
||||||
|
if result[0].Name != "Modified New Year" {
|
||||||
|
t.Errorf("Caching not working properly. Got %v, expected %v", result[0].Name, "Modified New Year")
|
||||||
|
}
|
||||||
|
}
|
24
models.go
Normal file
24
models.go
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
package holidayspl
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type HolidayType int
|
||||||
|
|
||||||
|
const (
|
||||||
|
FIXED HolidayType = iota
|
||||||
|
MOVABLE
|
||||||
|
)
|
||||||
|
|
||||||
|
func (d HolidayType) String() string {
|
||||||
|
return [...]string{"fixed", "movable"}[d]
|
||||||
|
}
|
||||||
|
|
||||||
|
type Holiday struct {
|
||||||
|
Name string
|
||||||
|
NamePL string
|
||||||
|
Month time.Month
|
||||||
|
Day int
|
||||||
|
HolidayType HolidayType
|
||||||
|
PlusEaster int
|
||||||
|
Date time.Time
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user