rspamd/src/memcached-test.c

#include <stdio.h>
#include <stdlib.h>
#include <netdb.h>
#include <errno.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <event.h>

#include "upstream.h"
#include "memcached.h"

#define HOST "127.0.0.1"
#define PORT 11211

memcached_param_t cur_param;

static void
test_memc_callback (memcached_ctx_t *ctx, memc_error_t error, void *data)
{	
	int s;
	int r;
	int *num = ((int *)data);
	printf ("result of memc command '%s' is '%s'\n", ctx->cmd, memc_strerror (error));
	/* Connect */
	if (*num == 0) {
		printf ("Setting value to memcached: %s -> %s\n", cur_param.key, (char *)cur_param.buf);
		s = 1;
		r = memc_set (ctx, &cur_param, &s, 60);
		(*num)++;
	}
	else if (*num == 1) {
		printf ("Getting value from memcached: %s -> %s\n", cur_param.key, (char *)cur_param.buf);
		s = 1;
		r = memc_get (ctx, &cur_param, &s);
		(*num)++;
	}
	else {
		printf ("Got value from memcached: %s -> %s\n", cur_param.key, (char *)cur_param.buf);
		event_loopexit (NULL);
	}
}


int 
main (int argc, char **argv)
{
	memcached_ctx_t mctx;
	char *addr, buf[512];
	int num = 0;
	
	event_init ();
	strcpy (cur_param.key, "testkey");
	strcpy (buf, "test_value");
	cur_param.buf = buf;
	cur_param.bufsize = sizeof ("test_value") - 1;

	if (argc == 2) {
		addr = argv[1];
	}
	else {
		addr = HOST;
	}
	
	mctx.protocol = TCP_TEXT;
	mctx.timeout.tv_sec = 1;
	mctx.timeout.tv_usec = 0;
	mctx.port = htons (PORT);
	mctx.options = MEMC_OPT_DEBUG;
	mctx.callback = test_memc_callback;
	/* XXX: it is wrong to use local variable pointer here */
	mctx.callback_data = (void *)&num;
	inet_aton (addr, &mctx.addr);

	memc_init_ctx (&mctx);
	
	event_loop (0);
	return 0;
}