Peewee+Postgresql+PooledPostgresqlDatabase重连机制

本文主要是介绍Peewee+Postgresql+PooledPostgresqlDatabase重连机制，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

需求：

Postgresql数据库服务重启后，需要业务代码正常读写数据库

方案：

1. 通过继承playhouse.shortcuts.ReconnectMixin和playhouse.pool.PooledPostgresqlDatabase来创建一个新的ReconnectPooledPostgresqlDatabase类
2. 修改reconnect_errors属性来适配Postgresql的错误类型
3. 使用ReconnectPooledPostgresqlDatabase来获取数据库连接

测试

1. 启动程序 -->重起数据库服务–>读写数据库操作正常
2. 启动应用程序–>关闭数据库服务–>读写数据库失败–>启动数据库服务–>读写数据库操作正常

示例：

from playhouse.pool import PooledPostgresqlDatabase
from peewee import OperationalError, InterfaceError
from playhouse.shortcuts import ReconnectMixinclass ReconnectPooledPostgresqlDatabase(ReconnectMixin, PooledPostgresqlDatabase):"""支持重连机制的数据库连接池类，可以通过扩展reconnect_errors来支持重连场景"""reconnect_errors = (# Postgres error examples:(OperationalError, 'terminat'),(InterfaceError, 'connection already closed'))class PostgresqlManager(object):"""PG数据库管理类"""__instance_lock = threading.Lock()__database = None@classmethoddef get_database(cls, **kwargs):if cls.__database is None:with cls.__instance_lock:cls.__database = PooledPostgresqlDatabase(database=kwargs.get("database"),host=kwargs.get("host", "127.0.0.1"),port=int(kwargs.get("port", 5432)),user=kwargs.get("user", "postgres"),password=kwargs.get("password", "postgres"),max_connections=int(kwargs.get("max_connections", 50)),stale_timeout=int(kwargs.get("stale_timeout", 600)))return cls.__database

这篇关于Peewee+Postgresql+PooledPostgresqlDatabase重连机制的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---