diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2022-08-11 19:53:15 +0000 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2022-08-11 19:53:15 +0000 |
commit | 50949b6a1e13276b467d219de23d5fb41383282e (patch) | |
tree | f5fc1c578aad1ce9ad88c6e15813eec3fb466c0b /fbench/src/fbench/fbench.cpp | |
parent | 046cae924701cf6446db73b12720194874daad01 (diff) |
Default content-type is json when using post.
Diffstat (limited to 'fbench/src/fbench/fbench.cpp')
-rw-r--r-- | fbench/src/fbench/fbench.cpp | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/fbench/src/fbench/fbench.cpp b/fbench/src/fbench/fbench.cpp index c69d4f5af18..8c3ea7b1eed 100644 --- a/fbench/src/fbench/fbench.cpp +++ b/fbench/src/fbench/fbench.cpp @@ -372,13 +372,18 @@ FBench::Main(int argc, char *argv[]) bool optError; optError = false; + std::string content_type = "Content-type:application/json"; while((opt = getopt(argc, argv, "H:A:T:C:K:Da:n:c:l:i:s:q:o:r:m:p:kdxyzP")) != -1) { switch(opt) { case 'A': authority = optarg; break; case 'H': - extraHeaders += std::string(optarg) + "\r\n"; + if (strncmp(optarg, "Content-type:", 13) == 0) { + content_type = optarg; + } else { + extraHeaders += std::string(optarg) + "\r\n"; + } if (strncmp(optarg, "Host:", 5) == 0) { fprintf(stderr, "Do not override 'Host:' header, use -A option instead\n"); return -1; @@ -458,6 +463,9 @@ FBench::Main(int argc, char *argv[]) break; } } + if (usePostMode) { + extraHeaders += content_type + "\r\n"; + } if ( argc < (optind + 2) || optError) { Usage(); @@ -477,8 +485,7 @@ FBench::Main(int argc, char *argv[]) short hosts = args / 2; - for (int i=0; i<hosts; ++i) - { + for (int i=0; i<hosts; ++i) { _hostnames.push_back(std::string(argv[optind+2*i])); int port = atoi(argv[optind+2*i+1]); if (port == 0) { @@ -490,9 +497,9 @@ FBench::Main(int argc, char *argv[]) // Find offset for each client if shared query file _queryfileOffset.push_back(0); + char filename[1024]; if (singleQueryFile) { // Open file to find offsets, with pattern as if client 0 - char filename[1024]; snprintf(filename, 1024, queryFilePattern, 0); queryFilePattern = filename; FileReader reader; |