feat: validate rate limit

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Jun 16, 2022 · a139aa3 · a139aa3
1 parent a7b4b92
commit a139aa3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/client/config/dfget.go b/client/config/dfget.go
@@ -164,7 +164,7 @@ func (cfg *ClientOption) Validate() error {
 	}
 
 	if int64(cfg.RateLimit.Limit) < DefaultMinRate.ToNumber() {
-		return fmt.Errorf("rate limit must be greater than %s", DefaultMinRate.String())
+		return errors.Wrapf(dferrors.ErrInvalidHeader, "rate limit must be greater than %s", DefaultMinRate.String())
 	}
 
 	return nil

diff --git a/client/config/dfget_linux.go b/client/config/dfget_linux.go
@@ -19,7 +19,10 @@
 
 package config
 
-import "d7y.io/dragonfly/v2/client/clientutil"
+import (
+	"d7y.io/dragonfly/v2/client/clientutil"
+	"golang.org/x/time/rate"
+)
 
 var dfgetConfig = ClientOption{
 	URL:     "",

diff --git a/client/config/peerhost.go b/client/config/peerhost.go
@@ -136,11 +136,11 @@ func (p *DaemonOption) Validate() error {
 	}
 
 	if int64(p.Download.TotalRateLimit.Limit) < DefaultMinRate.ToNumber() {
-		return fmt.Errorf("rate limit must be greater than %s", DefaultMinRate.String())
+		return errors.Errorf("rate limit must be greater than %s", DefaultMinRate.String())
 	}
 
 	if int64(p.Upload.RateLimit.Limit) < DefaultMinRate.ToNumber() {
-		return fmt.Errorf("rate limit must be greater than %s", DefaultMinRate.String())
+		return errors.Errorf("rate limit must be greater than %s", DefaultMinRate.String())
 	}
 
 	switch p.Download.DefaultPattern {