anonfunc/gositemap
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 4 Wiki Activity

make sitemap url limit adjustable

Browse Source
This commit is contained in:
Giovanni Harting
2023-03-04 15:59:06 +01:00
parent ec86a0396a
commit 4e3f22a528
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
sitemap.go
View File

@@ -46,8 +46,9 @@ type sitemapXML struct {
}
type SiteMap struct {
urls []*urlXML
BaseURL string
urls []*urlXML
BaseURL string
MaxURLPerSiteMap int
}
func (m *SiteMap) AddURL(url string, lastMod *string, freq *ChangeFrequency, prio *float64) {
@@ -64,8 +65,7 @@ func prefixXML(content []byte) []byte {
}
func (m *SiteMap) SiteMapIndex() ([]byte, error) {
nSiteMaps := int(math.Floor(float64(len(m.urls))/50000 + 1))
nSiteMaps := int(math.Floor(float64(len(m.urls))/float64(m.MaxURLPerSiteMap) + 1))
index := &siteMapIndexXML{}
for i := 0; i < nSiteMaps; i++ {
@@ -83,17 +83,17 @@ func (m *SiteMap) SiteMapIndex() ([]byte, error) {
}
func (m *SiteMap) SiteMap(id int) ([]byte, error) {
if 50000*(id) >= len(m.urls) {
if m.MaxURLPerSiteMap*(id) >= len(m.urls) {
return nil, nil
}
siteMap := &urlSetXML{}
var urls []*urlXML
if 50000*(id+1) >= len(m.urls) {
urls = m.urls[50000*id:]
if m.MaxURLPerSiteMap*(id+1) >= len(m.urls) {
urls = m.urls[m.MaxURLPerSiteMap*id:]
} else {
urls = m.urls[50000*id : 50000*(id+1)]
urls = m.urls[m.MaxURLPerSiteMap*id : m.MaxURLPerSiteMap*(id+1)]
}
siteMap.URLs = append(siteMap.URLs, urls...)