Go: How to Generate sitemap.xml

sitemap.go

package main

import (
	"encoding/xml"
	"io"
)

const defaultXmlNs = "http://www.sitemaps.org/schemas/sitemap/0.9"

type Url struct {
	Loc string `xml:"loc"`
}

type Sitemap struct {
	XMLName xml.Name `xml:"urlset"`
	XmlNs   string   `xml:"xmlns,attr"`
	Url     []*Url   `xml:"url"`
}

func NewSitemap() *Sitemap {
	return &Sitemap{XmlNs: defaultXmlNs}
}

func (s *Sitemap) AddUrl(u string) {
	s.Url = append(s.Url, &Url{u})
}

func (s *Sitemap) Encode(w io.Writer) error {
	_, err := w.Write([]byte(xml.Header))
	if err != nil {
		return err
	}
	return xml.NewEncoder(w).Encode(s)
}