Function array_recv_callable

server/functions/array.go:179–222

array_recv_callable is the function definition of array_recv.

(ctx *sql.Context, t [4]*pgtypes.DoltgresType, val1, val2, val3 any)

// array_recv_callable is the function definition of array_recv.
func array_recv_callable(ctx *sql.Context, t [4]*pgtypes.DoltgresType, val1, val2, val3 any) (any, error) {
	data := val1.([]byte)
	if data == nil {
		return nil, nil
	}
	typeColl, err := core.GetTypesCollectionFromContext(ctx, "")
	if err != nil {
		return nil, err
	}
	reader := utils.NewWireReader(data)
	dimensions := reader.ReadInt32()
	_ = reader.ReadInt32() // Whether the array has a null, doesn't seem useful
	baseTypeID := id.Type(id.Cache().ToInternal(reader.ReadUint32()))
	baseType, err := typeColl.GetType(ctx, baseTypeID)
	if err != nil {
		return nil, err
	}
	if baseType == nil {
		return nil, pgtypes.ErrTypeDoesNotExist.New(baseTypeID.TypeName())
	}
	// TODO: handle more than 1 dimension
	if dimensions > 1 {
		return nil, errors.Errorf("array dimensions greater than 1 are not yet supported")
	}
	var vals []any
	for dimensionIdx := int32(0); dimensionIdx < dimensions; dimensionIdx++ {
		elementsCount := reader.ReadInt32()
		_ = reader.ReadInt32() // Lower bound, not sure what to do with this
		for i := int32(0); i < elementsCount; i++ {
			elementLen := reader.ReadInt32()
			if elementLen != -1 {
				valBytes := reader.ReadBytes(uint32(elementLen))
				val, err := baseType.CallReceive(ctx, valBytes)
				if err != nil {
					return nil, err
				}
				vals = append(vals, val)
			} else {
				vals = append(vals, nil)
			}
		}
	}
	return vals, nil
}

// array_send represents the PostgreSQL function of array type IO send.
var array_send = framework.Function1{