json document query builder

Author: size12

internal/params/builder_test.go

@@ -390,6 +390,25 @@ func TestBuilder(t *testing.T) {
 				},
 			},
 		},
+		{
+			name: xtest.CurrentFileLine(),
+			builder: Builder{}.
+				Param("$x").JsonDocument(`{"a": 1,"b": "B"}`).Build(),
+			params: map[string]*Ydb.TypedValue{
+				"$x": {
+					Type: &Ydb.Type{
+						Type: &Ydb.Type_TypeId{
+							TypeId: Ydb.Type_JSON_DOCUMENT,
+						},
+					},
+					Value: &Ydb.Value{
+						Value: &Ydb.Value_TextValue{
+							TextValue: `{"a": 1,"b": "B"}`,
+						},
+					},
+				},
+			},
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			a := allocator.New()

internal/params/optional.go

@@ -140,3 +140,9 @@ func (p *optional) Json(v string) *optionalBuilder {
 
 	return &optionalBuilder{opt: p}
 }
+
+func (p *optional) JsonDocument(v string) *optionalBuilder {
+	p.value = value.JSONDocumentValue(v)
+
+	return &optionalBuilder{opt: p}
+}

internal/params/optional_test.go

@@ -477,6 +477,30 @@ func TestOptional(t *testing.T) {
 				},
 			},
 		},
+		{
+			name:    xtest.CurrentFileLine(),
+			builder: Builder{}.Param("$x").Optional().JsonDocument(`{"a": 1,"b": "B"}`).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_JSON_DOCUMENT,
+									},
+								},
+							},
+						},
+					},
+					Value: &Ydb.Value{
+						Value: &Ydb.Value_TextValue{
+							TextValue: `{"a": 1,"b": "B"}`,
+						},
+					},
+				},
+			},
+		},
 	} {
 		t.Run(tt.name, func(t *testing.T) {
 			a := allocator.New()

internal/params/parameters.go

@@ -249,6 +249,13 @@ func (p *Parameter) Json(v string) Builder {
 	return p.parent
 }
 
+func (p *Parameter) JsonDocument(v string) Builder {
+	p.value = value.JSONDocumentValue(v)
+	p.parent.params = append(p.parent.params, p)
+
+	return p.parent
+}
+
 func Declare(p *Parameter) string {
 	return fmt.Sprintf(
 		"DECLARE %s AS %s",