aljaz/lightning
1
0
Fork 0
mirror of https://github.com/aljazceru/lightning.git synced 2025-12-21 16:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

plugins/renepay/dijkstra: improve API to remove global.

Browse Source 
The global is an *internal* hack because dijkstra_item_mover doesn't
take a context arg!  It should be used with care.

Easy, since all the accessors exist: we just hand in the struct dijkstra.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Rusty Russell
2023-07-31 11:21:25 +09:30
parent 79486c1e3b
commit b5da85e85d
4 changed files with 155 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
plugins/renepay/test/run-dijkstra.c
View File

@@ -11,70 +11,70 @@
static void insertion_in_increasing_distance(const tal_t *ctx)
{
dijkstra_malloc(ctx,10);
struct dijkstra *dijkstra = dijkstra_new(ctx,10);
for(int i=0;i<dijkstra_maxsize();++i)
for(int i=0;i<dijkstra_maxsize(dijkstra);++i)
{
dijkstra_update(i,10+i);
assert(dijkstra_size()==(i+1));
dijkstra_update(dijkstra,i,10+i);
assert(dijkstra_size(dijkstra)==(i+1));
}
dijkstra_update(3,3);
assert(dijkstra_top()==3);
dijkstra_update(dijkstra,3,3);
assert(dijkstra_top(dijkstra)==3);
dijkstra_update(3,15);
assert(dijkstra_top()==0);
dijkstra_update(dijkstra,3,15);
assert(dijkstra_top(dijkstra)==0);
dijkstra_update(3,-1);
assert(dijkstra_top()==3);
dijkstra_update(dijkstra,3,-1);
assert(dijkstra_top(dijkstra)==3);
dijkstra_pop();
assert(dijkstra_size()==9);
assert(dijkstra_top()==0);
dijkstra_pop(dijkstra);
assert(dijkstra_size(dijkstra)==9);
assert(dijkstra_top(dijkstra)==0);
// Insert again
dijkstra_update(3,3+10);
dijkstra_update(dijkstra,3,3+10);
u32 top=0;
while(!dijkstra_empty())
while(!dijkstra_empty(dijkstra))
{
assert(top==dijkstra_top());
assert(top==dijkstra_top(dijkstra));
top++;
dijkstra_pop();
dijkstra_pop(dijkstra);
}
}
static void insertion_in_decreasing_distance(const tal_t *ctx)
{
dijkstra_malloc(ctx,10);
struct dijkstra *dijkstra = dijkstra_new(ctx,10);
for(int i=0;i<dijkstra_maxsize();++i)
for(int i=0;i<dijkstra_maxsize(dijkstra);++i)
{
dijkstra_update(i,10-i);
assert(dijkstra_size()==(i+1));
dijkstra_update(dijkstra,i,10-i);
assert(dijkstra_size(dijkstra)==(i+1));
}
dijkstra_update(3,-3);
assert(dijkstra_top()==3);
dijkstra_update(dijkstra,3,-3);
assert(dijkstra_top(dijkstra)==3);
dijkstra_update(3,15);
assert(dijkstra_top()==9);
dijkstra_update(dijkstra,3,15);
assert(dijkstra_top(dijkstra)==9);
dijkstra_update(3,-1);
assert(dijkstra_top()==3);
dijkstra_update(dijkstra,3,-1);
assert(dijkstra_top(dijkstra)==3);
dijkstra_pop();
assert(dijkstra_size()==9);
assert(dijkstra_top()==9);
dijkstra_pop(dijkstra);
assert(dijkstra_size(dijkstra)==9);
assert(dijkstra_top(dijkstra)==9);
// Insert again
dijkstra_update(3,10-3);
dijkstra_update(dijkstra,3,10-3);
u32 top=9;
while(!dijkstra_empty())
while(!dijkstra_empty(dijkstra))
{
assert(top==dijkstra_top());
assert(top==dijkstra_top(dijkstra));
top--;
dijkstra_pop();
dijkstra_pop(dijkstra);
}
}
@@ -82,14 +82,6 @@ int main(int argc, char *argv[])
{
common_setup(argv[0]);
insertion_in_increasing_distance(NULL);
insertion_in_decreasing_distance(tmpctx);
// test dijkstra_free
dijkstra_free();
// we can call it twice, no problem
dijkstra_free();
// does tal_free() cleansup correctly?
const tal_t *this_ctx = tal(NULL,tal_t);
insertion_in_increasing_distance(this_ctx);