mirror of
https://gitlab.nic.cz/labs/bird.git
synced 2024-12-22 17:51:53 +00:00
First try of loop balancing
If a thread encounters timeout == 0 for poll, it considers itself "busy" and with some hysteresis it tries to drop loops for others to pick and thus better distribute work between threads.
This commit is contained in:
Maria Matejka
parent
9f25dd79b8
commit
fa973c2c15
@ -406,7 +406,7 @@ socket_changed(sock *s)
|
||||
struct birdloop *loop = s->loop;
|
||||
ASSERT_DIE(birdloop_inside(loop));
|
||||
|
||||
loop->sock_changed++;
|
||||
loop->sock_changed = 1;
|
||||
birdloop_ping(loop);
|
||||
}
|
||||
|
||||
@ -552,7 +552,7 @@ sockets_fire(struct birdloop *loop)
|
||||
continue;
|
||||
|
||||
if (!sk_tx_pending(s))
|
||||
loop->thread->sock_changed++;
|
||||
loop->thread->sock_changed = 1;
|
||||
}
|
||||
|
||||
if (rev & POLLIN)
|
||||
@ -590,6 +590,7 @@ struct birdloop_pickup_group {
|
||||
list threads;
|
||||
uint thread_count;
|
||||
uint loop_count;
|
||||
uint thread_busy_count;
|
||||
btime max_latency;
|
||||
event start_threads;
|
||||
} pickup_groups[2] = {
|
||||
@ -624,42 +625,40 @@ birdloop_set_thread(struct birdloop *loop, struct bird_thread *thr, struct birdl
|
||||
}
|
||||
/* Now we are free of running pings */
|
||||
|
||||
if (loop->thread = thr)
|
||||
/* Update the thread value */
|
||||
loop->thread = thr;
|
||||
|
||||
/* Put into appropriate lists */
|
||||
if (thr)
|
||||
{
|
||||
add_tail(&thr->loops, &loop->n);
|
||||
thr->loop_count++;
|
||||
ev_send_loop(loop->thread->meta, &loop->event);
|
||||
}
|
||||
else
|
||||
{
|
||||
old->loop_count--;
|
||||
|
||||
/* Unschedule from Meta */
|
||||
ev_postpone(&loop->event);
|
||||
tm_stop(&loop->timer);
|
||||
|
||||
/* Request local socket reload */
|
||||
this_thread->sock_changed = 1;
|
||||
|
||||
/* Put into pickup list */
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
add_tail(&group->loops, &loop->n);
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
}
|
||||
|
||||
/* Finished */
|
||||
/* Allow pings */
|
||||
atomic_fetch_and_explicit(&loop->thread_transition, ~LTT_MOVE, memory_order_acq_rel);
|
||||
|
||||
/* Request to run by force */
|
||||
ev_send_loop(loop->thread->meta, &loop->event);
|
||||
}
|
||||
|
||||
static struct birdloop *
|
||||
birdloop_take(struct birdloop_pickup_group *group)
|
||||
static void
|
||||
bird_thread_pickup_next(struct birdloop_pickup_group *group)
|
||||
{
|
||||
struct birdloop *loop = NULL;
|
||||
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
if (!EMPTY_LIST(group->loops))
|
||||
{
|
||||
/* Take the first loop from the pickup list and unlock */
|
||||
loop = SKIP_BACK(struct birdloop, n, HEAD(group->loops));
|
||||
rem_node(&loop->n);
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
|
||||
birdloop_set_thread(loop, this_thread, group);
|
||||
|
||||
/* This thread goes to the end of the pickup list */
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
rem_node(&this_thread->n);
|
||||
@ -668,32 +667,56 @@ birdloop_take(struct birdloop_pickup_group *group)
|
||||
/* If there are more loops to be picked up, wakeup the next thread in order */
|
||||
if (!EMPTY_LIST(group->loops))
|
||||
wakeup_do_kick(SKIP_BACK(struct bird_thread, n, HEAD(group->threads)));
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
}
|
||||
|
||||
static struct birdloop *
|
||||
birdloop_take(struct birdloop_pickup_group *group)
|
||||
{
|
||||
struct birdloop *loop = NULL;
|
||||
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
|
||||
int drop =
|
||||
this_thread->busy_active &&
|
||||
(group->thread_busy_count < group->thread_count) &&
|
||||
(this_thread->loop_count > 1);
|
||||
int take = !EMPTY_LIST(group->loops);
|
||||
|
||||
if (drop)
|
||||
{
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
node *n;
|
||||
WALK_LIST2(loop, n, this_thread->loops, n)
|
||||
if (ev_active(&loop->event))
|
||||
{
|
||||
LOOP_TRACE(loop, "Moving to another thread");
|
||||
|
||||
/* Pass to another thread */
|
||||
rem_node(&loop->n);
|
||||
birdloop_set_thread(loop, NULL, group);
|
||||
bird_thread_pickup_next(group);
|
||||
break;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
else if (take)
|
||||
{
|
||||
/* Take the first loop from the pickup list and unlock */
|
||||
loop = SKIP_BACK(struct birdloop, n, HEAD(group->loops));
|
||||
rem_node(&loop->n);
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
|
||||
birdloop_set_thread(loop, this_thread, group);
|
||||
bird_thread_pickup_next(group);
|
||||
return loop;
|
||||
}
|
||||
|
||||
static void
|
||||
birdloop_drop(struct birdloop *loop, struct birdloop_pickup_group *group)
|
||||
{
|
||||
/* Remove loop from this thread's list */
|
||||
rem_node(&loop->n);
|
||||
|
||||
/* Unset loop's thread */
|
||||
if (birdloop_inside(loop))
|
||||
birdloop_set_thread(loop, NULL, group);
|
||||
else
|
||||
{
|
||||
birdloop_enter(loop);
|
||||
birdloop_set_thread(loop, NULL, group);
|
||||
birdloop_leave(loop);
|
||||
}
|
||||
|
||||
/* Put loop into pickup list */
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
add_tail(&group->loops, &loop->n);
|
||||
UNLOCK_DOMAIN(resource, group->domain);
|
||||
return NULL;
|
||||
}
|
||||
}
|
||||
|
||||
static int
|
||||
@ -707,6 +730,38 @@ poll_timeout(struct birdloop *loop)
|
||||
return remains TO_MS + ((remains TO_MS) MS < remains);
|
||||
}
|
||||
|
||||
static void
|
||||
bird_thread_busy_update(struct bird_thread *thr, int val)
|
||||
{
|
||||
if (thr->busy_active == val)
|
||||
return;
|
||||
|
||||
if (val)
|
||||
thr->busy_counter++;
|
||||
else
|
||||
thr->busy_counter--;
|
||||
|
||||
switch (thr->busy_counter)
|
||||
{
|
||||
case 0:
|
||||
thr->busy_active = 0;
|
||||
break;
|
||||
case 4:
|
||||
thr->busy_active = 1;
|
||||
break;
|
||||
|
||||
default:
|
||||
return;
|
||||
}
|
||||
|
||||
LOCK_DOMAIN(resource, thr->group->domain);
|
||||
if (val)
|
||||
thr->group->thread_busy_count++;
|
||||
else
|
||||
thr->group->thread_busy_count--;
|
||||
UNLOCK_DOMAIN(resource, thr->group->domain);
|
||||
}
|
||||
|
||||
static void *
|
||||
bird_thread_main(void *arg)
|
||||
{
|
||||
@ -795,6 +850,9 @@ bird_thread_main(void *arg)
|
||||
else if (timeout > 5000)
|
||||
flush_local_pages();
|
||||
|
||||
bird_thread_busy_update(thr, (timeout == 0));
|
||||
|
||||
account_to(&this_thread->idle);
|
||||
poll_retry:;
|
||||
int rv = poll(pfd.pfd.data, pfd.pfd.used, timeout);
|
||||
if (rv < 0)
|
||||
@ -804,6 +862,8 @@ poll_retry:;
|
||||
bug("poll in %p: %m", thr);
|
||||
}
|
||||
|
||||
account_to(&this_thread->overhead);
|
||||
|
||||
/* Drain wakeup fd */
|
||||
if (pfd.pfd.data[0].revents & POLLIN)
|
||||
{
|
||||
@ -934,7 +994,22 @@ bird_thread_shutdown(void * _ UNUSED)
|
||||
|
||||
/* Drop loops including the thread dropper itself */
|
||||
while (!EMPTY_LIST(thr->loops))
|
||||
birdloop_drop(HEAD(thr->loops), group);
|
||||
{
|
||||
struct birdloop *loop = HEAD(thr->loops);
|
||||
|
||||
/* Remove loop from this thread's list */
|
||||
rem_node(&loop->n);
|
||||
|
||||
/* Unset loop's thread */
|
||||
if (birdloop_inside(loop))
|
||||
birdloop_set_thread(loop, NULL, group);
|
||||
else
|
||||
{
|
||||
birdloop_enter(loop);
|
||||
birdloop_set_thread(loop, NULL, group);
|
||||
birdloop_leave(loop);
|
||||
}
|
||||
}
|
||||
|
||||
/* Let others know about new loops */
|
||||
LOCK_DOMAIN(resource, group->domain);
|
||||
@ -1022,11 +1097,12 @@ struct bird_thread_show_data {
|
||||
u8 show_loops;
|
||||
uint line_pos;
|
||||
uint line_max;
|
||||
const char *lines[];
|
||||
const char **lines;
|
||||
};
|
||||
|
||||
#define tsd_append(...) do { \
|
||||
ASSERT_DIE(tsd->line_pos < tsd->line_max); \
|
||||
if (tsd->line_pos >= tsd->line_max) \
|
||||
tsd->lines = mb_realloc(tsd->lines, sizeof (const char *) * (tsd->line_max *= 2)); \
|
||||
tsd->lines[tsd->line_pos++] = lp_sprintf(tsd->lp, __VA_ARGS__); \
|
||||
} while (0)
|
||||
|
||||
@ -1090,6 +1166,7 @@ bird_thread_show(void *data)
|
||||
{
|
||||
tsd_append(" Total working time: %t", total_time_ns NS);
|
||||
bird_thread_show_spent_time(tsd, "Overhead", &this_thread->overhead);
|
||||
bird_thread_show_spent_time(tsd, "Idle ", &this_thread->idle);
|
||||
}
|
||||
else
|
||||
tsd_append("Thread %p working %t s overhead %t s",
|
||||
@ -1175,9 +1252,8 @@ cmd_show_threads(int show_loops)
|
||||
}
|
||||
|
||||
/* Total number of lines must be recalculated when changing the code! */
|
||||
uint total_lines = total_threads * (show_loops + 1) + total_loops * 3 * show_loops + 10;
|
||||
|
||||
struct bird_thread_show_data *tsd = mb_allocz(p, sizeof(struct bird_thread_show_data) + total_lines * sizeof(const char *));
|
||||
struct bird_thread_show_data *tsd = mb_allocz(p, sizeof(struct bird_thread_show_data));
|
||||
tsd->cli = this_cli;
|
||||
tsd->pool = p;
|
||||
tsd->lp = lp_new(p);
|
||||
@ -1188,7 +1264,8 @@ cmd_show_threads(int show_loops)
|
||||
.data = tsd,
|
||||
};
|
||||
tsd->line_pos = 0;
|
||||
tsd->line_max = total_lines;
|
||||
tsd->line_max = 64;
|
||||
tsd->lines = mb_allocz(p, sizeof(const char *) * tsd->line_max);
|
||||
|
||||
this_cli->cont = bird_thread_show_cli_cont;
|
||||
this_cli->cleanup = bird_thread_show_cli_cleanup;
|
||||
@ -1294,7 +1371,7 @@ birdloop_stop_internal(struct birdloop *loop)
|
||||
birdloop_leave(loop);
|
||||
|
||||
/* Request local socket reload */
|
||||
this_thread->sock_changed++;
|
||||
this_thread->sock_changed = 1;
|
||||
|
||||
/* Call the stopped hook from the main loop */
|
||||
loop->event.hook = loop->stopped;
|
||||
@ -1353,7 +1430,7 @@ birdloop_run(void *_loop)
|
||||
ev_send_loop(this_thread->meta, &loop->event);
|
||||
|
||||
/* Collect socket change requests */
|
||||
this_thread->sock_changed += loop->sock_changed;
|
||||
this_thread->sock_changed |= loop->sock_changed;
|
||||
loop->sock_changed = 0;
|
||||
|
||||
account_to(&this_thread->overhead);
|
||||
|
||||
@ -51,7 +51,7 @@ struct birdloop
|
||||
list sock_list;
|
||||
struct birdsock *sock_active;
|
||||
int sock_num;
|
||||
uint sock_changed;
|
||||
uint sock_changed:1;
|
||||
|
||||
uint ping_pending;
|
||||
|
||||
@ -93,13 +93,15 @@ struct bird_thread
|
||||
|
||||
event cleanup_event;
|
||||
|
||||
int sock_changed;
|
||||
u8 sock_changed;
|
||||
u8 busy_active;
|
||||
u16 busy_counter;
|
||||
uint loop_count;
|
||||
|
||||
u64 max_latency_ns;
|
||||
u64 max_loop_time_ns;
|
||||
|
||||
struct spent_time overhead;
|
||||
struct spent_time overhead, idle;
|
||||
};
|
||||
|
||||
#endif
|
||||
|
||||
Loading…
Reference in New Issue
Block a user