Function VitessTypeToObjectID

pgserver/connection_data.go:125–184

VitessTypeToObjectID returns a type, as defined by Vitess, into a type as defined by Postgres. OIDs can be obtained with the following query: `SELECT oid, typname FROM pg_type ORDER BY 1;`

(typ query.Type)

}

type PreparedStatementData struct {
	Statement    ConvertedStatement
	ReturnFields []pgproto3.FieldDescription
	BindVarTypes []uint32
	Stmt         *duckdb.Stmt
	Closed       *atomic.Bool
}

// VitessTypeToObjectID returns a type, as defined by Vitess, into a type as defined by Postgres.
// OIDs can be obtained with the following query: `SELECT oid, typname FROM pg_type ORDER BY 1;`
func VitessTypeToObjectID(typ query.Type) (uint32, error) {
	switch typ {
	case query.Type_INT8:
		// Postgres doesn't make use of a small integer type for integer returns, which presents a bit of a conundrum.
		// GMS defines boolean operations as the smallest integer type, while Postgres has an explicit bool type.
		// We can't always assume that `INT8` means bool, since it could just be a small integer. As a result, we'll
		// always return this as though it's an `INT16`, which also means that we can't support bools right now.
		// OIDs 16 (bool) and 18 (char, ASCII only?) are the only single-byte types as far as I'm aware.
		return uint32(oid.T_int2), nil
	case query.Type_INT16:
		// The technically correct OID is 21 (2-byte integer), however it seems like some clients don't actually expect
		// this, so I'm not sure when it's actually used by Postgres. Because of this, we'll just pretend it's an `INT32`.
		return uint32(oid.T_int2), nil
	case query.Type_INT24:
		// Postgres doesn't have a 3-byte integer type, so just pretend it's `INT32`.
		return uint32(oid.T_int4), nil
	case query.Type_INT32:
		return uint32(oid.T_int4), nil
	case query.Type_INT64:
		return uint32(oid.T_int8), nil
	case query.Type_UINT8:
		return uint32(oid.T_int4), nil
	case query.Type_UINT16:
		return uint32(oid.T_int4), nil
	case query.Type_UINT24:
		return uint32(oid.T_int4), nil
	case query.Type_UINT32:
		// Since this has an upperbound greater than `INT32`, we'll treat it as `INT64`
		return uint32(oid.T_oid), nil
	case query.Type_UINT64:
		// Since this has an upperbound greater than `INT64`, we'll treat it as `NUMERIC`
		return uint32(oid.T_numeric), nil
	case query.Type_FLOAT32:
		return uint32(oid.T_float4), nil
	case query.Type_FLOAT64:
		return uint32(oid.T_float8), nil
	case query.Type_DECIMAL:
		return uint32(oid.T_numeric), nil
	case query.Type_CHAR:
		return uint32(oid.T_char), nil
	case query.Type_VARCHAR:
		return uint32(oid.T_varchar), nil
	case query.Type_TEXT:
		return uint32(oid.T_text), nil
	case query.Type_BLOB:
		return uint32(oid.T_bytea), nil
	case query.Type_JSON:
		return uint32(oid.T_json), nil