diff options
Diffstat (limited to 'modules/ratelimit')
| -rw-r--r-- | modules/ratelimit/ratelimit.go | 5 | ||||
| -rw-r--r-- | modules/ratelimit/ratelimit_test.go | 2 |
2 files changed, 1 insertions, 6 deletions
diff --git a/modules/ratelimit/ratelimit.go b/modules/ratelimit/ratelimit.go index b02f5d5..be622f6 100644 --- a/modules/ratelimit/ratelimit.go +++ b/modules/ratelimit/ratelimit.go @@ -21,10 +21,6 @@ type Module struct { semaphore chan struct{} } -func (m *Module) ID() string { - return "ratelimit" -} - func (m *Module) OnLoad(v flyscrape.Visitor) { rate := time.Duration(float64(time.Second) / m.Rate) @@ -47,7 +43,6 @@ func (m *Module) OnComplete() { } var ( - _ flyscrape.Module = (*Module)(nil) _ flyscrape.OnRequest = (*Module)(nil) _ flyscrape.OnLoad = (*Module)(nil) _ flyscrape.OnComplete = (*Module)(nil) diff --git a/modules/ratelimit/ratelimit_test.go b/modules/ratelimit/ratelimit_test.go index c166371..5e91f8f 100644 --- a/modules/ratelimit/ratelimit_test.go +++ b/modules/ratelimit/ratelimit_test.go @@ -17,7 +17,7 @@ import ( func TestRatelimit(t *testing.T) { scraper := flyscrape.NewScraper() - scraper.LoadModule(&starturl.Module{URL: "http://www.example.com/"}) + scraper.LoadModule(&starturl.Module{URL: "http://www.example.com"}) scraper.LoadModule(&followlinks.Module{}) scraper.LoadModule(&ratelimit.Module{ Rate: 100, |