Add database enumerator interface to perform iterative searches

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/database.c b/src/database.c
index 563546a130d93b45dc14c1878d14ecdf4057b6ef..e37baa4d250443912e1b62047fae2837837c21ab 100644 (file)
--- a/src/database.c
+++ b/src/database.c
@@ -61,6 +61,12 @@ struct loc_database {
        struct loc_stringpool* pool;
 };
 
        struct loc_stringpool* pool;
 };
 

+struct loc_database_enumerator {
+       struct loc_ctx* ctx;
+       struct loc_database* db;
+       int refcount;
+};
+

 static int loc_database_read_magic(struct loc_database* db, FILE* f) {
        struct loc_database_magic magic;
 
 static int loc_database_read_magic(struct loc_database* db, FILE* f) {
        struct loc_database_magic magic;
 


@@ -578,3 +584,48 @@ LOC_EXPORT int loc_database_lookup_from_string(struct loc_database* db,
 
        return loc_database_lookup(db, &address, network);
 }
 
        return loc_database_lookup(db, &address, network);
 }

+
+// Enumerator
+
+LOC_EXPORT int loc_database_enumerator_new(struct loc_database_enumerator** enumerator, struct loc_database* db) {
+       struct loc_database_enumerator* e = calloc(1, sizeof(*e));
+       if (!e)
+               return -ENOMEM;
+
+       // Reference context
+       e->ctx = loc_ref(db->ctx);
+       e->db = loc_database_ref(db);
+       e->refcount = 1;
+
+       DEBUG(e->ctx, "Database enumerator object allocated at %p\n", e);
+
+       *enumerator = e;
+       return 0;
+}
+
+LOC_EXPORT struct loc_database_enumerator* loc_database_enumerator_ref(struct loc_database_enumerator* enumerator) {
+       enumerator->refcount++;
+
+       return enumerator;
+}
+
+static void loc_database_enumerator_free(struct loc_database_enumerator* enumerator) {
+       DEBUG(enumerator->ctx, "Releasing database enumerator %p\n", enumerator);
+
+       // Release all references
+       loc_database_unref(enumerator->db);
+       loc_unref(enumerator->ctx);
+
+       free(enumerator);
+}
+
+LOC_EXPORT struct loc_database_enumerator* loc_database_enumerator_unref(struct loc_database_enumerator* enumerator) {
+       if (!enumerator)
+               return NULL;
+
+       if (--enumerator->refcount > 0)
+               return enumerator;
+
+       loc_database_enumerator_free(enumerator);
+       return NULL;
+}

diff --git a/src/loc/database.h b/src/loc/database.h
index 2f46e612319cfcb51bc9a6ce428d8d6a5fda1f15..e126f78ce0add1b87abcba8ec1ec3f8caab2cf34 100644 (file)
--- a/src/loc/database.h
+++ b/src/loc/database.h
@@ -43,4 +43,9 @@ int loc_database_lookup(struct loc_database* db,
 int loc_database_lookup_from_string(struct loc_database* db,
                const char* string, struct loc_network** network);
 
 int loc_database_lookup_from_string(struct loc_database* db,
                const char* string, struct loc_network** network);
 

+struct loc_database_enumerator;
+int loc_database_enumerator_new(struct loc_database_enumerator** enumerator, struct loc_database* db);
+struct loc_database_enumerator* loc_database_enumerator_ref(struct loc_database_enumerator* enumerator);
+struct loc_database_enumerator* loc_database_enumerator_unref(struct loc_database_enumerator* enumerator);
+

 #endif
 #endif