diff options
| author | chenwanli <chenwanli@sensetime.com> | 2019-03-20 16:38:11 +0800 |
|---|---|---|
| committer | chenwanli <chenwanli@sensetime.com> | 2019-03-20 16:38:11 +0800 |
| commit | 45a52b17fd83a9ffcda494de5cd800a93de97c9c (patch) | |
| tree | 41d3d20a43e79fbd5964731d8a667106a4f51d43 | |
| parent | 88ab932f7d439715698183fbdfbd4ade71596b67 (diff) | |
| download | seaweedfs-45a52b17fd83a9ffcda494de5cd800a93de97c9c.tar.xz seaweedfs-45a52b17fd83a9ffcda494de5cd800a93de97c9c.zip | |
benchmark: add replication flag
| -rw-r--r-- | weed/command/benchmark.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/weed/command/benchmark.go b/weed/command/benchmark.go index 44601e567..aa54946a7 100644 --- a/weed/command/benchmark.go +++ b/weed/command/benchmark.go @@ -36,6 +36,7 @@ type BenchmarkOptions struct { read *bool sequentialRead *bool collection *string + replication *string cpuprofile *string maxCpu *int grpcDialOption grpc.DialOption @@ -61,6 +62,7 @@ func init() { b.read = cmdBenchmark.Flag.Bool("read", true, "enable read") b.sequentialRead = cmdBenchmark.Flag.Bool("readSequentially", false, "randomly read by ids from \"-list\" specified file") b.collection = cmdBenchmark.Flag.String("collection", "benchmark", "write data to this collection") + b.replication = cmdBenchmark.Flag.String("replication", "000", "replication type") b.cpuprofile = cmdBenchmark.Flag.String("cpuprofile", "", "cpu profile output file") b.maxCpu = cmdBenchmark.Flag.Int("maxCpu", 0, "maximum number of CPUs. 0 means all available CPUs") sharedBytes = make([]byte, 1024) @@ -228,8 +230,9 @@ func writeFiles(idChan chan int, fileIdLineChan chan string, s *stat) { MimeType: "image/bench", // prevent gzip benchmark content } ar := &operation.VolumeAssignRequest{ - Count: 1, - Collection: *b.collection, + Count: 1, + Collection: *b.collection, + Replication: *b.replication, } if assignResult, err := operation.Assign(masterClient.GetMaster(), b.grpcDialOption, ar); err == nil { fp.Server, fp.Fid, fp.Collection = assignResult.Url, assignResult.Fid, *b.collection |