Skip to content

Commit

Permalink
tpl/partials: Make sure a cached partial is invoked only once
Browse files Browse the repository at this point in the history
This commit revises the locking strategy for `partialCached`. We have added a benchmark that may be a little artificial, but it should at least show that we're not losing any performance over this:

```bash
name              old time/op    new time/op    delta
IncludeCached-10    12.2ms ± 2%    11.3ms ± 1%   -7.36%  (p=0.029 n=4+4)

name              old alloc/op   new alloc/op   delta
IncludeCached-10    7.17MB ± 0%    5.09MB ± 0%  -29.00%  (p=0.029 n=4+4)

name              old allocs/op  new allocs/op  delta
IncludeCached-10      128k ± 1%       70k ± 0%  -45.42%  (p=0.029 n=4+4)
```

This commit also revises the template metrics hints logic a little, and add a test for it, which output is currently this:

```bash
 cumulative       average       maximum      cache  percent  cached  total
       duration      duration      duration  potential   cached   count  count  template
     ----------      --------      --------  ---------  -------  ------  -----  --------
      163.334µs     163.334µs     163.334µs          0        0       0      1  index.html
       23.749µs       5.937µs      19.916µs         25       50       2      4  partials/dynamic1.html
        9.625µs       4.812µs        6.75µs        100       50       1      2  partials/static1.html
        7.625µs       7.625µs       7.625µs        100        0       0      1  partials/static2.html
```

Some notes:

* The duration now includes the cached invocations (which should be very short)
* A cached template gets executed once before it gets cached, so the "percent cached" will never be 100.

Fixes gohugoio#4086
Fixes gohugoio#9506
  • Loading branch information
bep committed Feb 16, 2022
1 parent d162283 commit 9e586dc
Show file tree
Hide file tree
Showing 6 changed files with 168 additions and 16 deletions.
4 changes: 2 additions & 2 deletions htesting/hqt/checkers.go
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ type stringChecker struct {
argNames
}

// Check implements Checker.Check by checking that got and args[0] represents the same normalized text (whitespace etc. trimmed).
// Check implements Checker.Check by checking that got and args[0] represents the same normalized text (whitespace etc. removed).
func (c *stringChecker) Check(got interface{}, args []interface{}, note func(key string, value interface{})) (err error) {
s1, s2 := cast.ToString(got), cast.ToString(args[0])

Expand All @@ -83,7 +83,7 @@ func (c *stringChecker) Check(got interface{}, args []interface{}, note func(key
func normalizeString(s string) string {
lines := strings.Split(strings.TrimSpace(s), "\n")
for i, line := range lines {
lines[i] = strings.TrimSpace(line)
lines[i] = strings.Join(strings.Fields(strings.TrimSpace(line)), "")
}

return strings.Join(lines, "\n")
Expand Down
12 changes: 9 additions & 3 deletions metrics/metrics.go
Original file line number Diff line number Diff line change
Expand Up @@ -201,9 +201,15 @@ type result struct {

type bySum []result

func (b bySum) Len() int { return len(b) }
func (b bySum) Swap(i, j int) { b[i], b[j] = b[j], b[i] }
func (b bySum) Less(i, j int) bool { return b[i].sum > b[j].sum }
func (b bySum) Len() int { return len(b) }
func (b bySum) Swap(i, j int) { b[i], b[j] = b[j], b[i] }
func (b bySum) Less(i, j int) bool {
// Make the sort more stable for testing.
if b[i].sum.Milliseconds() == b[j].sum.Milliseconds() {
return b[i].key < b[j].key
}
return b[i].sum > b[j].sum
}

// howSimilar is a naive diff implementation that returns
// a number between 0-100 indicating how similar a and b are.
Expand Down
4 changes: 2 additions & 2 deletions tpl/partials/init.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,14 +18,14 @@ import (
"github.com/gohugoio/hugo/tpl/internal"
)

const name = "partials"
const namespaceName = "partials"

func init() {
f := func(d *deps.Deps) *internal.TemplateFuncsNamespace {
ctx := New(d)

ns := &internal.TemplateFuncsNamespace{
Name: name,
Name: namespaceName,
Context: func(args ...interface{}) (interface{}, error) { return ctx, nil },
}

Expand Down
2 changes: 1 addition & 1 deletion tpl/partials/init_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ func TestInit(t *testing.T) {
BuildStartListeners: &deps.Listeners{},
Log: loggers.NewErrorLogger(),
})
if ns.Name == name {
if ns.Name == namespaceName {
found = true
break
}
Expand Down
133 changes: 133 additions & 0 deletions tpl/partials/integration_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,13 @@
package partials_test

import (
"bytes"
"fmt"
"regexp"
"strings"
"testing"

"github.com/gohugoio/hugo/htesting/hqt"
"github.com/gohugoio/hugo/hugolib"
)

Expand Down Expand Up @@ -68,3 +73,131 @@ partialCached: foo
partialCached: foo
`)
}

func TestIncludeCacheHints(t *testing.T) {
t.Parallel()

files := `
-- config.toml --
baseURL = 'http://example.com/'
templateMetrics=true
templateMetricsHints=true
disableKinds = ["page", "section", "taxonomy", "term", "sitemap"]
[outputs]
home = ["HTML"]
-- layouts/index.html --
{{ partials.IncludeCached "static1.html" . }}
{{ partials.IncludeCached "static1.html" . }}
{{ partials.Include "static2.html" . }}
Expext: Total count 4, cache count 2.
D1I: {{ partials.Include "dynamic1.html" . }}
D1C: {{ partials.IncludeCached "dynamic1.html" . }}
D1C: {{ partials.IncludeCached "dynamic1.html" . }}
D1C: {{ partials.IncludeCached "dynamic1.html" . }}
-- layouts/partials/static1.html --
P1
-- layouts/partials/static2.html --
P2
-- layouts/partials/dynamic1.html --
{{ now.UnixNano }}
-- layouts/partials/dynamic2.html --
{{ now.UnixNano }}
-- layouts/partials/halfdynamic1.html --
D1
{{ now.UnixNano }}
-- layouts/partials/halfdynamic2.html --
D2
{{ now.UnixNano }}
`

b := hugolib.NewIntegrationTestBuilder(
hugolib.IntegrationTestConfig{
T: t,
TxtarString: files,
},
).Build()

// fmt.Println(b.FileContent("public/index.html"))

var buf bytes.Buffer
b.H.Metrics.WriteMetrics(&buf)

got := buf.String()

// Get rid of all the durations, they are never the same.
durationRe := regexp.MustCompile(`\b.*(ms|µs)\b`)

normalize := func(s string) string {
s = durationRe.ReplaceAllString(s, "")
lines := strings.Split(s, "\n")
var result strings.Builder
for i, l := range lines {
if i < 3 {
continue
}
l = strings.TrimSpace(l)
if l == "" {
continue
}

result.WriteString(l + "\n")
}

return result.String()
}

got = normalize(got)

expect := `
0 0 0 1 index.html
25 50 2 4 partials/dynamic1.html
100 50 1 2 partials/static1.html
100 0 0 1 partials/static2.html
`

b.Assert(got, hqt.IsSameString, expect)
}

// gobench --package ./tpl/partials
func BenchmarkIncludeCached(b *testing.B) {
files := `
-- config.toml --
baseURL = 'http://example.com/'
-- layouts/index.html --
-- layouts/_default/single.html --
{{ partialCached "heavy.html" "foo" }}
-- layouts/partials/heavy.html --
{{ $result := slice }}
{{ range site.RegularPages }}
{{ $result = $result | append (dict "title" .Title "link" .RelPermalink "readingTime" .ReadingTime) }}
{{ end }}
{{ range $result }}
* {{ .title }} {{ .link }} {{ .readingTime }}
{{ end }}
`

for i := 1; i < 100; i++ {
files += fmt.Sprintf("\n-- content/p%d.md --\n---\ntitle: page\n---\n"+strings.Repeat("FOO ", i), i)
}

cfg := hugolib.IntegrationTestConfig{
T: b,
TxtarString: files,
}
builders := make([]*hugolib.IntegrationTestBuilder, b.N)

for i, _ := range builders {
builders[i] = hugolib.NewIntegrationTestBuilder(cfg)
}

b.ResetTimer()

for i := 0; i < b.N; i++ {
builders[i].Build()
}
}
29 changes: 21 additions & 8 deletions tpl/partials/partials.go
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ import (
"reflect"
"strings"
"sync"
"time"

texttemplate "github.com/gohugoio/hugo/tpl/internal/go_templates/texttemplate"

Expand All @@ -44,6 +45,13 @@ type partialCacheKey struct {
variant interface{}
}

func (k partialCacheKey) templateName() string {
if !strings.HasPrefix(k.name, "partials/") {
return "partials/" + k.name
}
return k.name
}

// partialCache represents a cache of partials protected by a mutex.
type partialCache struct {
sync.RWMutex
Expand Down Expand Up @@ -211,6 +219,7 @@ func createKey(name string, variants ...interface{}) (partialCacheKey, error) {
var errUnHashable = errors.New("unhashable")

func (ns *Namespace) getOrCreate(key partialCacheKey, context interface{}) (result interface{}, err error) {
start := time.Now()
defer func() {
if r := recover(); r != nil {
err = r.(error)
Expand All @@ -226,24 +235,28 @@ func (ns *Namespace) getOrCreate(key partialCacheKey, context interface{}) (resu
ns.cachedPartials.RUnlock()

if ok {
if ns.deps.Metrics != nil {
ns.deps.Metrics.TrackValue(key.templateName(), p, true)
// The templates that gets executed is measued in Execute.
// We need to track the time spent in the cache to
// get the totals correct.
ns.deps.Metrics.MeasureSince(key.templateName(), start)

}
return p, nil
}

ns.cachedPartials.Lock()
defer ns.cachedPartials.Unlock()

var name string
name, p, err = ns.include(key.name, context)
if err != nil {
return nil, err
}

if ns.deps.Metrics != nil {
ns.deps.Metrics.TrackValue(name, p, true)
}

ns.cachedPartials.Lock()
defer ns.cachedPartials.Unlock()
// Double-check.
if p2, ok := ns.cachedPartials.p[key]; ok {
return p2, nil
ns.deps.Metrics.TrackValue(name, p, false)
}
ns.cachedPartials.p[key] = p

Expand Down

0 comments on commit 9e586dc

Please sign in to comment.