diff options
| author | chenwanli <chenwanli@sensetime.com> | 2019-01-10 16:43:44 +0800 |
|---|---|---|
| committer | chenwanli <chenwanli@sensetime.com> | 2019-01-10 19:41:03 +0800 |
| commit | 0a3e83a36a067a282a3cecec06d7bb577f1822ba (patch) | |
| tree | 01ff3a3bf86f1cf81a34e0bc958ba1b0f93138b9 /weed/operation/stats.go | |
| parent | 2a75a36b275ea7f47a95ad2efc70be22d5ed3fe2 (diff) | |
| download | seaweedfs-0a3e83a36a067a282a3cecec06d7bb577f1822ba.tar.xz seaweedfs-0a3e83a36a067a282a3cecec06d7bb577f1822ba.zip | |
Set timeout for master and volume non-streaming rpc
Diffstat (limited to 'weed/operation/stats.go')
| -rw-r--r-- | weed/operation/stats.go | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/weed/operation/stats.go b/weed/operation/stats.go index 277b0e835..364727272 100644 --- a/weed/operation/stats.go +++ b/weed/operation/stats.go @@ -2,6 +2,7 @@ package operation import ( "context" + "time" "github.com/chrislusf/seaweedfs/weed/pb/master_pb" ) @@ -9,7 +10,10 @@ import ( func Statistics(server string, req *master_pb.StatisticsRequest) (resp *master_pb.StatisticsResponse, err error) { err = withMasterServerClient(server, func(masterClient master_pb.SeaweedClient) error { - grpcResponse, grpcErr := masterClient.Statistics(context.Background(), req) + ctx, cancel := context.WithTimeout(context.Background(), time.Duration(5*time.Second)) + defer cancel() + + grpcResponse, grpcErr := masterClient.Statistics(ctx, req) if grpcErr != nil { return grpcErr } |