Include rate limiting and stats in BenchmarkConnectionMainReadLoop

[btrtrc.git] / client.go

diff --git a/client.go b/client.go
index 8b8f9b4f86bd90db09df81f46bba4a42fb9c5f02..5f28b12d078833232158e2920991316a4813a4ef 100644 (file)
--- a/client.go
+++ b/client.go
@@ -192,12 +192,12 @@ func NewClient(cfg *Config) (cl *Client, err error) {
                cl.Close()
        }()
        if cfg.UploadRateLimiter == nil {
-               cl.uploadLimit = rate.NewLimiter(rate.Inf, 0)
+               cl.uploadLimit = unlimited
        } else {
                cl.uploadLimit = cfg.UploadRateLimiter
        }
        if cfg.DownloadRateLimiter == nil {
-               cl.downloadLimit = rate.NewLimiter(rate.Inf, 0)
+               cl.downloadLimit = unlimited
        } else {
                cl.downloadLimit = cfg.DownloadRateLimiter
        }