Merge pull request #1088 from size12/set-query-params

Set query params

Author: asmyasnikov

internal/params/parameters.go

@@ -109,6 +109,13 @@ func (p *Parameter) List() *list {
 	}
 }
 
+func (p *Parameter) Set() *set {
+	return &set{
+		parent: p.parent,
+		name:   p.name,
+	}
+}
+
 func (p *Parameter) Text(v string) Builder {
 	p.value = value.TextValue(v)
 	p.parent.params = append(p.parent.params, p)

internal/params/set.go

@@ -0,0 +1,143 @@
+package params
+
+import (
+	"time"
+
+	"github.com/ydb-platform/ydb-go-sdk/v3/internal/value"
+)
+
+type (
+	set struct {
+		parent Builder
+		name   string
+		values []value.Value
+	}
+
+	setItem struct {
+		parent *set
+	}
+)
+
+func (s *set) AddItem() *setItem {
+	return &setItem{
+		parent: s,
+	}
+}
+
+func (s *set) Build() Builder {
+	s.parent.params = append(s.parent.params, &Parameter{
+		parent: s.parent,
+		name:   s.name,
+		value:  value.SetValue(s.values...),
+	})
+
+	return s.parent
+}
+
+func (s *setItem) Text(v string) *set {
+	s.parent.values = append(s.parent.values, value.TextValue(v))
+
+	return s.parent
+}
+
+func (s *setItem) Bytes(v []byte) *set {
+	s.parent.values = append(s.parent.values, value.BytesValue(v))
+
+	return s.parent
+}
+
+func (s *setItem) Bool(v bool) *set {
+	s.parent.values = append(s.parent.values, value.BoolValue(v))
+
+	return s.parent
+}
+
+func (s *setItem) Uint64(v uint64) *set {
+	s.parent.values = append(s.parent.values, value.Uint64Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Int64(v int64) *set {
+	s.parent.values = append(s.parent.values, value.Int64Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Uint32(v uint32) *set {
+	s.parent.values = append(s.parent.values, value.Uint32Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Int32(v int32) *set {
+	s.parent.values = append(s.parent.values, value.Int32Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Uint16(v uint16) *set {
+	s.parent.values = append(s.parent.values, value.Uint16Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Int16(v int16) *set {
+	s.parent.values = append(s.parent.values, value.Int16Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Uint8(v uint8) *set {
+	s.parent.values = append(s.parent.values, value.Uint8Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Int8(v int8) *set {
+	s.parent.values = append(s.parent.values, value.Int8Value(v))
+
+	return s.parent
+}
+
+func (s *setItem) Float(v float32) *set {
+	s.parent.values = append(s.parent.values, value.FloatValue(v))
+
+	return s.parent
+}
+
+func (s *setItem) Double(v float64) *set {
+	s.parent.values = append(s.parent.values, value.DoubleValue(v))
+
+	return s.parent
+}
+
+func (s *setItem) Decimal(v [16]byte, precision, scale uint32) *set {
+	s.parent.values = append(s.parent.values, value.DecimalValue(v, precision, scale))
+
+	return s.parent
+}
+
+func (s *setItem) Timestamp(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.TimestampValueFromTime(v))
+
+	return s.parent
+}
+
+func (s *setItem) Date(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.DateValueFromTime(v))
+
+	return s.parent
+}
+
+func (s *setItem) Datetime(v time.Time) *set {
+	s.parent.values = append(s.parent.values, value.DatetimeValueFromTime(v))
+
+	return s.parent
+}
+
+func (s *setItem) Interval(v time.Duration) *set {
+	s.parent.values = append(s.parent.values, value.IntervalValueFromDuration(v))
+
+	return s.parent
+}