From 6fa7f2ac3efda8359d208c8f962d3d761fca3fc7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bj=C3=B8rn=20Erik=20Pedersen?=
 <bjorn.erik.pedersen@gmail.com>
Date: Sat, 21 Nov 2015 23:29:58 +0100
Subject: [PATCH] Avoid multiple locks for shortcode template

Fixes #1600
---
 hugolib/shortcode.go | 23 ++++++++++-------------
 1 file changed, 10 insertions(+), 13 deletions(-)

diff --git a/hugolib/shortcode.go b/hugolib/shortcode.go
index 4d3037dc8..a306023a7 100644
--- a/hugolib/shortcode.go
+++ b/hugolib/shortcode.go
@@ -160,21 +160,24 @@ var isInnerShortcodeCache = struct {
 // to avoid potential costly look-aheads for closing tags we look inside the template itself
 // we could change the syntax to self-closing tags, but that would make users cry
 // the value found is cached
-func isInnerShortcode(t *template.Template) bool {
+func isInnerShortcode(t *template.Template) (bool, error) {
 	isInnerShortcodeCache.RLock()
 	m, ok := isInnerShortcodeCache.m[t.Name()]
 	isInnerShortcodeCache.RUnlock()
 
 	if ok {
-		return m
+		return m, nil
 	}
 
 	isInnerShortcodeCache.Lock()
 	defer isInnerShortcodeCache.Unlock()
+	if t.Tree == nil {
+		return false, errors.New("Template failed to compile")
+	}
 	match, _ := regexp.MatchString("{{.*?\\.Inner.*?}}", t.Tree.Root.String())
 	isInnerShortcodeCache.m[t.Name()] = match
 
-	return match
+	return match, nil
 }
 
 func createShortcodePlaceholder(id int) string {
@@ -344,14 +347,11 @@ Loop:
 				return sc, fmt.Errorf("Unable to locate template for shortcode '%s' in page %s", sc.name, p.BaseFileName())
 			}
 
-			// TODO(bep) Refactor/rename this lock strategy
-			isInnerShortcodeCache.RLock()
-			if tmpl.Tree == nil {
-				isInnerShortcodeCache.RUnlock()
-				return sc, fmt.Errorf("Template for shortcode '%s' failed to compile for page '%s'", sc.name, p.BaseFileName())
+			var err error
+			isInner, err = isInnerShortcode(tmpl)
+			if err != nil {
+				return sc, fmt.Errorf("Failed to handle template for shortcode '%s' for page '%s': %s", sc.name, p.BaseFileName(), err)
 			}
-			isInnerShortcodeCache.RUnlock()
-			isInner = isInnerShortcode(tmpl)
 
 		case tScParam:
 			if !pt.isValueNext() {
@@ -523,9 +523,6 @@ func renderShortcodeWithPage(tmpl *template.Template, data *ShortcodeWithPage) s
 	buffer := bp.GetBuffer()
 	defer bp.PutBuffer(buffer)
 
-	// TODO(bep) Refactor/rename this lock strategy
-	isInnerShortcodeCache.Lock()
-	defer isInnerShortcodeCache.Unlock()
 	err := tmpl.Execute(buffer, data)
 	if err != nil {
 		jww.ERROR.Println("error processing shortcode", tmpl.Name(), "\n ERR:", err)