Added Datetime64 and Timestamp64 to params builder

Author: asmyasnikov

internal/params/builder_test.go

@@ -249,6 +249,21 @@ func TestBuilder(t *testing.T) {
 				},
 			},
 		},
+		{
+			method: "Datetime64",
+			args:   []any{time.Unix(123456789, 456)},
+
+			expected: expected{
+				Type: &Ydb.Type{
+					Type: &Ydb.Type_TypeId{TypeId: Ydb.Type_DATETIME64},
+				},
+				Value: &Ydb.Value{
+					Value: &Ydb.Value_Int64Value{
+						Int64Value: 123456789,
+					},
+				},
+			},
+		},
 		{
 			method: "Date",
 			args:   []any{time.Unix(123456789, 456)},
@@ -264,6 +279,21 @@ func TestBuilder(t *testing.T) {
 				},
 			},
 		},
+		{
+			method: "Date32",
+			args:   []any{time.Unix(123456789, 456)},
+
+			expected: expected{
+				Type: &Ydb.Type{
+					Type: &Ydb.Type_TypeId{TypeId: Ydb.Type_DATE32},
+				},
+				Value: &Ydb.Value{
+					Value: &Ydb.Value_Int32Value{
+						Int32Value: 1428,
+					},
+				},
+			},
+		},
 		{
 			method: "Timestamp",
 			args:   []any{time.Unix(123456789, 456)},
@@ -279,6 +309,21 @@ func TestBuilder(t *testing.T) {
 				},
 			},
 		},
+		{
+			method: "Timestamp64",
+			args:   []any{time.Unix(123456789, 456)},
+
+			expected: expected{
+				Type: &Ydb.Type{
+					Type: &Ydb.Type_TypeId{TypeId: Ydb.Type_TIMESTAMP64},
+				},
+				Value: &Ydb.Value{
+					Value: &Ydb.Value_Int64Value{
+						Int64Value: 123456789000000,
+					},
+				},
+			},
+		},
 		{
 			method: "Decimal",
 			args:   []any{[...]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2, 3, 4, 5, 6}, uint32(22), uint32(9)},

internal/params/parameters.go

@@ -273,6 +273,13 @@ func (p *Parameter) Timestamp(v time.Time) Builder {
 	return p.parent
 }
 
+func (p *Parameter) Timestamp64(v time.Time) Builder {
+	p.value = value.Timestamp64ValueFromTime(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
 func (p *Parameter) Date(v time.Time) Builder {
 	p.value = value.DateValueFromTime(v)
 	p.parent.params = append(p.parent.params, p)
@@ -294,6 +301,13 @@ func (p *Parameter) Datetime(v time.Time) Builder {
 	return p.parent
 }
 
+func (p *Parameter) Datetime64(v time.Time) Builder {
+	p.value = value.Datetime64ValueFromTime(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
 func (p *Parameter) Interval(v time.Duration) Builder {
 	p.value = value.IntervalValueFromDuration(v)
 	p.parent.params = append(p.parent.params, p)