Merge pull request #1164 from size12/variant-tuple

feat: variant tuple params builder
ydb-platform · Mar 26, 2024 · 0ffb7af · 0ffb7af
2 parents 30d0d7f + 1fec927
commit 0ffb7af
Show file tree

Hide file tree

Showing 7 changed files with 1,339 additions and 588 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,5 @@
+* Added `Tuple` support for `Variant` in `ydb.ParamsBuilder()`
+
 ## v3.60.1
 * Added additional traces for coordination service client internals
 

diff --git a/internal/params/parameters.go b/internal/params/parameters.go
@@ -141,6 +141,13 @@ func (p *Parameter) BeginStruct() *structure {
 	}
 }
 
+func (p *Parameter) BeginVariant() *variant {
+	return &variant{
+		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)

diff --git a/internal/params/tuple.go b/internal/params/tuple.go
@@ -170,3 +170,21 @@ func (t *tupleItem) UUID(v [16]byte) *tuple {
 
 	return t.parent
 }
+
+func (t *tupleItem) TzDate(v time.Time) *tuple {
+	t.parent.values = append(t.parent.values, value.TzDateValueFromTime(v))
+
+	return t.parent
+}
+
+func (t *tupleItem) TzTimestamp(v time.Time) *tuple {
+	t.parent.values = append(t.parent.values, value.TzTimestampValueFromTime(v))
+
+	return t.parent
+}
+
+func (t *tupleItem) TzDatetime(v time.Time) *tuple {
+	t.parent.values = append(t.parent.values, value.TzDatetimeValueFromTime(v))
+
+	return t.parent
+}