diff --git a/types/daterange.go b/types/daterange.go index 5ac1089..ef46d3f 100644 --- a/types/daterange.go +++ b/types/daterange.go @@ -91,9 +91,10 @@ func (dr *DateRange) UnmarshalJSON(data []byte) error { return nil } -func (dr *DateRange) SetLower(lower Date, bound pgtype.BoundType) { +func (dr *DateRange) SetLower(lower Date, bound ...pgtype.BoundType) { + bound = append(bound, pgtype.Inclusive) dr.Range.Lower = lower - dr.Range.LowerType = bound + dr.Range.LowerType = bound[0] } func (dr *DateRange) SetLowerUnbounded() { @@ -101,9 +102,10 @@ func (dr *DateRange) SetLowerUnbounded() { dr.Range.LowerType = pgtype.Unbounded } -func (dr *DateRange) SetUpper(upper Date, bound pgtype.BoundType) { +func (dr *DateRange) SetUpper(upper Date, bound ...pgtype.BoundType) { + bound = append(bound, pgtype.Inclusive) dr.Range.Upper = upper - dr.Range.UpperType = bound + dr.Range.UpperType = bound[0] } func (dr *DateRange) SetUpperUnbounded() { diff --git a/types/datetimerange.go b/types/datetimerange.go index 419a035..5927d8d 100644 --- a/types/datetimerange.go +++ b/types/datetimerange.go @@ -91,9 +91,10 @@ func (dr *DateTimeRange) UnmarshalJSON(data []byte) error { return nil } -func (dr *DateTimeRange) SetLower(lower DateTime, bound pgtype.BoundType) { +func (dr *DateTimeRange) SetLower(lower DateTime, bound ...pgtype.BoundType) { + bound = append(bound, pgtype.Inclusive) dr.Range.Lower = lower - dr.Range.LowerType = bound + dr.Range.LowerType = bound[0] } func (dr *DateTimeRange) SetLowerUnbounded() { @@ -101,9 +102,10 @@ func (dr *DateTimeRange) SetLowerUnbounded() { dr.Range.LowerType = pgtype.Unbounded } -func (dr *DateTimeRange) SetUpper(upper DateTime, bound pgtype.BoundType) { +func (dr *DateTimeRange) SetUpper(upper DateTime, bound ...pgtype.BoundType) { + bound = append(bound, pgtype.Inclusive) dr.Range.Upper = upper - dr.Range.UpperType = bound + dr.Range.UpperType = bound[0] } func (dr *DateTimeRange) SetUpperUnbounded() {