hashmap.c

来自「NAT打洞」· C语言 代码 · 共 90 行

#include "hashmap.h"

HASHMAP *hashmap(void){
  HASHMAP *res = (HASHMAP*)malloc(sizeof(HASHMAP));
  int i;
  memset(res, 0, sizeof(HASHMAP));
  for(i = 0; i < 1024; i++){
  	res->list[i] = NULL;
  }
  return res;
}

void hashmap_put(HASHMAP *hm, unsigned int key, void *value){
  unsigned int hash = key % 1024;

  HASHMAP_ENTRY *ptr = hm->list[hash];
  while(ptr != NULL){
    if(ptr->key == key){
    	ptr->value = (char*)value;
    	return;
    }
    ptr = ptr->next;
  }

  ptr = (HASHMAP_ENTRY*)malloc(sizeof(HASHMAP_ENTRY));
  ptr->next = hm->list[hash];
  ptr->key = key;
  ptr->value = (char*)value;

  hm->list[hash] = ptr;
}

void* hashmap_get(HASHMAP *hm, unsigned int key){
  unsigned int hash = key % 1024;

  HASHMAP_ENTRY *ptr = hm->list[hash];
  while(ptr != NULL){
    if(ptr->key == key){
    	return ptr->value;
    }
    ptr = ptr->next;
  }

  return NULL;
}

void *hashmap_remove(HASHMAP *hm, unsigned int key){
  unsigned int hash = key % 1024;
  void *res;

  HASHMAP_ENTRY *ptr = hm->list[hash];
  while(ptr != NULL){
    if(ptr->key == key){
    	res = ptr->value;
    	ptr->value = NULL;
      return res;
    }
    ptr = ptr->next;
  }
  return NULL;

}

void hashmap_print(HASHMAP *hm, void (__cdecl *print_value)(char*)){
	int i, j, n = 0;
  if(hm == NULL){
  	return;
  }
  printf("[");
  for(j = 0; j < 1024; j++){
	  HASHMAP_ENTRY *ptr = hm->list[j];
  	while(ptr != NULL){
      if(ptr->value != NULL){
	    	if(n > 0){
  	    	printf(", ");
    	  }
        printf("%d", ptr->key);
        if(print_value != NULL){
          print_value(ptr->value);
        }
        n++;
      }
	    ptr = ptr->next;
    }
  }
  printf("]\n");
}