lib/massive_sitemap/builder/rotating.rb in massive_sitemap-2.0.0.rc3 vs lib/massive_sitemap/builder/rotating.rb in massive_sitemap-2.0.0.rc4
- old
+ new
@@ -5,21 +5,17 @@
module Builder
class Rotating < Base
NUM_URLS = 1..50_000
OPTS = Base::OPTS.merge(
- :max_per_sitemap => NUM_URLS.max
+ :max_urls => NUM_URLS.max
)
def initialize(writer, options = {}, &block)
@rotations = 0
@urls = 0
- if options[:max_per_sitemap] && !NUM_URLS.member?(options[:max_per_sitemap])
- raise ArgumentError, %Q(":max_per_sitemap" must be greater than #{NUM_URLS.min} and smaller than #{NUM_URLS.max})
- end
-
super
end
# On rotation, close current file, and reopen a new one
# with same file name but -<counter> appendend
@@ -32,10 +28,10 @@
header!(&block)
end
end
def add_url!(location, attrs = {})
- if @urls >= @options[:max_per_sitemap]
+ if @urls >= @options[:max_urls]
close!
end
super
@urls += 1
end