d/mutilate
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added support for big document sizes up to 8MB

Browse Source 
Added support for big document sizes up to 8MB
This commit is contained in:
Deepu Bhatia 2018-04-18 16:25:00 -07:00
parent d65c6ef7c2
commit 52e6f4acf2
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mutilate.cc
View File

@ -39,11 +39,12 @@
#include "util.h" #include "util.h"
#define MIN(a,b) ((a) < (b) ? (a) : (b)) #define MIN(a,b) ((a) < (b) ? (a) : (b))
#define MAX_VALUE_SIZE 1024 * 1024 * 8
using namespace std; using namespace std;
gengetopt_args_info args; gengetopt_args_info args;
char random_char[2 * 1024 * 1024]; // Buffer used to generate random values. char random_char[MAX_VALUE_SIZE]; // Buffer used to generate random values.
#ifdef HAVE_LIBZMQ #ifdef HAVE_LIBZMQ
vector<zmq::socket_t*> agent_sockets; vector<zmq::socket_t*> agent_sockets;
@ -421,8 +422,12 @@ int main(int argc, char **argv) {
if (args.quiet_given) log_level = QUIET; if (args.quiet_given) log_level = QUIET;
if (args.depth_arg < 1) DIE("--depth must be >= 1"); if (args.depth_arg < 1) DIE("--depth must be >= 1");
// if (args.valuesize_arg < 1 || args.valuesize_arg > 1024*1024)
// DIE("--valuesize must be >= 1 and <= 1024*1024"); if (atoi(args.valuesize_arg) < 1 || atoi(args.valuesize_arg) > MAX_VALUE_SIZE) {
DIE("\"--valuesize must be >= 1 and <= %i", MAX_VALUE_SIZE);
}
if (args.qps_arg < 0) DIE("--qps must be >= 0"); if (args.qps_arg < 0) DIE("--qps must be >= 0");
if (args.update_arg < 0.0 || args.update_arg > 1.0) if (args.update_arg < 0.0 || args.update_arg > 1.0)
DIE("--update must be >= 0.0 and <= 1.0"); DIE("--update must be >= 0.0 and <= 1.0");
@ -470,8 +475,9 @@ int main(int argc, char **argv) {
pthread_barrier_init(&barrier, NULL, options.threads); pthread_barrier_init(&barrier, NULL, options.threads);
vector<string> servers; vector<string> servers;
for (unsigned int s = 0; s < args.server_given; s++) for (unsigned int s = 0; s < args.server_given; s++) {
servers.push_back(name_to_ipaddr(string(args.server_arg[s]))); servers.push_back(name_to_ipaddr(string(args.server_arg[s])));
}
ConnectionStats stats; ConnectionStats stats;