tz types support

Author: size12

internal/params/dict.go

@@ -209,6 +209,30 @@ func (d *dictPair) UUID(v [16]byte) *dictValue {
 	}
 }
 
+func (d *dictPair) TzDate(v time.Time) *dictValue {
+	d.keyValue = value.TzDateValueFromTime(v)
+
+	return &dictValue{
+		pair: d,
+	}
+}
+
+func (d *dictPair) TzTimestamp(v time.Time) *dictValue {
+	d.keyValue = value.TzTimestampValueFromTime(v)
+
+	return &dictValue{
+		pair: d,
+	}
+}
+
+func (d *dictPair) TzDatetime(v time.Time) *dictValue {
+	d.keyValue = value.TzDatetimeValueFromTime(v)
+
+	return &dictValue{
+		pair: d,
+	}
+}
+
 func (d *dictValue) Text(v string) *dict {
 	d.pair.parent.values = append(d.pair.parent.values, value.DictValueField{
 		K: d.pair.keyValue,
@@ -407,6 +431,33 @@ func (d *dictValue) UUID(v [16]byte) *dict {
 	return d.pair.parent
 }
 
+func (d *dictValue) TzDate(v time.Time) *dict {
+	d.pair.parent.values = append(d.pair.parent.values, value.DictValueField{
+		K: d.pair.keyValue,
+		V: value.TzDateValueFromTime(v),
+	})
+
+	return d.pair.parent
+}
+
+func (d *dictValue) TzTimestamp(v time.Time) *dict {
+	d.pair.parent.values = append(d.pair.parent.values, value.DictValueField{
+		K: d.pair.keyValue,
+		V: value.TzTimestampValueFromTime(v),
+	})
+
+	return d.pair.parent
+}
+
+func (d *dictValue) TzDatetime(v time.Time) *dict {
+	d.pair.parent.values = append(d.pair.parent.values, value.DictValueField{
+		K: d.pair.keyValue,
+		V: value.TzDatetimeValueFromTime(v),
+	})
+
+	return d.pair.parent
+}
+
 func (d *dict) EndDict() Builder {
 	d.parent.params = append(d.parent.params, &Parameter{
 		parent: d.parent,

internal/params/list.go

@@ -170,3 +170,21 @@ func (l *listItem) UUID(v [16]byte) *list {
 
 	return l.parent
 }
+
+func (l *listItem) TzDate(v time.Time) *list {
+	l.parent.values = append(l.parent.values, value.TzDateValueFromTime(v))
+
+	return l.parent
+}
+
+func (l *listItem) TzTimestamp(v time.Time) *list {
+	l.parent.values = append(l.parent.values, value.TzTimestampValueFromTime(v))
+
+	return l.parent
+}
+
+func (l *listItem) TzDatetime(v time.Time) *list {
+	l.parent.values = append(l.parent.values, value.TzDatetimeValueFromTime(v))
+
+	return l.parent
+}

internal/params/optional.go

@@ -158,3 +158,21 @@ func (p *optional) UUID(v [16]byte) *optionalBuilder {
 
 	return &optionalBuilder{opt: p}
 }
+
+func (p *optional) TzDate(v time.Time) *optionalBuilder {
+	p.value = value.TzDateValueFromTime(v)
+
+	return &optionalBuilder{opt: p}
+}
+
+func (p *optional) TzTimestamp(v time.Time) *optionalBuilder {
+	p.value = value.TzTimestampValueFromTime(v)
+
+	return &optionalBuilder{opt: p}
+}
+
+func (p *optional) TzDatetime(v time.Time) *optionalBuilder {
+	p.value = value.TzDatetimeValueFromTime(v)
+
+	return &optionalBuilder{opt: p}
+}

internal/params/parameters.go

@@ -281,6 +281,27 @@ func (p *Parameter) UUID(v [16]byte) Builder {
 	return p.parent
 }
 
+func (p *Parameter) TzDate(v time.Time) Builder {
+	p.value = value.TzDateValueFromTime(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
+func (p *Parameter) TzTimestamp(v time.Time) Builder {
+	p.value = value.TzTimestampValueFromTime(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
+func (p *Parameter) TzDatetime(v time.Time) Builder {
+	p.value = value.TzDatetimeValueFromTime(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
 func Declare(p *Parameter) string {
 	return fmt.Sprintf(
 		"DECLARE %s AS %s",

internal/params/set.go

@@ -171,3 +171,21 @@ func (s *setItem) UUID(v [16]byte) *set {
 
 	return s.parent
 }
+
+func (s *setItem) TzDate(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.TzDateValueFromTime(v))
+
+	return s.parent
+}
+
+func (s *setItem) TzTimestamp(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.TzTimestampValueFromTime(v))
+
+	return s.parent
+}
+
+func (s *setItem) TzDatetime(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.TzDatetimeValueFromTime(v))
+
+	return s.parent
+}