Skip to content
Snippets Groups Projects
Forked from Knot projects / Knot DNS
12466 commits behind the upstream repository.
ahtable.h 3.51 KiB
/*
 * This file is part of hat-trie.
 *
 * Copyright (c) 2011 by Daniel C. Jones <dcjones@cs.washington.edu>
 *
 *
 * This is an implementation of the 'cache-conscious' hash tables described in,
 *
 *    Askitis, N., & Zobel, J. (2005). Cache-conscious collision resolution in
 *    string hash tables. String Processing and Information Retrieval (pp.
 *    91–102). Springer.
 *
 * Briefly, the idea is, as opposed to separate chaining with linked lists, to
 * store keys contiguously in one big array, thereby improving the caching
 * behavior, and reducing space requirments.
 *
 */

#ifndef HATTRIE_AHTABLE_H
#define HATTRIE_AHTABLE_H

#ifdef __cplusplus
extern "C" {
#endif

#include <stdlib.h>
#include <stdbool.h>
#include <stdint.h>
#include "libknot/common.h"
#include "common/hattrie/hat-trie.h"

typedef unsigned char* slot_t;

typedef struct ahtable_t_
{
    /* these fields are reserved for hattrie to fiddle with */
    uint8_t flag;
    unsigned char c0;
    unsigned char c1;

    size_t n;        // number of slots
    size_t m;        // number of key/value pairs stored
    size_t max_m;    // number of stored keys before we resize

    uint32_t*  slot_sizes;
    slot_t*  slots;
    slot_t*  index;  // order index (optional)
} ahtable_t;

ahtable_t* ahtable_create   (void);         // Create an empty hash table.
ahtable_t* ahtable_create_n (size_t n);     // Create an empty hash table, with
                                            //  n slots reserved.

void       ahtable_free   (ahtable_t*);       // Free all memory used by a table.
void       ahtable_clear  (ahtable_t*);       // Remove all entries.
size_t     ahtable_size   (const ahtable_t*); // Number of stored keys.


/** Find the given key in the table, inserting it if it does not exist, and
 * returning a pointer to its key.
 *
 * This pointer is not guaranteed to be valid after additional calls to
 * ahtable_get, ahtable_del, ahtable_clear, or other functions that modifies the
 * table.
 */
value_t* ahtable_get (ahtable_t*, const char* key, size_t len);

/** Find a given key in the table, returning a NULL pointer if it does not
 * exist. */
value_t* ahtable_tryget (ahtable_t*, const char* key, size_t len);

/** Return value from index.
 */
value_t *ahtable_indexval(ahtable_t*, unsigned i);

/** Build order index for fast ordered lookup.
 */
void ahtable_build_index(ahtable_t*);

/** Find a key that is exact match or lexicographic predecessor.
 *  \retval  0 if exact match
 *  \retval  1 if couldn't find and no predecessor is found
 *  \retval -1 if found predecessor
 */
int ahtable_find_leq (ahtable_t*, const char* key, size_t len, value_t** dst);


/** Insert given key and value without checking for existence.
 */
void ahtable_insert (ahtable_t* T, const char* key, size_t len, value_t val);


int ahtable_del(ahtable_t*, const char* key, size_t len);

typedef struct ahtable_iter_t_
{
    unsigned flags;
    ahtable_t* T; // parent
    uint32_t i; // current key
    union {
        slot_t* xs; // pointers to keys
        slot_t s;           // slot position
    } d;

} ahtable_iter_t;

void            ahtable_iter_begin     (ahtable_t*, ahtable_iter_t*, bool sorted);
void            ahtable_iter_next      (ahtable_iter_t*);
void            ahtable_iter_del       (ahtable_iter_t*);
bool            ahtable_iter_finished  (ahtable_iter_t*);
void            ahtable_iter_free      (ahtable_iter_t*);
const char*     ahtable_iter_key       (ahtable_iter_t*, size_t* len);
value_t*        ahtable_iter_val       (ahtable_iter_t*);


#ifdef __cplusplus
}
#endif

#endif