Add cancellation to iterators over avltree(n) and genset(dyn)

1 files changed, 18 insertions, 0 deletions

diff --git a/src/libdatastruct/gensetdyn_iter_withcancel.c b/src/libdatastruct/gensetdyn_iter_withcancel.c
new file mode 100644
index 0000000..ffac1e0
--- /dev/null
+++ b/src/libdatastruct/gensetdyn_iter_withcancel.c
@@ -0,0 +1,18 @@
+/* ISC license. */
+
+#include <errno.h>
+#include <skalibs/functypes.h>
+#include <skalibs/gensetdyn.h>
+
+int gensetdyn_iter_withcancel (gensetdyn *g, iterfunc_t_ref f, iterfunc_t_ref cancelf, void *stuff)
+{
+  unsigned int n = gensetdyn_iter(g, f, stuff) ;
+  if (n < g->storage.len)
+  {
+    int e = errno ;
+    gensetdyn_iter_nocancel(g, n, cancelf, stuff) ;
+    errno = e ;
+    return 0 ;
+  }
+  return 1 ;
+}