Fix and optimize request rate wait time

cloudflare · Nov 15, 2018 · a5dd590 · a5dd590
1 parent ebc9ec8
commit a5dd590
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/cmd/logshare-cli/main.go b/cmd/logshare-cli/main.go
@@ -119,15 +119,11 @@ func run(conf *config) func(c *cli.Context) error {
 			hourStart := conf.startTime
 			hourEnd := hourStart + 3600
 			var previousRequest time.Time
-			var requestGap time.Duration
 			for hourStart < conf.endTime {
 				if hourEnd > conf.endTime {
 					hourEnd = conf.endTime
 				}
-				requestGap = time.Since(previousRequest)
-				if requestGap.Seconds() < 5 {
-					time.Sleep(requestGap)
-				}
+				time.Sleep(5 * time.Second - time.Since(previousRequest))
 				previousRequest = time.Now()
 				meta, err = client.GetFromTimestamp(
 					conf.zoneID, hourStart, hourEnd, conf.count)