yson query builder

Author: size12

internal/params/builder_test.go

@@ -409,6 +409,25 @@ func TestBuilder(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: xtest.CurrentFileLine(),
+			builder: Builder{}.
+				Param("$x").Yson([]byte(`[ 1; 2; 3; 4; 5 ]`)).Build(),
+			params: map[string]*Ydb.TypedValue{
+				"$x": {
+					Type: &Ydb.Type{
+						Type: &Ydb.Type_TypeId{
+							TypeId: Ydb.Type_YSON,
+						},
+					},
+					Value: &Ydb.Value{
+						Value: &Ydb.Value_BytesValue{
+							BytesValue: []byte(`[ 1; 2; 3; 4; 5 ]`),
+						},
+					},
+				},
+			},
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			a := allocator.New()

internal/params/optional.go

@@ -146,3 +146,9 @@ func (p *optional) JsonDocument(v string) *optionalBuilder {
 
 	return &optionalBuilder{opt: p}
 }
+
+func (p *optional) Yson(v []byte) *optionalBuilder {
+	p.value = value.YSONValue(v)
+
+	return &optionalBuilder{opt: p}
+}

internal/params/optional_test.go

@@ -501,6 +501,30 @@ func TestOptional(t *testing.T) {
 				},
 			},
 		},
+		{
+			name:    xtest.CurrentFileLine(),
+			builder: Builder{}.Param("$x").Optional().Yson([]byte(`[ 1; 2; 3; 4; 5 ]`)).Build(),
+			params: map[string]*Ydb.TypedValue{
+				"$x": {
+					Type: &Ydb.Type{
+						Type: &Ydb.Type_OptionalType{
+							OptionalType: &Ydb.OptionalType{
+								Item: &Ydb.Type{
+									Type: &Ydb.Type_TypeId{
+										TypeId: Ydb.Type_YSON,
+									},
+								},
+							},
+						},
+					},
+					Value: &Ydb.Value{
+						Value: &Ydb.Value_BytesValue{
+							BytesValue: []byte(`[ 1; 2; 3; 4; 5 ]`),
+						},
+					},
+				},
+			},
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			a := allocator.New()

internal/params/parameters.go

@@ -256,6 +256,13 @@ func (p *Parameter) JsonDocument(v string) Builder {
 	return p.parent
 }
 
+func (p *Parameter) Yson(v []byte) Builder {
+	p.value = value.YSONValue(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
 func Declare(p *Parameter) string {
 	return fmt.Sprintf(
 		"DECLARE %s AS %s",