haproxy/include/haproxy/tinfo.h
Willy Tarreau 97da942ba6 MINOR: thread: keep a bitmask of enabled groups in thread_set
We're only checking for 0, 1, or >1 groups enabled there, and we'll soon
need to be more precise and know quickly which groups are non-empty.
Let's just replace the count with a mask of enabled groups. This will
allow to quickly spot the presence of any such group in a set.
2023-04-13 16:57:51 +02:00

121 lines
3.6 KiB
C

/*
* include/haproxy/tinfo.h
* Export of ha_thread_info[] and ti pointer.
*
* Copyright (C) 2020 Willy Tarreau - w@1wt.eu
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation, version 2.1
* exclusively.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#ifndef _HAPROXY_TINFO_H
#define _HAPROXY_TINFO_H
#include <haproxy/api.h>
#include <haproxy/tinfo-t.h>
#include <haproxy/intops.h>
/* the structs are in thread.c */
extern struct tgroup_info ha_tgroup_info[MAX_TGROUPS];
extern THREAD_LOCAL const struct tgroup_info *tg;
extern struct thread_info ha_thread_info[MAX_THREADS];
extern THREAD_LOCAL const struct thread_info *ti; /* thread_info for the current thread */
extern struct tgroup_ctx ha_tgroup_ctx[MAX_TGROUPS];
extern THREAD_LOCAL struct tgroup_ctx *tg_ctx; /* ha_tgroup_ctx for the current thread */
extern struct thread_ctx ha_thread_ctx[MAX_THREADS];
extern THREAD_LOCAL struct thread_ctx *th_ctx; /* ha_thread_ctx for the current thread */
/* returns the number of threads set in set <ts>. */
static inline int thread_set_count(const struct thread_set *ts)
{
int i, n;
/* iterating over tgroups guarantees to visit all possible threads, the
* opposite is not true.
*/
for (i = n = 0; i < MAX_TGROUPS; i++)
n += my_popcountl(ts->rel[i]);
return n;
}
/* returns zero if the thread set <ts> has at least one thread set,
* otherwise non-zero.
*/
static inline int thread_set_is_empty(const struct thread_set *ts)
{
int i;
/* iterating over tgroups guarantees to visit all possible threads, the
* opposite is not true.
*/
for (i = 0; i < MAX_TGROUPS; i++)
if (ts->rel[i])
return 0;
return 1;
}
/* returns the number starting at 1 of the <n>th thread-group set in thread set
* <ts>, or zero if the set is empty or if thread numbers are only absolute.
* <n> starts at zero and corresponds to the number of non-empty groups to be
* skipped (i.e. 0 returns the first one).
*/
static inline int thread_set_nth_group(const struct thread_set *ts, int n)
{
int i;
if (ts->grps) {
for (i = 0; i < MAX_TGROUPS; i++)
if (ts->rel[i] && !n--)
return i + 1;
}
return 0;
}
/* returns the thread mask of the <n>th assigned thread-group in the thread
* set <ts> for relative sets, the first thread mask at all in case of absolute
* sets, or zero if the set is empty. This is only used temporarily to ease the
* transition. <n> starts at zero and corresponds to the number of non-empty
* groups to be skipped (i.e. 0 returns the first one).
*/
static inline ulong thread_set_nth_tmask(const struct thread_set *ts, int n)
{
int i;
if (ts->grps) {
for (i = 0; i < MAX_TGROUPS; i++)
if (ts->rel[i] && !n--)
return ts->rel[i];
}
return ts->abs[0];
}
/* Pins the thread set to the specified thread mask on group 1 (use ~0UL for
* all threads). This is for compatibility with some rare legacy code. If a
* "thread" directive on a bind line is parsed, this one will be overwritten.
*/
static inline void thread_set_pin_grp1(struct thread_set *ts, ulong mask)
{
int i;
ts->grps = 1;
ts->rel[0] = mask;
for (i = 1; i < MAX_TGROUPS; i++)
ts->rel[i] = 0;
}
#endif /* _HAPROXY_TINFO_H */