database/openGauss-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

支持idle_in_transaction_session_timeout

Browse Source
This commit is contained in:
xue_meng_en
2022-08-23 20:53:33 +08:00
parent e94d589c4f
commit 2bcb702760
13 changed files with 234 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/gausskernel/process/tcop/postgres.cpp
View File

@ -723,8 +723,22 @@ static int ReadCommand(StringInfo inBuf)
u_sess->postgres_cxt.doing_extended_query_message = false;
/* Start a timer for session timeout. */
if (!enable_session_sig_alarm(u_sess->attr.attr_common.SessionTimeout * 1000))
if (!enable_session_sig_alarm(u_sess->attr.attr_common.SessionTimeout * 1000)) {
ereport(FATAL, (errcode(ERRCODE_SYSTEM_ERROR), errmsg("could not set timer for session timeout")));
}
#ifndef ENABLE_MULTIPLE_NODES
/* if idle_in_transaction_session_timeout > 0 and it is in a transaction in idle,
* then start a timer for idle_in_transaction_session.
*/
if (u_sess->attr.attr_common.IdleInTransactionSessionTimeout > 0 &&
(IsAbortedTransactionBlockState() || IsTransactionOrTransactionBlock())) {
if (!enable_idle_in_transaction_session_sig_alarm(
u_sess->attr.attr_common.IdleInTransactionSessionTimeout * 1000)) {
ereport(FATAL, (errcode(ERRCODE_SYSTEM_ERROR), errmsg("could not set timer for idle-in-transaction timeout")));
}
}
#endif
if (t_thrd.postgres_cxt.whereToSendOutput == DestRemote)
result = SocketBackend(inBuf);
@ -734,8 +748,18 @@ static int ReadCommand(StringInfo inBuf)
result = EOF;
/* Disable a timer for session timeout. */
if (!disable_session_sig_alarm())
if(!disable_session_sig_alarm()) {
ereport(FATAL, (errcode(ERRCODE_SYSTEM_ERROR), errmsg("could not disable timer for session timeout")));
}
#ifndef ENABLE_MULTIPLE_NODES
/* Disable a timer for idle_in_transaction_session. */
if (u_sess->attr.attr_common.IdleInTransactionSessionTimeout > 0 &&
(IsAbortedTransactionBlockState() || IsTransactionOrTransactionBlock()) &&
!disable_idle_in_transaction_session_sig_alarm()) {
ereport(FATAL, (errcode(ERRCODE_SYSTEM_ERROR), errmsg("could not disable timer for idle-in-transaction timeout")));
}
#endif
u_sess->proc_cxt.firstChar = (char)result;
return result;