mirrors/haproxy
1
0
Fork 0
mirror of https://git.haproxy.org/git/haproxy.git/ synced 2025-09-22 06:11:32 +02:00
Code Issues Projects Releases Wiki Activity

MINOR: queue: split __pendconn_unlink() in per-srv and per-prx

Browse Source 
The function is called with the lock held and does too many tests for
things that are already known from its callers. Let's split it in two
so that its callers call either the per-server or per-proxy function
depending on where the element is (since they had to determine it
prior to taking the lock).
This commit is contained in:
Willy Tarreau 2020-10-21 11:20:07 +02:00
parent b7ba1d9011
commit 3e3ae2524d
1 changed files with 46 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
src/queue.c
View File

@ -123,22 +123,32 @@ unsigned int srv_dynamic_maxconn(const struct server *s)
return max; return max;
} }
/* Remove the pendconn from the server/proxy queue. At this stage, the /* Remove the pendconn from the server's queue. At this stage, the connection
* connection is not really dequeued. It will be done during the * is not really dequeued. It will be done during the process_stream. It also
* process_stream. It also decreases the pending count. * decrements the pending count.
* *
* The caller must own the lock on the queue containing the pendconn. The * The caller must own the lock on the server queue. The pendconn must still be
* pendconn must still be queued. * queued (p->node.leaf_p != NULL) and must be in a server (p->srv != NULL).
*/ */
static void __pendconn_unlink(struct pendconn *p) static void __pendconn_unlink_srv(struct pendconn *p)
{ {
if (p->srv) {
p->strm->logs.srv_queue_pos += p->srv->queue_idx - p->queue_idx; p->strm->logs.srv_queue_pos += p->srv->queue_idx - p->queue_idx;
p->srv->nbpend--; p->srv->nbpend--;
} else { _HA_ATOMIC_SUB(&p->px->totpend, 1);
eb32_delete(&p->node);
}
/* Remove the pendconn from the proxy's queue. At this stage, the connection
* is not really dequeued. It will be done during the process_stream. It also
* decrements the pending count.
*
* The caller must own the lock on the proxy queue. The pendconn must still be
* queued (p->node.leaf_p != NULL) and must be in the proxy (p->srv == NULL).
*/
static void __pendconn_unlink_prx(struct pendconn *p)
{
p->strm->logs.prx_queue_pos += p->px->queue_idx - p->queue_idx; p->strm->logs.prx_queue_pos += p->px->queue_idx - p->queue_idx;
p->px->nbpend--; p->px->nbpend--;
}
_HA_ATOMIC_SUB(&p->px->totpend, 1); _HA_ATOMIC_SUB(&p->px->totpend, 1);
eb32_delete(&p->node); eb32_delete(&p->node);
} }
@ -172,16 +182,24 @@ static inline void pendconn_queue_unlock(struct pendconn *p)
* This function takes all the required locks for the operation. The pendconn * This function takes all the required locks for the operation. The pendconn
* must be valid, though it doesn't matter if it was already unlinked. Prefer * must be valid, though it doesn't matter if it was already unlinked. Prefer
* pendconn_cond_unlink() to first check <p>. When the locks are already held, * pendconn_cond_unlink() to first check <p>. When the locks are already held,
* please use __pendconn_unlink() instead. * please use __pendconn_unlink_{srv,prx}() instead.
*/ */
void pendconn_unlink(struct pendconn *p) void pendconn_unlink(struct pendconn *p)
{ {
pendconn_queue_lock(p); if (p->srv) {
/* queued in the server */
HA_SPIN_LOCK(SERVER_LOCK, &p->srv->lock);
if (p->node.node.leaf_p) if (p->node.node.leaf_p)
__pendconn_unlink(p); __pendconn_unlink_srv(p);
HA_SPIN_UNLOCK(SERVER_LOCK, &p->srv->lock);
pendconn_queue_unlock(p); }
else {
/* queued in the proxy */
HA_RWLOCK_WRLOCK(PROXY_LOCK, &p->px->lock);
if (p->node.node.leaf_p)
__pendconn_unlink_prx(p);
HA_RWLOCK_WRUNLOCK(PROXY_LOCK, &p->px->lock);
}
} }
/* Retrieve the first pendconn from tree <pendconns>. Classes are always /* Retrieve the first pendconn from tree <pendconns>. Classes are always
@ -280,17 +298,17 @@ static int pendconn_process_next_strm(struct server *srv, struct proxy *px)
use_pp: use_pp:
/* Let's switch from the server pendconn to the proxy pendconn */ /* Let's switch from the server pendconn to the proxy pendconn */
__pendconn_unlink_prx(pp);
px->queue_idx++;
p = pp; p = pp;
goto unlinked;
use_p: use_p:
__pendconn_unlink(p); __pendconn_unlink_srv(p);
srv->queue_idx++;
unlinked:
p->strm_flags |= SF_ASSIGNED; p->strm_flags |= SF_ASSIGNED;
p->target = srv; p->target = srv;
if (p != pp)
srv->queue_idx++;
else
px->queue_idx++;
_HA_ATOMIC_ADD(&srv->served, 1); _HA_ATOMIC_ADD(&srv->served, 1);
_HA_ATOMIC_ADD(&srv->proxy->served, 1); _HA_ATOMIC_ADD(&srv->proxy->served, 1);
__ha_barrier_atomic_store(); __ha_barrier_atomic_store();
@ -412,7 +430,7 @@ int pendconn_redistribute(struct server *s)
continue; continue;
/* it's left to the dispatcher to choose a server */ /* it's left to the dispatcher to choose a server */
__pendconn_unlink(p); __pendconn_unlink_srv(p);
p->strm_flags &= ~(SF_DIRECT | SF_ASSIGNED | SF_ADDR_SET); p->strm_flags &= ~(SF_DIRECT | SF_ASSIGNED | SF_ADDR_SET);
task_wakeup(p->strm->task, TASK_WOKEN_RES); task_wakeup(p->strm->task, TASK_WOKEN_RES);
@ -450,7 +468,7 @@ int pendconn_grab_from_px(struct server *s)
if (s->maxconn && s->served + xferred >= maxconn) if (s->maxconn && s->served + xferred >= maxconn)
break; break;
__pendconn_unlink(p); __pendconn_unlink_prx(p);
p->target = s; p->target = s;
task_wakeup(p->strm->task, TASK_WOKEN_RES); task_wakeup(p->strm->task, TASK_WOKEN_RES);