chore: import upstream snapshot with attribution

2026-07-13 12:32:25 +08:00
commit e014feafe1
2285 changed files with 1131979 additions and 0 deletions
@@ -0,0 +1,138 @@
+// Copyright 2024 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package types
+
+import (
+	"github.com/cockroachdb/apd/v3"
+	"github.com/cockroachdb/errors"
+	"github.com/dolthub/go-mysql-server/sql"
+	gmstypes "github.com/dolthub/go-mysql-server/sql/types"
+
+	"github.com/dolthub/doltgresql/core/id"
+	"github.com/dolthub/doltgresql/utils"
+)
+
+// JsonB is the deserialized and structured version of JSON that deals with JsonDocument.
+var JsonB = &DoltgresType{
+	ID:                  toInternal("jsonb"),
+	TypLength:           int16(-1),
+	PassedByVal:         false,
+	TypType:             TypeType_Base,
+	TypCategory:         TypeCategory_UserDefinedTypes,
+	IsPreferred:         false,
+	IsDefined:           true,
+	Delimiter:           ",",
+	RelID:               id.Null,
+	SubscriptFunc:       toFuncID("jsonb_subscript_handler", toInternal("internal")),
+	Elem:                internalNullType,
+	Array:               internalNullType,
+	InputFunc:           toFuncID("jsonb_in", toInternal("cstring")),
+	OutputFunc:          toFuncID("jsonb_out", toInternal("jsonb")),
+	ReceiveFunc:         toFuncID("jsonb_recv", toInternal("internal")),
+	SendFunc:            toFuncID("jsonb_send", toInternal("jsonb")),
+	ModInFunc:           toFuncID("-"),
+	ModOutFunc:          toFuncID("-"),
+	AnalyzeFunc:         toFuncID("-"),
+	Align:               TypeAlignment_Int,
+	Storage:             TypeStorage_Extended,
+	NotNull:             false,
+	BaseTypeType:        internalNullType,
+	TypMod:              -1,
+	NDims:               0,
+	TypCollation:        id.NullCollation,
+	DefaulBin:           "",
+	Default:             "",
+	Acl:                 nil,
+	Checks:              nil,
+	attTypMod:           -1,
+	CompareFunc:         toFuncID("jsonb_cmp", toInternal("jsonb"), toInternal("jsonb")),
+	SerializationFunc:   serializeTypeJsonB,
+	DeserializationFunc: deserializeTypeJsonB,
+}
+
+// serializeTypeJsonB handles serialization from the standard representation to our serialized representation that is
+// written in Dolt. This is used for the legacy ExtendedEnc storage path.
+// Deprecated. These values are now serialized and deserialized by Dolt natively.
+func serializeTypeJsonB(ctx *sql.Context, t *DoltgresType, val any) ([]byte, error) {
+	res, err := sql.UnwrapAny(ctx, val)
+	if err != nil {
+		return nil, err
+	}
+
+	var doc JsonDocument
+	switch v := res.(type) {
+	case sql.JSONWrapper:
+		j, err := v.ToInterface(ctx)
+		if err != nil {
+			return nil, err
+		}
+		jsonVal, err := ConvertToJsonDocument(j)
+		if err != nil {
+			return nil, err
+		}
+		doc = JsonDocument{Value: jsonVal}
+	default:
+		return nil, errors.Newf("jsonb: unexpected types %T, %T", res, val)
+	}
+
+	writer := utils.NewWriter(256)
+	JsonValueSerialize(writer, doc.Value)
+	return writer.Data(), nil
+}
+
+// deserializeTypeJsonB handles deserialization from the Dolt serialized format to our standard representation used by
+// expressions and nodes. This is used for the legacy ExtendedEnc storage path.
+// Deprecated. These values are now serialized and deserialized by Dolt natively, but previous releases still write
+// values in this old format.
+func deserializeTypeJsonB(ctx *sql.Context, t *DoltgresType, data []byte) (any, error) {
+	if len(data) == 0 {
+		return nil, nil
+	}
+	reader := utils.NewReader(data)
+	jsonValue, err := JsonValueDeserialize(reader)
+	if err != nil {
+		return nil, err
+	}
+	return gmstypes.JSONDocument{Val: jsonValueToInterface(jsonValue)}, nil
+}
+
+// jsonValueToInterface converts a legacy JsonValue to a native Go interface value.
+func jsonValueToInterface(value JsonValue) any {
+	switch v := value.(type) {
+	case JsonValueObject:
+		obj := make(map[string]any, len(v.Items))
+		for _, item := range v.Items {
+			obj[item.Key] = jsonValueToInterface(item.Value)
+		}
+		return obj
+	case JsonValueArray:
+		arr := make([]any, len(v))
+		for i, item := range v {
+			arr[i] = jsonValueToInterface(item)
+		}
+		return arr
+	case JsonValueString:
+		return string(v)
+	case JsonValueNumber:
+		d := apd.Decimal(v)
+		return &d
+	case JsonValueBoolean:
+		return bool(v)
+	case JsonValueNull:
+		return nil
+	default:
+		return nil
+	}
+}