# original function to get these records
async def set_array_type_codecs(conn):
"""
Cockroachdb sets array types of native types in the same namespace as pg native type in 'pg_catalog.pg_type'
asyncpg assumes that all new composite types are not going to be in there, so it does not have native codecs for arrays of native types, and goes through a very expensive introspection process to infer the types.

This lists all array types in cockroachdb and creates codecs for them and registers them as well.

Some array types have been commented out since they're not used in the code.

Solution loosely based on the one in here https://github.com/MagicStack/asyncpg/issues/413
"""
start = time()
# got the array types by querying:
# """SELECT *
# FROM pg_catalog.pg_type
# where typcategory = 'A';"""
typenames = [
 "_bit",
 "_bool",
 # '_box2d',
 "_bpchar",
 "_bytea",
 "_char",
 "_date",
 "_float4",
 "_float8",
 # '_geography',
 # '_geometry',
 # '_inet',
 "_int2",
 # '_int2vector',
 "_int4",
 "_int8",
 "_interval",
 # "_jsonb",
 "_name",
 "_numeric",
 # '_oid',
 # '_oidvector',
 # '_record',
 # '_regclass',
 # '_regnamespace',
 # '_regproc',
 # '_regprocedure',
 # '_regrole',
 # '_regtype',
 "_text",
 "_time",
 "_timestamp",
 "_timestamptz",
 "_timetz",
 # '_tsquery',
 # '_tsvector',
 "_uuid",
 # "_varbit",
 "_varchar",
]
schema = "pg_catalog"
conn._check_open()
timeout = 5 # seconds
TYPE_BY_NAMES = f"""\
SELECT
 t.oid,
 t.typelem AS elemtype,
 t.typtype AS kind,
 t.typname as name,
 parent.typname as elemtype_name,
 'pg_catalog' as ns,
 parent.typdelim as elemdelim,
 0 as depth,
 NULL as basetype,
 NULL as range_subtype,
 NULL as attrtypoids,
 NULL as attrnames,
 NULL as basetype_name,
 NULL as range_subtype_name
FROM
 pg_catalog.pg_type AS t
LEFT JOIN
 pg_catalog.pg_type as parent on parent.oid = t.typelem
WHERE
 t.typname in {str(tuple(typenames))}
"""
array_types_records = []
query_args = []
array_type = await conn._execute(
 TYPE_BY_NAMES,
 query_args,
 0,
 timeout,
 ignore_custom_codec=True,
)
array_types_records.extend(array_type)
conn._protocol.get_settings().register_data_types(ARRAY_TYPES)
end = time()
print(f"Finished in {end-start} seconds")
@event.listens_for(self.engine.sync_engine, "connect")
def register_custom_types(dbapi_connection, connection_record):
 dbapi_connection.run_async(
 lambda connection: set_array_type_codecs(connection)
 )