生成sitemap.xml示范demo

package services

import (
	"encoding/xml"
	"fmt"
	"log"
	"os"
	"github.com/gin-gonic/gin"
	"time"
	model "btcinfo/models"
	"context"
	"math" //向上取整需要
	"go.mongodb.org/mongo-driver/bson"

)

// URL 结构体定义了 sitemap.xml 文件中的每个 URL 条目
type URL struct {
	XMLName   xml.Name `xml:"url"`
	Loc       string   `xml:"loc"`
	LastMod   string   `xml:"lastmod,omitempty"`
	ChangeFreq string   `xml:"changefreq,omitempty"`
	Priority  float64  `xml:"priority,omitempty"`
}

// Sitemap 结构体定义了 sitemap.xml 文件的根元素
type Sitemap struct {
	XMLName xml.Name `xml:"urlset"`
	Xmlns   string   `xml:"xmlns,attr"`
	URLs    []URL    `xml:"url"`
}





//创建函数
func UpSitemap(c *gin.Context) {
	var Urls []URL
	log.Print("sitemap运行")

	//遍历语言代码并追加
	for _, lang := range Langs {
		//首页
		index := URL{
			Loc:       "https://info.com/"+lang,
			LastMod:   "2024-03-28",
			ChangeFreq: "yearly",
			Priority: 0.6,
		}
		Urls = append(Urls, index) //追加

		//列表faq
		//查询条件
		filter := bson.D{{Key: "dalei", Value: "faq"},{Key: "lang", Value: lang}}
		//取当前条件的总条数
		zongtiaoshu, _ := model.FaqModel.CountDocuments(context.TODO(), filter)
		//根据总条数计算总页数
		zongyeshu := int64(math.Ceil(float64(zongtiaoshu) / 12))
		log.Print("faq总页数",zongyeshu)
		//循环
		for i := int64(0); i < zongyeshu; i++ {
			log.Print("循环",i+1)
			lists1 := URL{
				Loc:       "https://info.com/"+lang+"/lists/faq/" +fmt.Sprint((i+1)),
				LastMod:   "2024-03-28",
				ChangeFreq: "weekly",
				Priority: 0.3,
			}
			Urls = append(Urls, lists1) //追加
		}

		//文章列表
		//查询条件
		filter = bson.D{{Key: "dalei", Value: "article"},{Key: "lang", Value: lang}}
		//取当前条件的总条数
		zongtiaoshu, _ = model.ArticleModel.CountDocuments(context.TODO(), filter)
		//根据总条数计算总页数
		zongyeshu = int64(math.Ceil(float64(zongtiaoshu) / 12))
		log.Print("文章总页数",zongyeshu)
		//循环
		for i := int64(0); i < zongyeshu; i++ {
			log.Print("循环",i+1)
			lists1 := URL{
				Loc:       "https://info.com/"+lang+"/lists/article/" +fmt.Sprint((i+1)),
				LastMod:   "2024-03-28",
				ChangeFreq: "weekly",
				Priority: 0.3,
			}
			Urls = append(Urls, lists1) //追加
		}

		//文章faq
		//定义一个空的切片来存储查询结果
		var results []model.Article_modelcha
		//查询条件
		filter = bson.D{{Key: "lang", Value: lang}}
		// 执行查询操作
		cursor, err := model.FaqModel.Find(context.TODO(), filter)
		if err != nil {
			c.JSON(400, gin.H{"db_error": "FAQ表查询错误"})
			log.Print("faq表查询错误:", err)
			return
		}
		//将上面的查询结果转码赋值给变量&results
		if err = cursor.All(context.TODO(), &results); err != nil {
			// 处理解码错误
			c.JSON(500, gin.H{"db_error": err})
			log.Print("路由查询解码错误:", err)
			return
		}

		//遍历文章
		for _, r := range results {
			var date time.Time //判断是否存在更新时间
			if r.Update != (time.Time{}) {
				date = r.Update
			} else {
				date = r.CreatedAt
			}
			
			list := URL{
				Loc:       "https://info.com/" + lang + "/article/faq/" + r.ID.Hex(),
				LastMod:   date.Format("2006-01-02"),
				ChangeFreq: "monthly",
				Priority: 0.7,
			}
			Urls = append(Urls, list) //追加
			
		}


		//文章

		// 执行查询操作
		cursor, err = model.ArticleModel.Find(context.TODO(), filter)
		if err != nil {
			c.JSON(400, gin.H{"db_error": "FAQ表查询错误"})
			log.Print("faq表查询错误:", err)
			return
		}
		//将上面的查询结果转码赋值给变量&results
		if err = cursor.All(context.TODO(), &results); err != nil {
			// 处理解码错误
			c.JSON(500, gin.H{"db_error": err})
			log.Print("路由查询解码错误:", err)
			return
		}

		//遍历文章
		for _, r := range results {
			var date time.Time //判断是否存在更新时间
			if r.Update != (time.Time{}) {
				date = r.Update
			} else {
				date = r.CreatedAt
			}
			
			list := URL{
				Loc:       "https://info.com/" + lang + "/article/article/" + r.ID.Hex(),
				LastMod:   date.Format("2006-01-02"),
				ChangeFreq: "monthly",
				Priority: 0.8,
			}
			Urls = append(Urls, list) //追加
			
		}

	}

	// 创建一个新的 sitemap 结构体
	sitemap := Sitemap{
		Xmlns: "http://www.sitemaps.org/schemas/sitemap/0.9",
		URLs: Urls,
	}

	// 创建一个新的文件并打开以供写入
	file, err := os.Create("jingtai/sitemap.xml")
	if err != nil {
		log.Println("Error creating file:", err)
		return
	}
	defer file.Close()

	// 使用 xml 包将 sitemap 结构体编码为 XML 格式，并写入到文件中
	encoder := xml.NewEncoder(file)
	encoder.Indent("", "  ")
	if err := encoder.Encode(sitemap); err != nil {
		log.Println("写入Sitemap.xml失败:", err)
		return
	}

	log.Println("Sitemap.xml更新完成!")
}