allow setting statement name in set_type_codec

This is necessary when using PgBouncer because otherwise the default simplistically-generated statement names can fight with other programs using asyncpg due to PgBouncer sharing the same underlying PostgreSQL connection and thus the prepared statement names. [Supposedly asyncpg can be made to use an unnamed statement here by turning the cache off][0], but I can't for the life of me figure out how to do that when using asyncpg through SQLAlchemy; I observe that setting `prepared_statement_cache_size` to `0` as described [here][1] doesn't cause `use_cache` to be `False`. Anyway, that's not really asyncpg's problem. I think this functionality should probably be provided either way since it'd be kinda weird to only allow configuring the names of *some* prepared statements. [0]: https://github.com/MagicStack/asyncpg/blob/1aab2094d82104d5eee2cffcfd0c7e7347d4c5b8/asyncpg/connection.py#L415-L424 [1]: https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#prepared-statement-cache

Author: CobaltCause

asyncpg/connection.py

@@ -534,7 +534,7 @@ async def _introspect_types(self, typeoids, timeout):
 
  return result
 
- async def _introspect_type(self, typename, schema):
+ async def _introspect_type(self, typename, schema, *, name=None):
  if (
  schema == 'pg_catalog'
  and typename.lower() in protocol.BUILTIN_TYPE_NAME_MAP
@@ -546,6 +546,7 @@ async def _introspect_type(self, typename, schema):
  limit=0,
  timeout=None,
  ignore_custom_codec=True,
+ name=name,
  )
  else:
  rows = await self._execute(
@@ -554,6 +555,7 @@ async def _introspect_type(self, typename, schema):
  limit=1,
  timeout=None,
  ignore_custom_codec=True,
+ name=name,
  )
 
  if not rows:
@@ -1215,7 +1217,7 @@ async def __anext__(self):
 
  async def set_type_codec(self, typename, *,
  schema='public', encoder, decoder,
- format='text'):
+ format='text', name=None):
  """Set an encoder/decoder pair for the specified data type.
 
  :param typename:
@@ -1336,7 +1338,7 @@ async def set_type_codec(self, typename, *,
  """
  self._check_open()
  settings = self._protocol.get_settings()
- typeinfo = await self._introspect_type(typename, schema)
+ typeinfo = await self._introspect_type(typename, schema, name)
  full_typeinfos = []
  if introspection.is_scalar_type(typeinfo):
  kind = 'scalar'
@@ -1788,7 +1790,8 @@ async def _execute(
  *,
  return_status=False,
  ignore_custom_codec=False,
- record_class=None
+ record_class=None,
+ name=None,
  ):
  with self._stmt_exclusive_section:
  result, _ = await self.__execute(
@@ -1799,6 +1802,7 @@ async def _execute(
  return_status=return_status,
  record_class=record_class,
  ignore_custom_codec=ignore_custom_codec,
+ name=name,
  )
  return result
 
@@ -1868,7 +1872,8 @@ async def __execute(
  *,
  return_status=False,
  ignore_custom_codec=False,
- record_class=None
+ record_class=None,
+ name=None,
  ):
  executor = lambda stmt, timeout: self._protocol.bind_execute(
  state=stmt,
@@ -1887,6 +1892,7 @@ async def __execute(
  timeout,
  record_class=record_class,
  ignore_custom_codec=ignore_custom_codec,
+ name=name,
  )
  else:
  result, stmt = await self._do_execute(
@@ -1895,6 +1901,7 @@ async def __execute(
  timeout,
  record_class=record_class,
  ignore_custom_codec=ignore_custom_codec,
+ name=name,
  )
  return result, stmt
 
@@ -1919,14 +1926,16 @@ async def _do_execute(
  retry=True,
  *,
  ignore_custom_codec=False,
- record_class=None
+ record_class=None,
+ name=None,
  ):
  if timeout is None:
  stmt = await self._get_statement(
  query,
  None,
  record_class=record_class,
  ignore_custom_codec=ignore_custom_codec,
+ named=True if name is None else name,
  )
  else:
  before = time.monotonic()
@@ -1935,6 +1944,7 @@ async def _do_execute(
  timeout,
  record_class=record_class,
  ignore_custom_codec=ignore_custom_codec,
+ named=True if name is None else name,
  )
  after = time.monotonic()
  timeout -= after - before