Initial implementation

holiday/model.go

@@ -0,0 +1,31 @@
+package holiday
+
+import (
+	"github.com/google/uuid"
+	"time"
+)
+
+type Holiday struct {
+	Id                 uuid.UUID `db:"id"`
+	Country            string    `db:"country"`
+	Date               time.Time `db:"date"`
+	Name               string    `db:"name"`
+	Description        string    `db:"description"`
+	IsStateHoliday     bool      `db:"is_state"`
+	IsReligiousHoliday bool      `db:"is_religious"`
+}
+
+type Paging struct {
+	PageSize int
+	Page     int
+}
+
+type Search struct {
+	Country          string
+	Year             *int
+	Date             *time.Time
+	RangeStart       *time.Time
+	RangeEnd         *time.Time
+	StateHoliday     *bool
+	ReligiousHoliday *bool
+}

holiday/service.go

@@ -0,0 +1,111 @@
+package holiday
+
+import (
+	"github.com/google/uuid"
+	"github.com/jmoiron/sqlx"
+	"strconv"
+	"strings"
+	"time"
+)
+
+type Service struct {
+	DB *sqlx.DB
+}
+
+func (s *Service) Find(search Search, paging Paging) ([]Holiday, error) {
+	var holidays []Holiday
+	var err error
+	if search.Date != nil {
+		holidays, err = s.findByDate(*search.Date, search.StateHoliday, search.ReligiousHoliday, search.Country)
+	} else if search.RangeStart != nil || search.RangeEnd != nil {
+		holidays, err = s.findForRange(getDateOrDefault(search.RangeStart, 1000), getDateOrDefault(search.RangeEnd, 3000), search.StateHoliday, search.ReligiousHoliday, search.Country)
+	} else if search.Year != nil {
+		holidays, err = s.findByYear(*search.Year, search.StateHoliday, search.ReligiousHoliday, search.Country)
+	} else {
+		holidays, err = s.find(search.StateHoliday, search.ReligiousHoliday, search.Country)
+	}
+	if err != nil {
+		return nil, err
+	}
+	return s.paginate(holidays, paging), err
+}
+
+func (s *Service) FindById(id uuid.UUID) (Holiday, error) {
+	var holiday Holiday
+	return holiday, s.DB.Get(&holiday, `SELECT * FROM "holiday" WHERE "id" = $1;`, id)
+}
+
+func (s *Service) findByDate(date time.Time, isState *bool, isReligious *bool, country string) ([]Holiday, error) {
+	var holidays []Holiday
+	return holidays, s.DB.Select(&holidays, `SELECT * FROM "holiday" WHERE "date" = $1 AND country = $2 `+s.filter(isState, isReligious)+";", date, country)
+}
+
+func (s *Service) findForRange(rangeStart time.Time, rangeEnd time.Time, isState *bool, isReligious *bool, country string) ([]Holiday, error) {
+	var holidays []Holiday
+	return holidays, s.DB.Select(&holidays, `SELECT * FROM "holiday" WHERE "date" BETWEEN $1 AND $2 AND country = $3`+s.filter(isState, isReligious)+";", rangeStart, rangeEnd, country)
+}
+
+func (s *Service) findByYear(year int, isState *bool, isReligious *bool, country string) ([]Holiday, error) {
+	var holidays []Holiday
+	return holidays, s.DB.Select(&holidays, `SELECT * FROM "holiday" WHERE extract(year from "date") = $1 AND country = $2 `+s.filter(isState, isReligious)+";", year, country)
+}
+
+func (s *Service) find(isState *bool, isReligious *bool, country string) ([]Holiday, error) {
+	var holidays []Holiday
+	return holidays, s.DB.Select(&holidays, `SELECT * FROM "holiday" WHERE country = $1 `+s.filter(isState, isReligious)+";", country)
+}
+
+func (s *Service) paginate(holidays []Holiday, paging Paging) []Holiday {
+	start := paging.Page * paging.PageSize
+	end := (paging.Page + 1) * paging.PageSize
+	if end < len(holidays) {
+		return holidays[start:end]
+	} else if start < len(holidays) {
+		return holidays[start:]
+	} else {
+		return []Holiday{}
+	}
+}
+
+func (s *Service) filter(isState *bool, isReligious *bool) string {
+	var filters []string
+	if isState != nil {
+		filters = append(filters, "is_state = "+strconv.FormatBool(*isState))
+	}
+	if isReligious != nil {
+		filters = append(filters, "is_religious = "+strconv.FormatBool(*isReligious))
+	}
+	if len(filters) > 0 {
+		return " AND " + strings.Join(filters, " AND ")
+	} else {
+		return ""
+	}
+}
+
+func (s *Service) Update(holiday Holiday) (Holiday, error) {
+	_, err := s.DB.Exec(`UPDATE holiday SET "country" = $1, "name" = $2, "description" = $3, "date" = $4, "is_state"=$5, "is_religious"=$6 WHERE "id" = $7`,
+		&holiday.Country, &holiday.Name, &holiday.Description, &holiday.Date, &holiday.IsStateHoliday, &holiday.IsReligiousHoliday, &holiday.Id,
+	)
+	return holiday, err
+}
+
+func (s *Service) Create(holiday Holiday) (Holiday, error) {
+	holiday.Id = uuid.Must(uuid.NewRandom())
+	_, err := s.DB.Exec(`INSERT INTO holiday (id, country, name, description, date, is_state, is_religious) values ($1, $2, $3, $4, $5, $6, $7)`,
+		&holiday.Id, &holiday.Country, &holiday.Name, &holiday.Description, &holiday.Date, &holiday.IsStateHoliday, &holiday.IsReligiousHoliday,
+	)
+	return holiday, err
+}
+
+func (s *Service) Delete(id uuid.UUID) error {
+	_, err := s.DB.Exec(`DELETE FROM holiday WHERE "id" = $1`, &id)
+	return err
+}
+
+func getDateOrDefault(date *time.Time, year int) time.Time {
+	if date == nil {
+		return time.Date(year, 1, 1, 0, 0, 0, 0, time.UTC)
+	} else {
+		return *date
+	}
+}