使用psycopg2的两阶段提交示例-改进版

以上是基于上一篇 CSDN 文章的改写内容。 

g_xid = self.pg_master_conn.xid(1, 'gtrid' ,'bqual')
for conn_csr in self.node_conn_csr.values():
	l_conn = conn_csr[0]
	l_csr = conn_csr[1]
	l_conn.tpc_begin(g_xid)
	pq_file = open(pq_code_csv_file, 'r')
	l_csr.copy_expert(copy_sql, pq_file)
	if pq_file:
		pq_file.close()
logger.info("=" * 20 + "tpc_prepare" + '=' * 20)

# tpc_prepare 1
try:
"""
	在预提交阶段的失败都可以回滚一致,除非Ctrl+C/kill刚好在这时发生
"""
for conn_csr in self.node_conn_csr.values():
	l_conn = conn_csr[0]
	l_conn.tpc_prepare()
	logger.info("tpc_prepare ok {}".format(l_conn))

except Exception as oe:
for conn_csr in self.node_conn_csr.values():
	try:
		l_conn = conn_csr[0]
		l_conn.tpc_rollback()
	except Exception as oe2:
		logger.info(oe2.message + str(l_conn))
raise customException('tpc_prepare: ' + oe.message, err_code)

# tpc_commit 2
"""
在提交阶段的失败不可以回滚，需要在下一次继续提交修复
# 如果2pc在commit提交阶段出现个别失败的情况，在失败的节点上会遗留预提交的事务，
# 这里的处理是，跳过失败的节点，继续执行其它节点，失败的节点在下次执行时重新提交该事务。
# 可以查询： select * from pg_prepared_xact();
# 在下次执行时，需要检查是否存在之前的遗留事物，有则重新提交。
# 事物ID与g_xid有关
"""
#标记tpc_commit是否失败
tpc_commit_fail = False
logger.info("=" * 20 + "tpc_commit" + '=' * 20)
for conn_csr in self.node_conn_csr.values():
l_conn = conn_csr[0]
try:
	l_conn.tpc_commit()
except Exception as ce:
	logger.info("tpc_commit 提交失败,继续" + str(l_conn))
	tpc_commit_fail = True
else:
	logger.info("tpc_commit ok {}".format(l_conn))
# import time
# time.sleep(10)
if tpc_commit_fail:
# 如果失败，抛出异常
raise customException('tpc_commit 阶段有节点失败!',err_code)