void cache_get_stats (struct cache *cache, tstat_t *ts, int *sizep,
int *incompletep, int *dirtyp)
{
zlist_t *keys;
struct cache_entry *hp;
char *ref;
int size = 0;
int incomplete = 0;
int dirty = 0;
if (!(keys = zhash_keys (cache->zh)))
oom ();
while ((ref = zlist_pop (keys))) {
hp = zhash_lookup (cache->zh, ref);
if (cache_entry_get_valid (hp)) {
int obj_size = strlen (Jtostr (hp->o));
size += obj_size;
tstat_push (ts, obj_size);
} else
incomplete++;
if (cache_entry_get_dirty (hp))
dirty++;
free (ref);
}
zlist_destroy (&keys);
if (sizep)
*sizep = size;
if (incompletep)
*incompletep = incomplete;
if (dirtyp)
*dirtyp = dirty;
}
int cache_expire_entries (struct cache *cache, int current_epoch, int thresh)
{
zlist_t *keys;
char *ref;
struct cache_entry *hp;
int count = 0;
if (!(keys = zhash_keys (cache->zh)))
oom ();
while ((ref = zlist_pop (keys))) {
if ((hp = zhash_lookup (cache->zh, ref))
&& !cache_entry_get_dirty (hp)
&& cache_entry_get_valid (hp)
&& (thresh == 0 || cache_entry_age (hp, current_epoch) > thresh)) {
zhash_delete (cache->zh, ref);
count++;
}
free (ref);
}
zlist_destroy (&keys);
return count;
}
int main (int argc, char *argv[])
{
struct cache *cache;
struct cache_entry *e1, *e2;
json_object *o1;
json_object *o2;
wait_t *w;
int count, i;
plan (NO_PLAN);
cache_destroy (NULL);
cache_entry_destroy (NULL);
diag ("cache_destroy and cache_entry_destroy accept NULL arg");
ok ((cache = cache_create ()) != NULL,
"cache_create works");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
cache_destroy (cache);
/* Play with one entry.
* N.B.: json ref is NOT incremented by create or get_json.
*/
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
ok ((e1 = cache_entry_create (o1)) != NULL,
"cache_entry_create works");
ok (cache_entry_get_valid (e1) == true,
"cache entry initially valid");
ok (cache_entry_get_dirty (e1) == false,
"cache entry initially not dirty");
cache_entry_set_dirty (e1, true);
ok (cache_entry_get_dirty (e1) == true,
"cache entry succcessfully set dirty");
ok ((o2 = cache_entry_get_json (e1)) != NULL,
"json retrieved from cache entry");
ok (Jget_int (o2, "foo", &i) == true && i == 42,
"expected json object found");
cache_entry_destroy (e1); /* destroys o1 */
/* Test cache entry waiters.
* N.B. waiter is destroyed when run.
*/
count = 0;
ok ((w = wait_create (wait_cb, &count)) != NULL,
"wait_create works");
ok ((e1 = cache_entry_create (NULL)) != NULL,
"cache_entry_create created empty object");
ok (cache_entry_get_valid (e1) == false,
"cache entry invalid, adding waiter");
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
cache_entry_wait_valid (e1, w);
cache_entry_set_json (e1, o1);
ok (cache_entry_get_valid (e1) == true,
"cache entry set valid with one waiter");
ok (count == 1,
"waiter callback ran");
count = 0;
ok ((w = wait_create (wait_cb, &count)) != NULL,
"wait_create works");
cache_entry_set_dirty (e1, true);
ok (cache_entry_get_dirty (e1) == true,
"cache entry set dirty, adding waiter");
cache_entry_wait_notdirty (e1, w);
cache_entry_set_dirty (e1, false);
ok (cache_entry_get_dirty (e1) == false,
"cache entry set not dirty with one waiter");
ok (count == 1,
"waiter callback ran");
cache_entry_destroy (e1); /* destroys o1 */
/* Put entry in cache and test lookup, expire
*/
ok ((cache = cache_create ()) != NULL,
"cache_create works");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
ok ((e1 = cache_entry_create (o1)) != NULL,
"cache_entry_create works");
cache_insert (cache, "xxx1", e1);
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry after insert");
ok (cache_lookup (cache, "yyy1", 0) == NULL,
"cache_lookup of wrong hash fails");
ok ((e2 = cache_lookup (cache, "xxx1", 42)) != NULL,
"cache_lookup of correct hash works (last use=42)");
i = 0;
ok ((o2 = cache_entry_get_json (e2)) != NULL
&& Jget_int (o2, "foo", &i) == true && i == 42,
"expected json object found");
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry");
ok (cache_expire_entries (cache, 43, 1) == 0,
"cache_expire_entries now=43 thresh=1 expired 0");
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry");
ok (cache_expire_entries (cache, 44, 1) == 1,
"cache_expire_entries now=44 thresh=1 expired 1");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
cache_destroy (cache);
done_testing ();
return (0);
}
