diff --git a/src/main/java/com/redfin/sitemapgenerator/AbstractSitemapGeneratorOptions.java b/src/main/java/com/redfin/sitemapgenerator/AbstractSitemapGeneratorOptions.java index 502fa98..91b0017 100644 --- a/src/main/java/com/redfin/sitemapgenerator/AbstractSitemapGeneratorOptions.java +++ b/src/main/java/com/redfin/sitemapgenerator/AbstractSitemapGeneratorOptions.java @@ -9,6 +9,7 @@ abstract class AbstractSitemapGeneratorOptions urls = new ArrayList(); private final W3CDateFormat dateFormat; @@ -40,6 +41,7 @@ abstract class SitemapGenerator write() { if (finished) throw new RuntimeException("Sitemap already printed; you must create a new generator to make more sitemaps"); - if (urls.size() == 0 && mapCount == 0) throw new RuntimeException("No URLs added, sitemap would be empty; you must add some URLs with addUrls"); + if (!allowEmptySitemap && urls.isEmpty() && mapCount == 0) throw new RuntimeException("No URLs added, sitemap would be empty; you must add some URLs with addUrls"); writeSiteMap(); finished = true; return outFiles; @@ -221,7 +223,7 @@ abstract class SitemapGenerator 0 || !allowEmptySitemap)) return; String fileNamePrefix; if (mapCount > 0) { fileNamePrefix = this.fileNamePrefix + mapCount; diff --git a/src/main/java/com/redfin/sitemapgenerator/SitemapIndexGenerator.java b/src/main/java/com/redfin/sitemapgenerator/SitemapIndexGenerator.java index 5ed204a..d426f36 100644 --- a/src/main/java/com/redfin/sitemapgenerator/SitemapIndexGenerator.java +++ b/src/main/java/com/redfin/sitemapgenerator/SitemapIndexGenerator.java @@ -19,6 +19,7 @@ import org.xml.sax.SAXException; public class SitemapIndexGenerator { private final URL baseUrl; private final File outFile; + private final boolean allowEmptyIndex; private final ArrayList urls = new ArrayList(); private final int maxUrls; private final W3CDateFormat dateFormat; @@ -32,6 +33,7 @@ public class SitemapIndexGenerator { private URL baseUrl; private File outFile; private W3CDateFormat dateFormat = null; + private boolean allowEmptyIndex = false; private int maxUrls = MAX_SITEMAPS_PER_INDEX; private Date defaultLastMod = new Date(); private boolean autoValidate = false; @@ -59,6 +61,18 @@ public class SitemapIndexGenerator { this.dateFormat = dateFormat; return this; } + + /** + * Permit writing an index that contains no URLs. + * + * @param allowEmptyIndex {@code true} if an empty index is permissible + * @return this instance, for chaining + */ + public Options allowEmptyIndex(boolean allowEmptyIndex) { + this.allowEmptyIndex = allowEmptyIndex; + return this; + } + /** * The maximum number of sitemaps to allow per sitemap index; the default is the * maximum allowed (1,000), but you can decrease it if you wish (for testing) @@ -116,6 +130,7 @@ public class SitemapIndexGenerator { private SitemapIndexGenerator(Options options) { this.baseUrl = options.baseUrl; this.outFile = options.outFile; + this.allowEmptyIndex = options.allowEmptyIndex; this.maxUrls = options.maxUrls; W3CDateFormat dateFormat = options.dateFormat; if (dateFormat == null) dateFormat = new W3CDateFormat(); @@ -206,7 +221,7 @@ public class SitemapIndexGenerator { /** Writes out the sitemap index */ public void write() { - if (urls.size() == 0) throw new RuntimeException("No URLs added, sitemap index would be empty; you must add some URLs with addUrls"); + if (!allowEmptyIndex && urls.isEmpty()) throw new RuntimeException("No URLs added, sitemap index would be empty; you must add some URLs with addUrls"); try { // TODO gzip? is that legal for a sitemap index? FileWriter out = new FileWriter(outFile); diff --git a/src/test/java/com/redfin/sitemapgenerator/SitemapGeneratorTest.java b/src/test/java/com/redfin/sitemapgenerator/SitemapGeneratorTest.java index cea3bd2..b64d740 100644 --- a/src/test/java/com/redfin/sitemapgenerator/SitemapGeneratorTest.java +++ b/src/test/java/com/redfin/sitemapgenerator/SitemapGeneratorTest.java @@ -347,6 +347,24 @@ public class SitemapGeneratorTest extends TestCase { assertEquals("First string didn't match", SITEMAP1, siteMapsAsStrings.get(0)); assertEquals("Second string didn't match", SITEMAP_PLUS_ONE, siteMapsAsStrings.get(1)); } + + public void testWriteEmptySitemap() throws Exception { + wsg = WebSitemapGenerator.builder("http://www.example.com", dir).allowEmptySitemap(true).build(); + String expected = "\n" + + "\n" + + ""; + String sitemap = writeSingleSiteMap(wsg); + assertEquals(expected, sitemap); + } + + public void testMaxUrlsAllowingEmptyDoesNotWriteExtraSitemap() throws Exception { + wsg = WebSitemapGenerator.builder("http://www.example.com", dir).allowEmptySitemap(true).maxUrls(10).build(); + for (int i = 0; i < 10; i++) { + wsg.addUrl("http://www.example.com/"+i); + } + String sitemap = writeSingleSiteMap(wsg); + assertEquals(SITEMAP1, sitemap); + } private String writeSingleSiteMap(WebSitemapGenerator wsg) { List files = wsg.write(); diff --git a/src/test/java/com/redfin/sitemapgenerator/SitemapIndexGeneratorTest.java b/src/test/java/com/redfin/sitemapgenerator/SitemapIndexGeneratorTest.java index 5b03f7e..70ab13e 100644 --- a/src/test/java/com/redfin/sitemapgenerator/SitemapIndexGeneratorTest.java +++ b/src/test/java/com/redfin/sitemapgenerator/SitemapIndexGeneratorTest.java @@ -86,6 +86,16 @@ public class SitemapIndexGeneratorTest extends TestCase { fail("Allowed write with no URLs"); } catch (RuntimeException e) {} } + + public void testNoUrlsEmptyIndexAllowed() throws Exception { + sig = new SitemapIndexGenerator.Options(EXAMPLE, outFile).allowEmptyIndex(true).build(); + sig.write(); + String expected = "\n" + + "\n" + + ""; + String actual = TestUtil.slurpFileAndDelete(outFile); + assertEquals(expected, actual); + } public void testMaxUrls() throws Exception { sig = new SitemapIndexGenerator.Options(EXAMPLE, outFile).autoValidate(true)