PG-1349 Fix deadlock
Artem Gavrilov
parent
fd43b75153
commit
ec1d449dc5
|
|
@ -1968,120 +1968,142 @@ pgsm_store(pgsmEntry *entry)
|
||||||
* we need to create the entry.
|
* we need to create the entry.
|
||||||
*/
|
*/
|
||||||
LWLockAcquire(pgsm->lock, LW_SHARED);
|
LWLockAcquire(pgsm->lock, LW_SHARED);
|
||||||
shared_hash_entry = (pgsmEntry *) pgsm_hash_find(get_pgsmHash(), &entry->key, &found);
|
|
||||||
|
|
||||||
if (!shared_hash_entry)
|
PG_TRY();
|
||||||
{
|
{
|
||||||
dsa_pointer dsa_query_pointer;
|
|
||||||
dsa_area *query_dsa_area;
|
|
||||||
char *query_buff;
|
|
||||||
|
|
||||||
/* New query, truncate length if necessary. */
|
shared_hash_entry = (pgsmEntry *) pgsm_hash_find(get_pgsmHash(), &entry->key, &found);
|
||||||
if (query_len > pgsm_query_max_len)
|
|
||||||
query_len = pgsm_query_max_len;
|
|
||||||
|
|
||||||
/* Save the query text in raw dsa area */
|
if (!shared_hash_entry)
|
||||||
query_dsa_area = get_dsa_area_for_query_text();
|
|
||||||
dsa_query_pointer = dsa_allocate_extended(query_dsa_area, query_len + 1, DSA_ALLOC_NO_OOM | DSA_ALLOC_ZERO);
|
|
||||||
if (!DsaPointerIsValid(dsa_query_pointer))
|
|
||||||
{
|
{
|
||||||
LWLockRelease(pgsm->lock);
|
dsa_pointer dsa_query_pointer;
|
||||||
return;
|
dsa_area *query_dsa_area;
|
||||||
}
|
char *query_buff;
|
||||||
|
|
||||||
/*
|
/* New query, truncate length if necessary. */
|
||||||
* Get the memory address from DSA pointer and copy the query text in
|
if (query_len > pgsm_query_max_len)
|
||||||
* local variable
|
query_len = pgsm_query_max_len;
|
||||||
*/
|
|
||||||
query_buff = dsa_get_address(query_dsa_area, dsa_query_pointer);
|
|
||||||
memcpy(query_buff, query, query_len);
|
|
||||||
|
|
||||||
LWLockRelease(pgsm->lock);
|
/* Save the query text in raw dsa area */
|
||||||
LWLockAcquire(pgsm->lock, LW_EXCLUSIVE);
|
query_dsa_area = get_dsa_area_for_query_text();
|
||||||
|
dsa_query_pointer = dsa_allocate_extended(query_dsa_area, query_len + 1, DSA_ALLOC_NO_OOM | DSA_ALLOC_ZERO);
|
||||||
/* OK to create a new hashtable entry */
|
if (!DsaPointerIsValid(dsa_query_pointer))
|
||||||
PGSM_DISABLE_ERROR_CAPUTRE();
|
|
||||||
{
|
|
||||||
PG_TRY();
|
|
||||||
{
|
|
||||||
shared_hash_entry = hash_entry_alloc(pgsm, &entry->key, GetDatabaseEncoding());
|
|
||||||
}
|
|
||||||
PG_CATCH();
|
|
||||||
{
|
{
|
||||||
LWLockRelease(pgsm->lock);
|
LWLockRelease(pgsm->lock);
|
||||||
|
return;
|
||||||
if (DsaPointerIsValid(dsa_query_pointer))
|
|
||||||
dsa_free(query_dsa_area, dsa_query_pointer);
|
|
||||||
PG_RE_THROW();
|
|
||||||
}
|
}
|
||||||
PG_END_TRY();
|
|
||||||
} PGSM_END_DISABLE_ERROR_CAPTURE();
|
|
||||||
|
|
||||||
if (shared_hash_entry == NULL)
|
|
||||||
{
|
|
||||||
LWLockRelease(pgsm->lock);
|
|
||||||
|
|
||||||
if (DsaPointerIsValid(dsa_query_pointer))
|
|
||||||
dsa_free(query_dsa_area, dsa_query_pointer);
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Out of memory; report only if the state has changed now.
|
* Get the memory address from DSA pointer and copy the query text
|
||||||
* Otherwise we risk filling up the log file with these message.
|
* in local variable
|
||||||
*/
|
*/
|
||||||
if (!IsSystemOOM())
|
query_buff = dsa_get_address(query_dsa_area, dsa_query_pointer);
|
||||||
{
|
memcpy(query_buff, query, query_len);
|
||||||
pgsm->pgsm_oom = true;
|
|
||||||
|
|
||||||
PGSM_DISABLE_ERROR_CAPUTRE();
|
/* Promotion to exclusive lock */
|
||||||
|
LWLockRelease(pgsm->lock);
|
||||||
|
LWLockAcquire(pgsm->lock, LW_EXCLUSIVE);
|
||||||
|
|
||||||
|
/* OK to create a new hashtable entry */
|
||||||
|
PGSM_DISABLE_ERROR_CAPUTRE();
|
||||||
|
{
|
||||||
|
PG_TRY();
|
||||||
{
|
{
|
||||||
ereport(WARNING,
|
shared_hash_entry = hash_entry_alloc(pgsm, &entry->key, GetDatabaseEncoding());
|
||||||
(errcode(ERRCODE_OUT_OF_MEMORY),
|
}
|
||||||
errmsg("[pg_stat_monitor] pgsm_store: Hash table is out of memory and can no longer store queries!"),
|
PG_CATCH();
|
||||||
errdetail("You may reset the view or when the buckets are deallocated, pg_stat_monitor will resume saving " \
|
{
|
||||||
"queries. Alternatively, try increasing the value of pg_stat_monitor.pgsm_max.")));
|
if (DsaPointerIsValid(dsa_query_pointer))
|
||||||
} PGSM_END_DISABLE_ERROR_CAPTURE();
|
dsa_free(query_dsa_area, dsa_query_pointer);
|
||||||
|
PG_RE_THROW();
|
||||||
|
}
|
||||||
|
PG_END_TRY();
|
||||||
|
} PGSM_END_DISABLE_ERROR_CAPTURE();
|
||||||
|
|
||||||
|
if (shared_hash_entry == NULL)
|
||||||
|
{
|
||||||
|
if (DsaPointerIsValid(dsa_query_pointer))
|
||||||
|
dsa_free(query_dsa_area, dsa_query_pointer);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Out of memory; report only if the state has changed now.
|
||||||
|
* Otherwise we risk filling up the log file with these
|
||||||
|
* message.
|
||||||
|
*/
|
||||||
|
if (!IsSystemOOM())
|
||||||
|
{
|
||||||
|
pgsm->pgsm_oom = true;
|
||||||
|
|
||||||
|
PGSM_DISABLE_ERROR_CAPUTRE();
|
||||||
|
{
|
||||||
|
ereport(WARNING,
|
||||||
|
(errcode(ERRCODE_OUT_OF_MEMORY),
|
||||||
|
errmsg("[pg_stat_monitor] pgsm_store: Hash table is out of memory and can no longer store queries!"),
|
||||||
|
errdetail("You may reset the view or when the buckets are deallocated, pg_stat_monitor will resume saving " \
|
||||||
|
"queries. Alternatively, try increasing the value of pg_stat_monitor.pgsm_max.")));
|
||||||
|
} PGSM_END_DISABLE_ERROR_CAPTURE();
|
||||||
|
}
|
||||||
|
|
||||||
|
LWLockRelease(pgsm->lock);
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
/* If we got a new entry, reset the oom value false */
|
||||||
|
pgsm->pgsm_oom = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
return;
|
/* If we already have the pointer set, free this one */
|
||||||
}
|
if (DsaPointerIsValid(shared_hash_entry->query_text.query_pos))
|
||||||
else
|
dsa_free(query_dsa_area, dsa_query_pointer);
|
||||||
{
|
else
|
||||||
/* If we got a new entry, reset the oom value false */
|
shared_hash_entry->query_text.query_pos = dsa_query_pointer;
|
||||||
pgsm->pgsm_oom = false;
|
|
||||||
|
shared_hash_entry->pgsm_query_id = entry->pgsm_query_id;
|
||||||
|
shared_hash_entry->encoding = entry->encoding;
|
||||||
|
shared_hash_entry->counters.info.cmd_type = entry->counters.info.cmd_type;
|
||||||
|
shared_hash_entry->counters.info.parent_query = InvalidDsaPointer;
|
||||||
|
|
||||||
|
snprintf(shared_hash_entry->datname, sizeof(shared_hash_entry->datname), "%s", entry->datname);
|
||||||
|
snprintf(shared_hash_entry->username, sizeof(shared_hash_entry->username), "%s", entry->username);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* If we already have the pointer set, free this one */
|
pgsm_update_entry(shared_hash_entry, /* entry */
|
||||||
if (DsaPointerIsValid(shared_hash_entry->query_text.query_pos))
|
query, /* query */
|
||||||
dsa_free(query_dsa_area, dsa_query_pointer);
|
comments, /* comments */
|
||||||
else
|
comments_len, /* comments length */
|
||||||
shared_hash_entry->query_text.query_pos = dsa_query_pointer;
|
&entry->counters.planinfo, /* PlanInfo */
|
||||||
|
&entry->counters.sysinfo, /* SysInfo */
|
||||||
|
&entry->counters.error, /* ErrorInfo */
|
||||||
|
entry->counters.plantime.total_time, /* plan_total_time */
|
||||||
|
entry->counters.time.total_time, /* exec_total_time */
|
||||||
|
entry->counters.calls.rows, /* rows */
|
||||||
|
&bufusage, /* bufusage */
|
||||||
|
&walusage, /* walusage */
|
||||||
|
&jitusage, /* jitusage */
|
||||||
|
reset, /* reset */
|
||||||
|
PGSM_STORE);
|
||||||
|
|
||||||
shared_hash_entry->pgsm_query_id = entry->pgsm_query_id;
|
memset(&entry->counters, 0, sizeof(entry->counters));
|
||||||
shared_hash_entry->encoding = entry->encoding;
|
|
||||||
shared_hash_entry->counters.info.cmd_type = entry->counters.info.cmd_type;
|
|
||||||
shared_hash_entry->counters.info.parent_query = InvalidDsaPointer;
|
|
||||||
|
|
||||||
snprintf(shared_hash_entry->datname, sizeof(shared_hash_entry->datname), "%s", entry->datname);
|
|
||||||
snprintf(shared_hash_entry->username, sizeof(shared_hash_entry->username), "%s", entry->username);
|
|
||||||
}
|
}
|
||||||
|
PG_CATCH();
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
* Force LWLock release in case of error because we need to aquire it
|
||||||
|
* again in pgsm_emit_log_hook
|
||||||
|
*
|
||||||
|
* On ereport with level >= ERROR Postgres resets
|
||||||
|
* InterruptHoldoffCount counter (see errfinish funcion in elog.c). We
|
||||||
|
* need to increment it to be able to release lock here.
|
||||||
|
*/
|
||||||
|
HOLD_INTERRUPTS(); /* Increments holdoff counter */
|
||||||
|
LWLockRelease(pgsm->lock); /* Decrements holdoff counter */
|
||||||
|
PG_RE_THROW();
|
||||||
|
}
|
||||||
|
PG_END_TRY();
|
||||||
|
|
||||||
pgsm_update_entry(shared_hash_entry, /* entry */
|
|
||||||
query, /* query */
|
|
||||||
comments, /* comments */
|
|
||||||
comments_len, /* comments length */
|
|
||||||
&entry->counters.planinfo, /* PlanInfo */
|
|
||||||
&entry->counters.sysinfo, /* SysInfo */
|
|
||||||
&entry->counters.error, /* ErrorInfo */
|
|
||||||
entry->counters.plantime.total_time, /* plan_total_time */
|
|
||||||
entry->counters.time.total_time, /* exec_total_time */
|
|
||||||
entry->counters.calls.rows, /* rows */
|
|
||||||
&bufusage, /* bufusage */
|
|
||||||
&walusage, /* walusage */
|
|
||||||
&jitusage, /* jitusage */
|
|
||||||
reset, /* reset */
|
|
||||||
PGSM_STORE);
|
|
||||||
|
|
||||||
memset(&entry->counters, 0, sizeof(entry->counters));
|
|
||||||
LWLockRelease(pgsm->lock);
|
LWLockRelease(pgsm->lock);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue