Richard Yen：调试RDS Proxy锁定：隐藏的JIT切换如何导致数千个锁定连接
内容提要
使用AWS RDS Proxy时，连接复用因会话状态变化被锁定，导致高CPU使用和大量活动连接。分析日志发现，asyncpg在注册JSON编码器时临时禁用JIT，影响RDS Proxy跟踪会话状态。通过添加连接钩子禁用JIT，成功减少锁定会话数量，提升性能。
关键要点
-
使用AWS RDS Proxy时，连接复用因会话状态变化被锁定，导致高CPU使用和大量活动连接。
-
通过分析日志发现，asyncpg在注册JSON编码器时临时禁用JIT，影响RDS Proxy跟踪会话状态。
-
添加连接钩子禁用JIT，成功减少锁定会话数量，提升性能。
-
SQLAlchemy在连接初始化时运行on_connect钩子，注册优化的JSON和JSONB编码器。
-
asyncpg在注册编码器时需要查找类型OID，触发内部函数introspect_types()。
-
asyncpg临时禁用JIT会导致RDS Proxy无法安全跟踪会话状态，从而锁定连接。
-
通过添加SQLAlchemy连接钩子，禁用asyncpg的JIT能力标志，避免会话锁定。
-
部署修复后，锁定会话数量显著减少，改善超过50%。
-
尝试其他修复方法未成功，最终解决方案是添加防止锁定的钩子。
-
调试经验教训包括：RDS Proxy锁定来源意外，小会话级变化也会影响复用，pg_stat_statements隐藏参数值。
延伸解读
RDS Proxy的连接复用挑战
在使用AWS RDS Proxy时，连接复用的效率受到会话状态变化的影响。文章指出，asyncpg在注册JSON编码器时临时禁用JIT，导致RDS Proxy无法安全跟踪会话状态，从而锁定连接。这一问题不仅影响性能，还可能导致高CPU使用和大量活动连接，开发者需对此保持警惕。
调试经验教训
调试过程中，作者总结了几个重要经验：RDS Proxy的锁定问题可能源于意想不到的地方，小的会话级变化也能影响连接复用。此外，pg_stat_statements无法显示参数值，可能隐藏关键线索。临时启用日志记录可以快速发现问题，建议开发者在调试时考虑这些因素。
解决方案的有效性
通过添加SQLAlchemy连接钩子来禁用asyncpg的JIT能力标志，成功减少了锁定会话数量，提升了性能超过50%。这一解决方案表明，针对特定问题的细致调试和代码调整可以显著改善系统表现，开发者在面对类似问题时应考虑这种方法。
延伸问答
使用AWS RDS Proxy时，连接复用被锁定的原因是什么？
连接复用被锁定是因为会话状态发生变化，RDS Proxy无法安全跟踪这些变化，从而将连接固定在特定的后端连接上。
如何通过SQLAlchemy解决RDS Proxy的连接锁定问题？
通过添加SQLAlchemy连接钩子，禁用asyncpg的JIT能力标志，避免会话锁定，从而解决问题。
asyncpg在注册JSON编码器时会发生什么？
asyncpg在注册JSON编码器时会查找类型OID，触发内部函数introspect_types()，并可能临时禁用JIT。
部署修复后，RDS Proxy的锁定会话数量有何变化？
部署修复后，锁定会话数量显著减少，改善超过50%。
在调试RDS Proxy时，pg_stat_statements的局限性是什么？
pg_stat_statements隐藏参数值，无法暴露绑定参数，这可能会掩盖关键线索。
RDS Proxy锁定会话的调试经验教训有哪些？
调试经验包括：锁定来源意外，小会话级变化影响复用，pg_stat_statements隐藏参数值。
