fix: add support for newer state impacting command

Add support for newer command which impact the state of a client
connection hence need special handling when returning connections to the
pool.

Also add missing type assertions for ConnWithContext on connection types.