beanstalk_standard_hash.c

/*
  +----------------------------------------------------------------------+
  | PHP Version 5                                                        |
  +----------------------------------------------------------------------+
  | Copyright (c) 1997-2007 The PHP Group                                |
  +----------------------------------------------------------------------+
  | This source file is subject to version 3.0 of the PHP license,       |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.php.net/license/3_0.txt.                                  |
  | If you did not receive a copy of the PHP license and are unable to   |
  | obtain it through the world-wide-web, please send a note to          |
  | license@php.net so we can mail you a copy immediately.               |
  +----------------------------------------------------------------------+
  | Authors: Antony Dovgal <tony2001@phpclub.net>                        |
  |          Mikael Johansson <mikael AT synd DOT info>                  |
  +----------------------------------------------------------------------+
*/

/* $Id: beanstalk_standard_hash.c 310129 2011-04-11 04:44:27Z hradtke $ */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "php.h"
#include "beanstalk_pool.h"

//ZEND_EXTERN_MODULE_GLOBALS(beanstalk)

typedef struct bsc_standard_state {
	int						num_servers;
	bsc					**buckets;
	int						num_buckets;
	bsc_hash_function_t		*hash;
} bsc_standard_state_t;

void *bsc_standard_create_state(bsc_hash_function_t *hash) /* {{{ */
{
	bsc_standard_state_t *state = emalloc(sizeof(bsc_standard_state_t));
	memset(state, 0, sizeof(bsc_standard_state_t));
	state->hash = hash;
	return state;
}
/* }}} */

void bsc_standard_free_state(void *s) /* {{{ */
{
	bsc_standard_state_t *state = s;
	if (state != NULL) {
		if (state->buckets != NULL) {
			efree(state->buckets);
		}
		efree(state);
	}
}
/* }}} */

bsc *bsc_standard_find_server(void *s, const char *key, unsigned int key_len TSRMLS_DC) /* {{{ */
{
	bsc_standard_state_t *state = s;

	if (state->num_servers > 1) {
		/* "new-style" hash */
		unsigned int hash = (bsc_hash(state->hash, key, key_len) >> 16) & 0x7fff; 
		return state->buckets[(hash ? hash : 1) % state->num_buckets];
	}

	return state->buckets[0];
}
/* }}} */

void bsc_standard_add_server(void *s, bsc *svr, unsigned int weight) /* {{{ */
{
	bsc_standard_state_t *state = s;
	int i;

	/* add weight number of buckets for this server */
	state->buckets = erealloc(state->buckets, sizeof(*state->buckets) * (state->num_buckets + weight));

	for (i=0; i<weight; i++) {
		state->buckets[state->num_buckets + i] = svr;
	}

	state->num_buckets += weight;
	state->num_servers++;
}
/* }}} */

bsc_hash_strategy_t bsc_standard_hash = {
	bsc_standard_create_state,
	bsc_standard_free_state,
	bsc_standard_find_server,
	bsc_standard_add_server
};

/*
 * Local variables:
 * tab-width: 4
 * c-basic-offset: 4
 * End:
 * vim600: noet sw=4 ts=4 fdm=marker
 * vim<600: noet sw=4 ts=4
 */