增加 performan 测试

example/performance/HosClientPerformance.cpp

@@ -0,0 +1,542 @@
+/*************************************************************************
+    > File Name: HosClientPerformance.cpp
+    > Author: pxz
+    > Created Time: Sat 10 Oct 2020 05:26:02 PM CST
+ ************************************************************************/
+extern "C"
+{
+#include<stdio.h>
+#include<stdlib.h>
+#include<unistd.h>
+#include<string.h>
+#include<time.h>
+#include<pthread.h>
+#include<dirent.h>
+#include<sys/stat.h>
+}
+#include"../../src/hos_client.h"
+
+#define MAX_THREAD_NUM 16
+#ifndef MIN
+#define MIN(a,b) ((a) > (b)) ? (b) : (a)
+#endif
+
+typedef struct conf_s
+{
+#define STRING_SIZE 128
+    char endpoint[STRING_SIZE];
+    char bucket[STRING_SIZE];
+    char object[STRING_SIZE];
+    char file[STRING_SIZE];
+    size_t pool_size;
+    size_t thread_sum;
+    size_t append_size;
+    int mode;
+}conf_t;
+
+typedef struct thread_info_s
+{
+    conf_t conf;
+    hos_client_handle handle;
+    size_t  thread_num;
+}thread_info_t;
+
+static void configuration_init(conf_t *conf)
+{
+    strcpy(conf->endpoint, "http://192.168.40.223:9098/hos/");
+    strcpy(conf->bucket, "mybucket");
+    strcpy(conf->object, "myobject");
+    strcpy(conf->file, "./file/test.txt");
+    conf->pool_size = 4000;
+    conf->append_size = 1024;
+    conf->thread_sum = 1;
+    conf->mode = BUFF_MODE;
+
+    return ;
+}
+
+typedef struct userdata_s
+{
+    struct timespec *finished;
+}userdata_t;
+
+static size_t  calc_time(struct timespec start, struct timespec end)
+{
+    return (end.tv_sec * 1000 * 1000 * 1000 + end.tv_nsec - 
+        (start.tv_sec * 1000 * 1000 * 1000 + start.tv_nsec));
+}
+
+int read_file_list(const char *path, char file_name[][256])
+{
+    DIR *dir;
+    struct dirent *ptr;
+    int path_len = strlen(path);
+    int file_num = 0;
+
+    if ((dir=opendir(path)) == NULL)
+    {
+        perror("Open dir error...");
+        exit(-1);
+    }
+    
+    while (((ptr=readdir(dir)) != NULL) && (file_num < 100))
+    {
+        if(strcmp(ptr->d_name,".")==0 || strcmp(ptr->d_name,"..")==0)    ///current dir OR parrent dir
+            continue;
+        else if((ptr->d_type == 8) || (ptr->d_type == 10))
+        {
+            memcpy(file_name[file_num], path, path_len);
+            strcat(file_name[file_num], ptr->d_name);
+        }
+        else if(ptr->d_type == 4)    ///dir
+        {
+            continue;
+        }
+        file_num++;
+    }
+    closedir(dir);
+    return 0;
+}
+
+static void callback(bool result, const char *error, void *userdata)
+{
+#if 0
+    userdata_t *data = (userdata_t *)userdata;
+    clock_gettime(CLOCK_MONOTONIC, data->finished);
+#endif
+    return ;
+}
+
+static int file_to_buffer(const char *file, char *buffer, size_t *len)
+{
+    FILE *fp = fopen(file, "r");
+    int num = 0;
+    *len = 0;
+    if (fp == NULL)
+    {
+        printf("fopen file failed:%s\n", file);
+        return -1;
+    }
+    do{
+        num = fread(&buffer[*len], 1, 4096, fp);
+        if (num < 0)
+        {
+            return -1;
+        }
+        *len += num;
+    }while(num == 4096);
+    fclose(fp);
+    return 0;
+}
+
+static int upload_file(char *file, char *buff, int buff_len, int test_times, thread_info_t *thread_info, char *performance_info)
+{
+    int i;
+    FILE *fp = NULL;
+    size_t fd[3000];
+    conf_t *conf = &thread_info->conf;
+    struct timespec tstart, tend, twrite;
+    size_t time_write, time_upload;
+    size_t len = strlen(performance_info);
+    char file_size[128];
+    //写文件
+    
+    clock_gettime(CLOCK_MONOTONIC, &tstart);
+    for (i = 0; i < test_times; i++)
+    {
+        fp = fopen(file, "w+");
+        if (fp == NULL)
+        {
+            printf("error:fopen failed\n");
+            return -1;
+        }
+        if (fwrite(buff, buff_len, 1, fp) != 1)
+        {
+            printf("error:fwrite failed\n");
+            fclose(fp);
+            return -1;
+        }
+        fclose(fp);
+    }
+    clock_gettime(CLOCK_MONOTONIC, &twrite);
+    time_write = calc_time(tstart, twrite);
+    time_write /= test_times;
+
+    //上传文件
+    clock_gettime(CLOCK_MONOTONIC, &tstart);
+    for (i = 0; i < test_times; i++)
+    {
+        fd[i] = hos_open_fd(thread_info->handle, conf->bucket, conf->object, callback, NULL, thread_info->thread_num, conf->mode);
+        if (hos_write(fd[i], file, 0, thread_info->thread_num, 0) != HOS_CLIENT_OK)
+        {
+            printf("error:hos_write file:%s\n", file);
+            return -1;
+        }
+    }
+    clock_gettime(CLOCK_MONOTONIC, &tend);
+    time_upload = calc_time(tstart, tend);
+    time_upload /= test_times;
+
+    sprintf(file_size, "%dk", buff_len / 1024);
+    sprintf(&performance_info[len], "%-20lu%-20s%-20lu%-20lu%-20lu\n",
+            thread_info->thread_num, file_size, time_write, time_upload, time_write + time_upload);
+
+    //hos_close_fd(fd, thread_info->thread_num);
+    
+    return 0;
+}
+
+static int upload_buff(char * buff, int buff_len, int test_times, thread_info_t *thread_info, char *performance_info)
+{
+    int i = 0;
+    int j = 0;
+    size_t fd = 0;
+    size_t tmp = 0;
+    size_t rest = 0;
+    struct timespec tstart, ttmp;
+    size_t time = 0;
+    size_t len;
+    conf_t *conf = &thread_info->conf;
+    char file_size[128];
+    char append_size[128];
+
+    fd = hos_open_fd(thread_info->handle, conf->bucket, conf->object, callback, NULL, thread_info->thread_num, conf->mode);
+#if 0
+    size_t time_tmp = 0;
+    while(1)
+    {
+        tmp = j * conf->append_size;
+        rest = buff_len - tmp;
+        if (rest < conf->append_size)
+        {
+            clock_gettime(CLOCK_MONOTONIC, &tstart);
+            for (i = 0; i < test_times; i++)
+            {
+                hos_write(fd, &buff[tmp], rest, thread_info->thread_num, j + 1);
+            }
+            clock_gettime(CLOCK_MONOTONIC, &ttmp);
+            time_tmp = calc_time(tstart, ttmp);
+            time_tmp /= test_times;
+            time += time_tmp;
+
+            sprintf(file_size, "%dk", buff_len / 1024);
+            sprintf(append_size, "%luk", conf->append_size / 1024);
+            len = strlen(performance_info);
+            sprintf(&performance_info[len], "%-20lu%-20s%-20s%-20lu%-20lu\n",
+                    thread_info->thread_num, file_size, append_size, time_tmp, time);
+            break;
+        }
+        clock_gettime(CLOCK_MONOTONIC, &tstart);
+        for (i = 0; i < test_times; i++)
+        {
+            hos_write(fd, &buff[tmp], conf->append_size, thread_info->thread_num, j + 1);
+        }
+        clock_gettime(CLOCK_MONOTONIC, &ttmp);
+        time_tmp = calc_time(tstart, ttmp);
+        time_tmp /= test_times;
+        time += time_tmp;
+
+        sprintf(file_size, "%dk", buff_len / 1024);
+        sprintf(append_size, "%luk", conf->append_size / 1024);
+        len = strlen(performance_info);
+        sprintf(&performance_info[len], "%-20lu%-20s%-20s%-20lu%-20lu\n",
+                thread_info->thread_num, file_size, append_size, time_tmp, time);
+
+        j++;
+    }
+#else
+    if (conf->mode & APPEND_MODE)
+    {
+        clock_gettime(CLOCK_MONOTONIC, &tstart);
+        for (i = 0; i < test_times; i++)
+        {
+            while(1)
+            {
+                tmp = j * conf->append_size;
+                rest = buff_len - tmp;
+                if (rest < conf->append_size)
+                {
+                    hos_write(fd, &buff[tmp], rest, thread_info->thread_num, j + 1);
+                    break;
+                }
+                hos_write(fd, &buff[tmp], conf->append_size, thread_info->thread_num, j + 1);
+                j++;
+            }
+        }
+        clock_gettime(CLOCK_MONOTONIC, &ttmp);
+        time = calc_time(tstart, ttmp);
+        time /= test_times;
+
+        sprintf(file_size, "%dk", buff_len / 1024);
+        sprintf(append_size, "%luk", conf->append_size / 1024);
+        len = strlen(performance_info);
+        sprintf(&performance_info[len], "%-20lu%-20s%-20s%-20d%-20lu\n",
+                thread_info->thread_num, file_size, append_size, 0, time);
+    }else
+    {
+        clock_gettime(CLOCK_MONOTONIC, &tstart);
+        for (i = 0; i < test_times; i++)
+        {
+            hos_write(fd, buff, buff_len, thread_info->thread_num, j + 1);
+        }
+        clock_gettime(CLOCK_MONOTONIC, &ttmp);
+        time = calc_time(tstart, ttmp);
+        time /= test_times;
+
+        sprintf(file_size, "%dk", buff_len / 1024);
+        sprintf(append_size, "%luk", conf->append_size / 1024);
+        len = strlen(performance_info);
+        sprintf(&performance_info[len], "%-20lu%-20s%-20d%-20d%-20lu\n",
+                thread_info->thread_num, file_size, 0, 0, time);
+    }
+
+
+#endif
+    hos_close_fd(fd, thread_info->thread_num);
+
+    return 0;
+}
+
+static void *put_object_thread(void *ptr)
+{
+    char *performance_info = NULL;
+    thread_info_t *thread_info = (thread_info_t *)ptr;
+    conf_t *conf = &thread_info->conf;
+    char file[128];
+    size_t buff_len;
+    int ret;
+    int i;
+    char *buff = NULL;
+    char file_name[100][256];
+    struct stat s_buf;
+    int test_times = 1;
+
+    buff = (char *)malloc(30 * 1024 * 1024);
+    if (buff == NULL)
+    {
+        perror(" ");
+        pthread_exit(NULL);
+    }
+
+    performance_info = (char *)malloc(1024 * 1024);
+    if (performance_info == NULL)
+    {
+        perror(" ");
+        free(buff);
+        pthread_exit(NULL);
+    }
+    memset(performance_info, 0, 10240);
+    memset(file_name, 0, 100 *256);
+
+    stat(conf->file, &s_buf);
+    if (S_ISDIR(s_buf.st_mode))
+    {
+        read_file_list(conf->file, file_name);
+        for (i = 0; i < 100; i++)
+        {
+            if (file_name[i][0] == '\0')
+                break;
+            ret = file_to_buffer(file_name[i], buff, &buff_len);
+            if (ret == -1)
+            {
+                free(buff);
+                free(performance_info);
+                pthread_exit(NULL);
+            }
+            if (conf->mode & BUFF_MODE)
+            {
+                upload_buff(buff, buff_len, test_times, thread_info, performance_info);
+            }else
+            {
+                sprintf(file, "./file/file_%lu_%d", thread_info->thread_num, i);
+                upload_file(file, buff, buff_len, test_times, thread_info, performance_info);
+            }
+        }
+    }else
+    {
+        ret = file_to_buffer(conf->file, buff, &buff_len);
+        if (ret == -1)
+        {
+            free(buff);
+            free(performance_info);
+            pthread_exit(NULL);
+        }
+
+        if (conf->mode & BUFF_MODE)
+        {
+            upload_buff(buff, buff_len, test_times, thread_info, performance_info);
+        }else
+        {
+            sprintf(file, "./file/file_%lu", thread_info->thread_num);
+            upload_file(file, buff, buff_len, test_times, thread_info, performance_info);
+        }
+    }
+    free(buff);
+    pthread_exit(performance_info);
+}
+
+int main(int argc, char *argv[])
+{
+    int ch;
+    int buf_size;
+    conf_t conf;
+    char *object;
+    char *retval;
+    size_t thread_num;
+    size_t thread[MAX_THREAD_NUM];
+    thread_info_t thread_info[MAX_THREAD_NUM];
+    cpu_set_t mask;
+    FILE *log = NULL;
+    char log_name[256];
+    const char *log_prefix = "./log/";
+    time_t timep;
+#if 0
+    if (argc <= 1)
+    {
+        printf("usage: HosClientPerformance \n[-e set endpoint] \n[-b set bucket] \n"
+                "[-o set object] \n[-f set file] \n[-p set pool size] \n"
+                "[-t set thread sum] \n[-B set BUFF_MODE] \n"
+                "[-F set FILE_MODE] \n[-A set APPEND_MODE] \n"
+                "[-h show help info] \n");
+        return -1;
+    }
+#endif
+    configuration_init(&conf);
+    //读取命令行配置
+    while((ch = getopt(argc, argv, "a:e:b:o:f:p:t:k:BFAh")) != -1)
+    {
+        switch(ch)
+        {
+            case 'a':
+                conf.append_size = 1024 * atoi(optarg);
+                break;
+            case 'e':
+                //endpoint
+                buf_size = MIN(STRING_SIZE,  strlen(optarg));
+                strncpy((char *)conf.endpoint, optarg, buf_size);
+                break; 
+            case 'b':
+                buf_size = MIN(STRING_SIZE,  strlen(optarg));
+                strncpy((char *)conf.bucket, optarg, buf_size);
+                conf.bucket[buf_size] = '\0';
+                break;          
+            case 'o':
+                buf_size = MIN(STRING_SIZE, strlen(optarg));
+                strncpy(conf.object, optarg, buf_size);
+                conf.object[buf_size] = '\0';
+                break;
+            case 'f':
+                buf_size = MIN(STRING_SIZE, strlen(optarg));
+                strncpy(conf.file, optarg, buf_size);
+                conf.file[buf_size] = '\0';
+                break;
+            case 'p':
+                conf.pool_size = atoi(optarg);
+                conf.pool_size = MIN(3000, conf.pool_size);
+                break;
+            case 't':
+                conf.thread_sum = atoi(optarg);
+                break;
+            case 'B':
+                conf.mode &= BUFF_MODE;
+                break;
+            case 'F':
+                conf.mode &= FILE_MODE;
+                break;
+            case 'A':
+                conf.mode |= APPEND_MODE;
+                break;
+            case 'k':
+                conf.append_size = 1024 * atoi(optarg);
+                break;
+            case 'h':
+            default:
+                printf("usage: HosClientPerformance \n[-e set endpoint] \n[-b set bucket] \n"
+                        "[-o set object] \n[-f set file] \n[-p set pool size] \n"
+                        "[-t set thread sum] \n[-B set BUFF_MODE] \n"
+                        "[-F set FILE_MODE] \n[-A set APPEND_MODE] \n"
+                        "[-h show help info] \n[-a set append size(K)]\n");
+                return -1;
+                break;
+        }
+    }
+
+    strcpy(log_name, log_prefix);
+    time(&timep);
+    strftime(&log_name[strlen(log_prefix)], sizeof(log_name) - strlen(log_prefix),"%Y%m%d%H%M%S.log", localtime(&timep));
+    log = fopen(log_name, "a+");
+    if (log == NULL)
+    {
+        perror(log_name);
+        return -1;
+    }
+    //创建client
+    hos_client_handle handle = hos_client_create(conf.endpoint, "default", "default", conf.thread_sum, conf.pool_size);
+    if (handle == NULL)
+    {
+        printf("error:hos_client_handle\n");
+        fclose(log);
+        return -1;
+    }
+
+    //创建bucket
+    if (hos_create_bucket(handle, conf.bucket))
+    {
+        printf("error:hos_create_bucket\n");
+        fclose(log);
+        hos_client_destory(handle);
+        return -1;
+    }
+
+    printf("\n==============================================================================================================================\n");
+    if (conf.mode & BUFF_MODE)
+    {
+        printf("%-20s%-20s%-20s%-20s%-20s\n", "thread_id", "file_size", "append_size", "upload_time", "total_time");
+    }else
+    {
+        printf("%-20s%-20s%-20s%-20s%-20s\n", "thread_id", "file_size", "write_time", "upload_time", "total_time");
+    }
+    for ( thread_num = 0; thread_num < conf.thread_sum; thread_num++ )
+    {
+        thread_info[thread_num].conf = conf;
+        object = thread_info[thread_num].conf.object;
+        sprintf(&object[strlen(object)], "-%lu", thread_num);
+        thread_info[thread_num].thread_num = thread_num;
+        thread_info[thread_num].handle = handle;
+        if(pthread_create(&thread[thread_num], NULL, put_object_thread, (void *)&thread_info[thread_num]))
+        {
+            perror(" ");
+            fclose(log);
+            hos_client_destory(handle);
+            return -1;
+        }
+
+        CPU_ZERO(&mask);
+        CPU_SET(thread_num + 8, &mask);
+        if (pthread_setaffinity_np(thread[thread_num], sizeof(mask), &mask) != 0)
+        {
+            printf("warning:could not set CPU affinity, continuing...\n");
+        }
+    }
+
+    for (thread_num = 0; thread_num < conf.thread_sum; thread_num++)
+    {
+        pthread_join(thread[thread_num], (void **)&retval);
+        if (retval)
+        {
+            printf("%s", retval);
+            fwrite(retval, strlen(retval), 1, log);
+            free(retval);
+        }
+    }
+
+    if (hos_client_destory(handle) == 0)
+    {
+        //time = calc_time(start, finished);
+        //time /= test_times;
+        //printf("hos upload finished spent %llu ns\n", time);
+    }
+
+    fclose(log);
+    return 0;
+}