diff options
| author | chrislu <chris.lu@gmail.com> | 2024-06-14 11:42:05 -0700 |
|---|---|---|
| committer | chrislu <chris.lu@gmail.com> | 2024-06-14 11:42:05 -0700 |
| commit | 32dbb6a2ceb86a44d7879bdb506ab759d59f056b (patch) | |
| tree | 94c1d9cd5c086b437c4df2847b1a42b6432b535b /weed/command/benchmark.go | |
| parent | 70b31e3a28b8169289bb9c55052355a3daacfd12 (diff) | |
| parent | 8aae82dd71735df1cd915a47a9b9d453d393adf6 (diff) | |
| download | seaweedfs-32dbb6a2ceb86a44d7879bdb506ab759d59f056b.tar.xz seaweedfs-32dbb6a2ceb86a44d7879bdb506ab759d59f056b.zip | |
Merge branch 'master' into mq
Diffstat (limited to 'weed/command/benchmark.go')
| -rw-r--r-- | weed/command/benchmark.go | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/weed/command/benchmark.go b/weed/command/benchmark.go index 7f9a23cf8..2a0db47c2 100644 --- a/weed/command/benchmark.go +++ b/weed/command/benchmark.go @@ -2,6 +2,7 @@ package command import ( "bufio" + "context" "fmt" "github.com/seaweedfs/seaweedfs/weed/pb" "io" @@ -128,8 +129,9 @@ func runBenchmark(cmd *Command, args []string) bool { } b.masterClient = wdclient.NewMasterClient(b.grpcDialOption, "", "client", "", "", "", *pb.ServerAddresses(*b.masters).ToServiceDiscovery()) - go b.masterClient.KeepConnectedToMaster() - b.masterClient.WaitUntilConnected() + ctx := context.Background() + go b.masterClient.KeepConnectedToMaster(ctx) + b.masterClient.WaitUntilConnected(ctx) if *b.write { benchWrite() @@ -210,7 +212,7 @@ func writeFiles(idChan chan int, fileIdLineChan chan string, s *stat) { } var jwtAuthorization security.EncodedJwt if isSecure { - jwtAuthorization = operation.LookupJwt(b.masterClient.GetMaster(), b.grpcDialOption, df.fp.Fid) + jwtAuthorization = operation.LookupJwt(b.masterClient.GetMaster(context.Background()), b.grpcDialOption, df.fp.Fid) } if e := util.Delete(fmt.Sprintf("http://%s/%s", df.fp.Server, df.fp.Fid), string(jwtAuthorization)); e == nil { s.completed++ |