diff --git a/src/backend/storage/ipc/shm_mq.c b/src/backend/storage/ipc/shm_mq.c index a49bd17feeb09403c368fcea765e4dcf9f66ee3f..bc735f3b5e896529ec4c02307652f0af1bfbb170 100644 --- a/src/backend/storage/ipc/shm_mq.c +++ b/src/backend/storage/ipc/shm_mq.c @@ -681,33 +681,37 @@ shm_mq_send_bytes(shm_mq_handle *mqh, Size nbytes, void *data, bool nowait, return SHM_MQ_DETACHED; } - if (available == 0) + if (available == 0 && !mqh->mqh_counterparty_attached) { - shm_mq_result res; - /* * The queue is full, so if the receiver isn't yet known to be * attached, we must wait for that to happen. */ - if (!mqh->mqh_counterparty_attached) + if (nowait) { - if (nowait) + if (shm_mq_get_receiver(mq) == NULL) { - if (shm_mq_get_receiver(mq) == NULL) - { - *bytes_written = sent; - return SHM_MQ_WOULD_BLOCK; - } - } - else if (!shm_mq_wait_internal(mq, &mq->mq_receiver, - mqh->mqh_handle)) - { - mq->mq_detached = true; *bytes_written = sent; - return SHM_MQ_DETACHED; + return SHM_MQ_WOULD_BLOCK; } - mqh->mqh_counterparty_attached = true; } + else if (!shm_mq_wait_internal(mq, &mq->mq_receiver, + mqh->mqh_handle)) + { + mq->mq_detached = true; + *bytes_written = sent; + return SHM_MQ_DETACHED; + } + mqh->mqh_counterparty_attached = true; + + /* + * The receiver may have read some data after attaching, so we + * must not wait without rechecking the queue state. + */ + } + else if (available == 0) + { + shm_mq_result res; /* Let the receiver know that we need them to read some data. */ res = shm_mq_notify_receiver(mq);