1 files changed, 60 insertions, 6 deletions
diff --git a/system/db/content.go b/system/db/content.go
index 49cba87..b503a60 100644
--- a/system/db/content.go
+++ b/system/db/content.go
@@ -9,8 +9,11 @@ import (
 	"sort"
 	"strconv"
 	"strings"
+	"sync"
+	"time"
 
 	"github.com/ponzu-cms/ponzu/system/item"
+	"github.com/ponzu-cms/ponzu/system/search"
 
 	"github.com/boltdb/bolt"
 	"github.com/gorilla/schema"
@@ -124,9 +127,9 @@ func update(ns, id string, data url.Values, existingContent *[]byte) (int, error
 	go func() {
 		// update data in search index
 		target := fmt.Sprintf("%s:%s", ns, id)
-		err = UpdateSearchIndex(target, string(j))
+		err = search.UpdateIndex(target, j)
 		if err != nil {
-			log.Println("[search] UpdateSearchIndex Error:", err)
+			log.Println("[search] UpdateIndex Error:", err)
 		}
 	}()
 
@@ -252,9 +255,9 @@ func insert(ns string, data url.Values) (int, error) {
 	go func() {
 		// add data to seach index
 		target := fmt.Sprintf("%s:%s", ns, cid)
-		err = UpdateSearchIndex(target, string(j))
+		err = search.UpdateIndex(target, j)
 		if err != nil {
-			log.Println("[search] UpdateSearchIndex Error:", err)
+			log.Println("[search] UpdateIndex Error:", err)
 		}
 	}()
 
@@ -321,9 +324,9 @@ func DeleteContent(target string) error {
 		// delete indexed data from search index
 		if !strings.Contains(ns, "__") {
 			target = fmt.Sprintf("%s:%s", ns, id)
-			err = DeleteSearchIndex(target)
+			err = search.DeleteIndex(target)
 			if err != nil {
-				log.Println("[search] DeleteSearchIndex Error:", err)
+				log.Println("[search] DeleteIndex Error:", err)
 			}
 		}
 	}()
@@ -563,10 +566,61 @@ func Query(namespace string, opts QueryOptions) (int, [][]byte) {
 	return total, posts
 }
 
+var sortContentCalls = make(map[string]time.Time)
+var waitDuration = time.Millisecond * 2000
+var sortMutex = &sync.Mutex{}
+
+func setLastInvocation(key string) {
+	sortMutex.Lock()
+	sortContentCalls[key] = time.Now()
+	sortMutex.Unlock()
+}
+
+func lastInvocation(key string) (time.Time, bool) {
+	sortMutex.Lock()
+	last, ok := sortContentCalls[key]
+	sortMutex.Unlock()
+	return last, ok
+}
+
+func enoughTime(key string) bool {
+	last, ok := lastInvocation(key)
+	if !ok {
+		// no invocation yet
+		// track next invocation
+		setLastInvocation(key)
+		return true
+	}
+
+	// if our required wait time has been met, return true
+	if time.Now().After(last.Add(waitDuration)) {
+		setLastInvocation(key)
+		return true
+	}
+
+	// dispatch a delayed invocation in case no additional one follows
+	go func() {
+		lastInvocationBeforeTimer, _ := lastInvocation(key) // zero value can be handled, no need for ok
+		enoughTimer := time.NewTimer(waitDuration)
+		<-enoughTimer.C
+		lastInvocationAfterTimer, _ := lastInvocation(key)
+		if !lastInvocationAfterTimer.After(lastInvocationBeforeTimer) {
+			SortContent(key)
+		}
+	}()
+
+	return false
+}
+
 // SortContent sorts all content of the type supplied as the namespace by time,
 // in descending order, from most recent to least recent
 // Should be called from a goroutine after SetContent is successful
 func SortContent(namespace string) {
+	// wait if running too frequently per namespace
+	if !enoughTime(namespace) {
+		return
+	}
+
 	// only sort main content types i.e. Post
 	if strings.Contains(namespace, "__") {
 		return