Improve site benchmarks

hugolib/site_benchmark_test.go

@@ -27,6 +27,7 @@ import (
 type siteBuildingBenchmarkConfig struct {
 	Frontmatter  string
 	NumPages     int
+	NumLangs     int
 	RootSections int
 	Render       bool
 	Shortcodes   bool
@@ -39,6 +40,8 @@ func (s siteBuildingBenchmarkConfig) String() string {
 	// To make it a short as possible, we only shows bools when enabled and ints when >= 0 (RootSections > 1)
 	sep := ","
 	id := s.Frontmatter + sep
+	id += fmt.Sprintf("num_langs=%d%s", s.NumLangs, sep)
+
 	if s.RootSections > 1 {
 		id += fmt.Sprintf("num_root_sections=%d%s", s.RootSections, sep)
 	}
@@ -61,9 +64,10 @@ func (s siteBuildingBenchmarkConfig) String() string {
 	}
 
 	return strings.TrimSuffix(id, sep)
-
 }
 
+var someLangs = []string{"en", "fr", "nn"}
+
 func BenchmarkSiteBuilding(b *testing.B) {
 	var (
 		// The below represents the full matrix of benchmarks. Big!
@@ -74,6 +78,7 @@ func BenchmarkSiteBuilding(b *testing.B) {
 		allNumPages        = []int{1, 10, 100, 500, 1000, 5000, 10000}
 		allDoRender        = []bool{false, true}
 		allDoShortCodes    = []bool{false, true}
+		allNumLangs        = []int{1, 3}
 	)
 
 	var runDefault bool
@@ -99,6 +104,8 @@ func BenchmarkSiteBuilding(b *testing.B) {
 	}
 
 	var conf siteBuildingBenchmarkConfig
+	for _, numLangs := range allNumLangs {
+		conf.NumLangs = numLangs
 		for _, frontmatter := range allFrontmatters {
 			conf.Frontmatter = frontmatter
 			for _, rootSections := range allNumRootSections {
@@ -121,6 +128,7 @@ func BenchmarkSiteBuilding(b *testing.B) {
 				}
 			}
 		}
+	}
 }
 
 func doBenchMarkSiteBuilding(conf siteBuildingBenchmarkConfig, b *testing.B) {
@@ -199,11 +207,29 @@ baseURL = "http://example.com/blog"
 paginate = 10
 defaultContentLanguage = "en"
 
+[languages]
+%s
+
 [Taxonomies]
 tag = "tags"
 category = "categories"
 `
 
+	langConfigTemplate := `
+[languages.%s]
+languageName = "Lang %s"
+weight = %d
+`
+
+	langConfig := ""
+
+	for i := 0; i < cfg.NumLangs; i++ {
+		langCode := someLangs[i]
+		langConfig += fmt.Sprintf(langConfigTemplate, langCode, langCode, i+1)
+	}
+
+	siteConfig = fmt.Sprintf(siteConfig, langConfig)
+
 	numTags := cfg.NumTags
 
 	if cfg.TagsPerPage > numTags {
@@ -248,7 +274,12 @@ category = "categories"
 
 		fs := th.Fs
 
-		pagesPerSection := cfg.NumPages / cfg.RootSections
+		pagesPerSection := cfg.NumPages / cfg.RootSections / cfg.NumLangs
+		for li := 0; li < cfg.NumLangs; li++ {
+			fileLangCodeID := ""
+			if li > 0 {
+				fileLangCodeID = "." + someLangs[li] + "."
+			}
 
 			for i := 0; i < cfg.RootSections; i++ {
 				for j := 0; j < pagesPerSection; j++ {
@@ -278,7 +309,14 @@ category = "categories"
 
 					content := fmt.Sprintf(pageTemplate, fmt.Sprintf("Title%d_%d", i, j), tagsStr, contentPagesContent[rand.Intn(3)])
 
-				writeSource(b, fs, filepath.Join("content", fmt.Sprintf("sect%d", i), fmt.Sprintf("page%d.md", j)), content)
+					contentFilename := fmt.Sprintf("page%d%s.md", j, fileLangCodeID)
+
+					writeSource(b, fs, filepath.Join("content", fmt.Sprintf("sect%d", i), contentFilename), content)
+				}
+
+				content := fmt.Sprintf(pageTemplate, fmt.Sprintf("Section %d", i), "[]", contentPagesContent[rand.Intn(3)])
+				indexContentFilename := fmt.Sprintf("_index%s.md", fileLangCodeID)
+				writeSource(b, fs, filepath.Join("content", fmt.Sprintf("sect%d", i), indexContentFilename), content)
 			}
 		}