update lazymap

add rate limitting for NotFound
webtor-io · Dec 17, 2024 · f667670 · f667670
1 parent b3ca0b3
commit f667670
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/services/store.go b/services/store.go
@@ -98,6 +98,7 @@ func (s *Store) getRate(h string) *atomic.Int64 {
 
 func (s *Store) touch(ctx context.Context, h string) (ok bool, err error) {
 	if !s.checkRate(h) {
+		s.incRate(h)
 		log.WithField("infohash", h).Warn("get rate limit")
 		return false, ErrNotFound
 	}
@@ -128,6 +129,7 @@ func (s *Store) touch(ctx context.Context, h string) (ok bool, err error) {
 
 func (s *Store) pull(ctx context.Context, h string, start int) (torrent []byte, err error) {
 	if !s.checkRate(h) {
+		s.incRate(h)
 		log.WithField("infohash", h).Warn("get rate limit")
 		return nil, ErrNotFound
 	}