Skip to content

Commit

Permalink
gossmap: optimize asserts().
Browse files Browse the repository at this point in the history
They are surprisingly expensive!

Running `time ./plugins/renepay/test/run-not_mcf-gossmap gossip_store-sgl.rustcorp.com.au-2022-04-19 024b9a1fa8e006f1e3937f65f66c408e6da8e1ca728ea43222a7381df1cc449605 02ebb3b8a2316b3e876ea3f3d8124a3ab97f30b128f619608eb06b5251235dc2d9 10000000000 0.1`:

Before (-Og):
	real	0m1.495s
Before (no opt):
	real	0m2.552s

After (-Og):
	real	0m0.579s
After (no opt):
	real	0m1.061s

Signed-off-by: Rusty Russell <[email protected]>
  • Loading branch information
rustyrussell committed Sep 19, 2022
1 parent 4cdb416 commit fd71dfc
Showing 1 changed file with 19 additions and 9 deletions.
28 changes: 19 additions & 9 deletions common/gossmap.c
Original file line number Diff line number Diff line change
Expand Up @@ -66,9 +66,13 @@ struct gossmap {

/* Array of nodes, so we can use simple index. */
struct gossmap_node *node_arr;
/* This is tal_count(node_arr), which we call very often in assert() */
size_t num_node_arr;

/* Array of chans, so we can use simple index */
struct gossmap_chan *chan_arr;
/* This is tal_count(chan_arr), which we call very often in assert() */
size_t num_chan_arr;

/* Linked list of freed ones, if any. */
u32 freed_nodes, freed_chans;
Expand Down Expand Up @@ -155,24 +159,26 @@ static int map_feature_test(const struct gossmap *map,
/* These values can change across calls to gossmap_check. */
u32 gossmap_max_node_idx(const struct gossmap *map)
{
return tal_count(map->node_arr);
assert(tal_count(map->node_arr) == map->num_node_arr);
return map->num_node_arr;
}

u32 gossmap_max_chan_idx(const struct gossmap *map)
{
return tal_count(map->chan_arr);
assert(tal_count(map->chan_arr) == map->num_chan_arr);
return map->num_chan_arr;
}

/* Each channel has a unique (low) index. */
u32 gossmap_node_idx(const struct gossmap *map, const struct gossmap_node *node)
{
assert(node - map->node_arr < tal_count(map->node_arr));
assert(node - map->node_arr < map->num_node_arr);
return node - map->node_arr;
}

u32 gossmap_chan_idx(const struct gossmap *map, const struct gossmap_chan *chan)
{
assert(chan - map->chan_arr < tal_count(map->chan_arr));
assert(chan - map->chan_arr < map->num_chan_arr);
return chan - map->chan_arr;
}

Expand Down Expand Up @@ -265,6 +271,7 @@ static struct gossmap_node *next_free_node(struct gossmap *map)
if (map->freed_nodes == UINT_MAX) {
/* Double in size, add second half to free list */
size_t n = tal_count(map->node_arr);
map->num_node_arr *= 2;
tal_resize(&map->node_arr, n * 2);
map->freed_nodes = init_node_arr(map->node_arr, n);
}
Expand Down Expand Up @@ -325,6 +332,7 @@ static struct gossmap_chan *next_free_chan(struct gossmap *map)
if (map->freed_chans == UINT_MAX) {
/* Double in size, add second half to free list */
size_t n = tal_count(map->chan_arr);
map->num_chan_arr *= 2;
tal_resize(&map->chan_arr, n * 2);
map->freed_chans = init_chan_arr(map->chan_arr, n);
}
Expand Down Expand Up @@ -680,9 +688,11 @@ static bool load_gossip_store(struct gossmap *map, size_t *num_rejected)
chanidx_htable_init_sized(&map->channels, map->map_size / 750 / 2);
nodeidx_htable_init_sized(&map->nodes, map->map_size / 2500 / 2);

map->chan_arr = tal_arr(map, struct gossmap_chan, map->map_size / 750 / 2 + 1);
map->num_chan_arr = map->map_size / 750 / 2 + 1;
map->chan_arr = tal_arr(map, struct gossmap_chan, map->num_chan_arr);
map->freed_chans = init_chan_arr(map->chan_arr, 0);
map->node_arr = tal_arr(map, struct gossmap_node, map->map_size / 2500 / 2 + 1);
map->num_node_arr = map->map_size / 2500 / 2 + 1;
map->node_arr = tal_arr(map, struct gossmap_node, map->num_node_arr);
map->freed_nodes = init_node_arr(map->node_arr, 0);

map->map_end = 1;
Expand Down Expand Up @@ -1031,7 +1041,7 @@ struct gossmap_chan *gossmap_nth_chan(const struct gossmap *map,
struct gossmap_chan *chan;

assert(n < node->num_chans);
assert(node->chan_idxs[n] < tal_count(map->chan_arr));
assert(node->chan_idxs[n] < map->num_chan_arr);
chan = map->chan_arr + node->chan_idxs[n];

if (which_half) {
Expand Down Expand Up @@ -1061,7 +1071,7 @@ size_t gossmap_num_nodes(const struct gossmap *map)

static struct gossmap_node *node_iter(const struct gossmap *map, size_t start)
{
for (size_t i = start; i < tal_count(map->node_arr); i++) {
for (size_t i = start; i < map->num_node_arr; i++) {
if (map->node_arr[i].chan_idxs != NULL)
return &map->node_arr[i];
}
Expand All @@ -1086,7 +1096,7 @@ size_t gossmap_num_chans(const struct gossmap *map)

static struct gossmap_chan *chan_iter(const struct gossmap *map, size_t start)
{
for (size_t i = start; i < tal_count(map->chan_arr); i++) {
for (size_t i = start; i < map->num_chan_arr; i++) {
if (map->chan_arr[i].plus_scid_off != 0)
return &map->chan_arr[i];
}
Expand Down

0 comments on commit fd71dfc

Please sign in to comment.