feat: add --delay flag to allow sleep between search requests

closes l3uddz#7
rubicon · Oct 22, 2021 · bdaf478 · bdaf478
1 parent ed343dc
commit bdaf478
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/cmd/missarr/radarr.go b/cmd/missarr/radarr.go
@@ -14,6 +14,7 @@ type RadarrCmd struct {
 	LastSearched    time.Duration `default:"672h" help:"How long before an item can be searched again"`
 	LastReleaseDate time.Duration `default:"72h" help:"How long before an item can be considered missing based on release date"`
 	SkipRefresh     bool          `default:"false" help:"Retrieve current missing from radarr"`
+	Delay           time.Duration `default:"0s" help:"Delay between search requests"`
 }
 
 func (r *RadarrCmd) Run(c *config, db *sql.DB, mg *migrate.Migrator) error {
@@ -95,6 +96,11 @@ func (r *RadarrCmd) Run(c *config, db *sql.DB, mg *migrate.Migrator) error {
 			return fmt.Errorf("update store: %w", err)
 		}
 
+		// sleep for delay
+		if r.Limit > 0 {
+			time.Sleep(r.Delay)
+		}
+
 		// decrease limit
 		r.Limit--
 	}

diff --git a/cmd/missarr/sonarr.go b/cmd/missarr/sonarr.go
@@ -15,6 +15,7 @@ type SonarrCmd struct {
 	LastAirDate  time.Duration `default:"72h" help:"How long before an item can be considered missing based on air date"`
 	AllowSpecial bool          `default:"false" help:"Allow specials to be considered missing"`
 	SkipRefresh  bool          `default:"false" help:"Retrieve current missing from sonarr"`
+	Delay        time.Duration `default:"0s" help:"Delay between search requests"`
 }
 
 func (r *SonarrCmd) Run(c *config, db *sql.DB, mg *migrate.Migrator) error {
@@ -98,6 +99,11 @@ func (r *SonarrCmd) Run(c *config, db *sql.DB, mg *migrate.Migrator) error {
 			return fmt.Errorf("update store: %w", err)
 		}
 
+		// sleep for delay
+		if r.Limit > 0 {
+			time.Sleep(r.Delay)
+		}
+
 		// decrease limit
 		r.Limit--
 	}