创建

cache/test_demo/cache_evbase_test.cpp

@@ -0,0 +1,224 @@
+#include <sys/ioctl.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <net/if.h>
+#include <unistd.h> 
+#include <stdio.h> 
+#include <stdlib.h>
+#include <assert.h>
+#include <errno.h>
+#include <sys/prctl.h>
+#include <sys/un.h>
+#include <unistd.h>
+#include <sys/poll.h>
+#include <curl/curl.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include <sys/cdefs.h>
+#include <MESA/MESA_handle_logger.h>
+
+#include "cache_evbase_client.h"
+
+struct cache_evbase_instance *instance_asyn;
+int still_runing=1;
+
+struct future_pdata
+{
+	struct future * future;
+	FILE *fp;
+	char filename[256];
+};
+
+void get_future_success(future_result_t* result, void * user)
+{
+	struct tango_cache_result *res = (struct tango_cache_result *)result;
+	struct future_pdata *pdata = (struct future_pdata *)user;
+	char buffer[1024];
+
+	if(res != NULL)
+	{
+		if(res->type==RESULT_TYPE_BODY)
+		{
+			fwrite(res->data_frag, res->size, 1, pdata->fp);
+		}
+		else
+		{
+			memcpy(buffer, res->data_frag, res->size>=1024?1023:res->size);
+			buffer[res->size] = '\0';
+			printf("%s", buffer);
+		}
+	}
+	else //<2F><><EFBFBD><EFBFBD>
+	{
+		future_destroy(pdata->future);
+		fclose(pdata->fp);
+		free(pdata);
+		still_runing = 0;
+	}
+}
+
+void get_future_failed(enum e_future_error err, const char * what, void * user)
+{
+	printf("GET fail: %s\n", what);
+	sleep(5);
+	still_runing = 0;
+}
+
+void put_future_success(future_result_t* result, void * user)
+{
+	struct future_pdata *pdata = (struct future_pdata *)user;
+
+	printf("PUT %s succ\n", pdata->filename);
+	future_destroy(pdata->future);
+	free(pdata);
+	still_runing = 0;
+}
+void put_future_failed(enum e_future_error err, const char * what, void * user)
+{
+	struct future_pdata *pdata = (struct future_pdata *)user;
+
+	printf("PUT %s fail: %s\n", what, pdata->filename);
+	future_destroy(pdata->future);
+	free(pdata);
+	still_runing = 0;
+}
+
+char * get_file_content(const char *filename, size_t *filelen_out)
+{
+	char *buffer;
+	FILE *fp;
+	size_t filelen = 0;
+	struct stat filestat;
+	int readlen;
+
+	fp = fopen(filename, "rb");
+	if(fp == NULL)
+	{
+		printf("fopen file %s failed.\n", filename);
+		return NULL;
+	}
+	if(fstat(fileno(fp), &filestat))
+	{
+		printf("fstat %s failed.\n", filename);
+		return NULL;
+	}
+
+	buffer = (char *)malloc(filestat.st_size);
+
+	while(filelen < (size_t)filestat.st_size)
+	{
+		readlen = fread(buffer + filelen, 1, filestat.st_size - filelen, fp);
+		if(readlen < 0)
+		{
+			printf("read error: %s\n", strerror(errno));
+			return NULL;
+		}
+
+		filelen += readlen;
+	}
+	fclose(fp);
+	*filelen_out = filestat.st_size;
+
+	return buffer;
+}
+
+int main(int argc, char **argv)
+{
+	int n, index=0;
+	char method[16], filename_in[256], filename_out[256], *p;
+	struct tango_cache_meta meta;
+	struct future_pdata *pdata;
+	struct cache_evbase_ctx *ctx;
+	void *runtime_log;
+
+	if(argc != 2 && argc!=3)
+	{
+		printf("USGAE: %s <PUT/PUTONCE/GET:filename> [get_out_file_index]\n", argv[0]);
+		return -1;
+	}
+	if(argc==3)
+	{
+		index = atoi(argv[2]);
+	}
+	
+	runtime_log = MESA_create_runtime_log_handle("./runtime.log", 10);
+	if(NULL==runtime_log)
+	{
+		return -1;
+	}
+		
+	instance_asyn = cache_evbase_instance_new("./pangu_tg_cahce.conf", "TANGO_CACHE", runtime_log);
+	assert(instance_asyn!=NULL);
+
+	pdata = (struct future_pdata *)malloc(sizeof(struct future_pdata));
+
+	if(sscanf(argv[1], "%[^:]:%1023s%n", method, filename_in, &n) != 2)
+	{
+		assert(0);
+	}
+	if(strlen(filename_in) > 0) 
+	{
+		memset(&meta, 0, sizeof(struct tango_cache_meta));
+		meta.url = filename_in;
+		meta.std_hdr[HDR_CONTENT_TYPE] = "Content-Type: maintype/subtype";
+		meta.std_hdr[HDR_CONTENT_ENCODING] = "Content-Encoding: gzip";
+		meta.other_hdr = "Etag: hgdkqkwdwqekdfjwjfjwelkjfkwfejwhf\r\n";
+		
+		p = method;
+		while(*p=='\r'||*p=='\n') p++;
+		assert(*p!='\0');
+
+		if(!strcasecmp(p, "GET"))
+		{
+			sprintf(filename_out, "file_index_%u.bin", index);
+			pdata->future = future_create(get_future_success, get_future_failed, pdata);
+			promise_set_ctx(future_to_promise(pdata->future), NULL, NULL);
+			pdata->fp = fopen(filename_out, "w");
+			
+			cache_evbase_fetch(instance_asyn, pdata->future, &meta);
+		}
+		else if(!strcasecmp(p, "PUTONCE"))
+		{
+			size_t filelen;
+			p = get_file_content(filename_in, &filelen);
+			pdata->future = future_create(put_future_success, put_future_failed, pdata);
+			promise_set_ctx(future_to_promise(pdata->future), NULL, NULL);
+			
+			cache_evbase_upload_once(instance_asyn, pdata->future, PUT_ONCE_FREE, p, filelen, &meta, pdata->filename, 256);
+		}
+		else
+		{
+			pdata->future = future_create(put_future_success, put_future_failed, pdata);
+			promise_set_ctx(future_to_promise(pdata->future), NULL, NULL);
+
+			ctx = cache_evbase_update_start(instance_asyn, pdata->future, &meta);
+			sprintf(pdata->filename, "%s", cache_evbase_get_object_key(ctx));
+
+			char buffer[1024];
+			FILE *fp = fopen(filename_in, "r");
+			while(!feof(fp))
+			{
+				n = fread(buffer, 1, 1024, fp);
+				assert(n>=0);
+				cache_evbase_update_frag(ctx, PUT_ONCE_COPY, buffer, n);
+			}
+			
+			cache_evbase_update_end(ctx);
+		}
+	}
+
+	printf("Waiting to finish.......\n");
+	while(still_runing)
+	{
+		sleep(1);
+	}
+
+	struct cache_statistics out;
+	cache_evbase_get_statistics(instance_asyn, &out);
+	printf("get_recv: %llu, get_succ: %llu, get_fail: %llu, put_recv: %llu, put_succ: %llu, put_fail: %llu, session: %llu, memory: %llu\n",
+		out.get_recv_num, out.get_succ_num, out.get_fail_num, out.put_recv_num, out.put_succ_num, out.put_fail_num, out.session_num, out.memory_used);
+	return 0;
+}
+