Commit fb92682e authored by Rusty Russell's avatar Rusty Russell

htable: htable_type add htable_copy.

Signed-off-by: default avatarRusty Russell <rusty@rustcorp.com.au>
parent f359fde1
...@@ -21,8 +21,9 @@ ...@@ -21,8 +21,9 @@
* *
* It also defines initialization and freeing functions: * It also defines initialization and freeing functions:
* void <name>_init(struct <name> *); * void <name>_init(struct <name> *);
* void <name>_init_sized(struct <name> *, size_t); * bool <name>_init_sized(struct <name> *, size_t);
* void <name>_clear(struct <name> *); * void <name>_clear(struct <name> *);
* bool <name>_copy(struct <name> *dst, const struct <name> *src);
* *
* Add function only fails if we run out of memory: * Add function only fails if we run out of memory:
* bool <name>_add(struct <name> *ht, const <type> *e); * bool <name>_add(struct <name> *ht, const <type> *e);
...@@ -63,15 +64,20 @@ ...@@ -63,15 +64,20 @@
{ \ { \
htable_init(&ht->raw, name##_hash, NULL); \ htable_init(&ht->raw, name##_hash, NULL); \
} \ } \
static inline UNNEEDED void name##_init_sized(struct name *ht, \ static inline UNNEEDED bool name##_init_sized(struct name *ht, \
size_t s) \ size_t s) \
{ \ { \
htable_init_sized(&ht->raw, name##_hash, NULL, s); \ return htable_init_sized(&ht->raw, name##_hash, NULL, s); \
} \ } \
static inline UNNEEDED void name##_clear(struct name *ht) \ static inline UNNEEDED void name##_clear(struct name *ht) \
{ \ { \
htable_clear(&ht->raw); \ htable_clear(&ht->raw); \
} \ } \
static inline UNNEEDED bool name##_copy(struct name *dst, \
const struct name *src) \
{ \
return htable_copy(&dst->raw, &src->raw); \
} \
static inline bool name##_add(struct name *ht, const type *elem) \ static inline bool name##_add(struct name *ht, const type *elem) \
{ \ { \
return htable_add(&ht->raw, hashfn(keyof(elem)), elem); \ return htable_add(&ht->raw, hashfn(keyof(elem)), elem); \
......
...@@ -110,13 +110,13 @@ static bool check_mask(struct htable *ht, const struct obj val[], unsigned num) ...@@ -110,13 +110,13 @@ static bool check_mask(struct htable *ht, const struct obj val[], unsigned num)
int main(int argc, char *argv[]) int main(int argc, char *argv[])
{ {
unsigned int i; unsigned int i;
struct htable_obj ht; struct htable_obj ht, ht2;
struct obj val[NUM_VALS], *result; struct obj val[NUM_VALS], *result;
unsigned int dne; unsigned int dne;
void *p; void *p;
struct htable_obj_iter iter; struct htable_obj_iter iter;
plan_tests(27); plan_tests(29);
for (i = 0; i < NUM_VALS; i++) for (i = 0; i < NUM_VALS; i++)
val[i].key = i; val[i].key = i;
dne = i; dne = i;
...@@ -171,8 +171,12 @@ int main(int argc, char *argv[]) ...@@ -171,8 +171,12 @@ int main(int argc, char *argv[])
find_vals(&ht, val, NUM_VALS); find_vals(&ht, val, NUM_VALS);
ok1(!htable_obj_get(&ht, &dne)); ok1(!htable_obj_get(&ht, &dne));
/* Check copy. */
ok1(htable_obj_copy(&ht2, &ht));
/* Delete them all by key. */ /* Delete them all by key. */
del_vals_bykey(&ht, val, NUM_VALS); del_vals_bykey(&ht, val, NUM_VALS);
del_vals_bykey(&ht2, val, NUM_VALS);
/* Write two of the same value. */ /* Write two of the same value. */
val[1] = val[0]; val[1] = val[0];
...@@ -201,5 +205,6 @@ int main(int argc, char *argv[]) ...@@ -201,5 +205,6 @@ int main(int argc, char *argv[])
} }
htable_obj_clear(&ht); htable_obj_clear(&ht);
htable_obj_clear(&ht2);
return exit_status(); return exit_status();
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment