attempt to process esi in parallel

Author: digitalkaoz

.github/workflows/middlewares.yml

@@ -16,7 +16,7 @@ jobs:
         name: Install Go
         uses: actions/setup-go@v3
         with:
-          go-version: 1.19
+          go-version: 1.22
       -
         name: Checkout code
         uses: actions/checkout@v3

.github/workflows/non-regression.yml

@@ -20,10 +20,10 @@ jobs:
         name: install Go
         uses: actions/setup-go@v3
         with:
-          go-version: 1.19
+          go-version: 1.22
       -
         name: golangci-lint
-        uses: golangci/golangci-lint-action@v3
+        uses: golangci/golangci-lint-action@v6
       -
         name: install xcaddy
         run: go install github.com/caddyserver/xcaddy/cmd/xcaddy@latest

.github/workflows/release.yml

@@ -13,7 +13,7 @@ jobs:
         name: Set up Go
         uses: actions/setup-go@v2
         with:
-          go-version: 1.19
+          go-version: 1.22
       -
         name: Checkout
         uses: actions/checkout@v2

.golangci.yml

@@ -7,18 +7,16 @@ linters:
   disable:
     - bodyclose
     - cyclop
-    - exhaustivestruct
+    - depguard
     - exhaustruct
     - forbidigo
     - gochecknoglobals
     - gci
-    - golint
     - gomnd
-    - ifshort
     - ireturn
     - nestif
     - nonamedreturns
-    - nosnakecase
+    - mnd
     - revive
     - testpackage
     - stylecheck

esi/esi.go

@@ -2,6 +2,8 @@ package esi
 
 import (
 	"net/http"
+	"slices"
+	"sync"
 )
 
 func findTagName(b []byte) Tag {
@@ -47,14 +49,6 @@ func HasOpenedTags(b []byte) bool {
 	return esi.FindIndex(b) != nil || escapeRg.FindIndex(b) != nil
 }
 
-func CanProcess(b []byte) bool {
-	if tag := findTagName(b); tag != nil {
-		return tag.HasClose(b)
-	}
-
-	return false
-}
-
 func ReadToTag(next []byte, pointer int) (startTagPosition, esiPointer int, t Tag) {
 	var isEscapeTag bool
 
@@ -83,35 +77,81 @@ func ReadToTag(next []byte, pointer int) (startTagPosition, esiPointer int, t Ta
 
 func Parse(b []byte, req *http.Request) []byte {
 	pointer := 0
+	includes := make(map[int]Tag)
 
 	for pointer < len(b) {
-		var escapeTag bool
-
 		next := b[pointer:]
 		tagIdx := esi.FindIndex(next)
 
 		if escIdx := escapeRg.FindIndex(next); escIdx != nil && (tagIdx == nil || escIdx[0] < tagIdx[0]) {
 			tagIdx = escIdx
 			tagIdx[1] = escIdx[0]
-			escapeTag = true
 		}
 
 		if tagIdx == nil {
 			break
 		}
 
-		esiPointer := tagIdx[1]
-		t := findTagName(next[esiPointer:])
+		t := findTagName(next[tagIdx[1]:])
 
-		if escapeTag {
-			esiPointer += 7
+		if t != nil {
+			includes[pointer+tagIdx[0]] = t
 		}
 
-		res, p := t.Process(next[esiPointer:], req)
-		esiPointer += p
+		pointer += tagIdx[1]
+	}
+
+	if len(includes) == 0 {
+		return b
+	}
+
+	return processEsiTags(b, req, includes)
+}
+
+func processEsiTags(b []byte, req *http.Request, includes map[int]Tag) []byte {
+	type Response struct {
+		Start int
+		End   int
+		Res   []byte
+		Tag   Tag
+	}
+
+	var wg sync.WaitGroup
+
+	responses := make(chan Response, len(includes))
+
+	for start, t := range includes {
+		wg.Add(1)
+
+		go func(s int, tag Tag) {
+			defer wg.Done()
+
+			data, length := tag.Process(b[s:], req)
+
+			responses <- Response{
+				Start: s,
+				End:   s + length,
+				Res:   data,
+				Tag:   tag,
+			}
+		}(start, t)
+	}
+
+	wg.Wait()
+	close(responses)
+
+	// we need to replace from the bottom to top
+	sorted := make([]Response, 0, len(responses))
+	for r := range responses {
+		sorted = append(sorted, r)
+	}
+
+	slices.SortFunc(sorted, func(a, b Response) int {
+		return b.Start - a.Start
+	})
 
-		b = append(b[:pointer], append(next[:tagIdx[0]], append(res, next[esiPointer:]...)...)...)
-		pointer += len(res) + tagIdx[0]
+	for _, r := range sorted {
+		b = append(b[:r.Start], append(r.Res, b[r.End:]...)...)
 	}
 
 	return b

esi/include.go

@@ -12,7 +12,7 @@ import (
 const include = "include"
 
 var (
-	closeInclude     = regexp.MustCompile("/>")
+	closeInclude     = regexp.MustCompile(">")
 	srcAttribute     = regexp.MustCompile(`src="?(.+?)"?( |/>)`)
 	altAttribute     = regexp.MustCompile(`alt="?(.+?)"?( |/>)`)
 	onErrorAttribute = regexp.MustCompile(`onerror="?(.+?)"?( |/>)`)

esi/type.go

@@ -6,9 +6,9 @@ import (
 
 type (
 	Tag interface {
-		Process([]byte, *http.Request) ([]byte, int)
-		HasClose([]byte) bool
-		GetClosePosition([]byte) int
+		Process(body []byte, request *http.Request) ([]byte, int)
+		HasClose(body []byte) bool
+		GetClosePosition(body []byte) int
 	}
 
 	baseTag struct {

esi/vars.go

@@ -76,6 +76,8 @@ type varsTag struct {
 	*baseTag
 }
 
+const varTagLength = len("<esi:vars>")
+
 // Input (e.g. comment text="This is a comment." />).
 func (c *varsTag) Process(b []byte, req *http.Request) ([]byte, int) {
 	found := closeVars.FindIndex(b)
@@ -85,7 +87,7 @@ func (c *varsTag) Process(b []byte, req *http.Request) ([]byte, int) {
 
 	c.length = found[1]
 
-	return interpretedVar.ReplaceAllFunc(b[5:found[0]], func(b []byte) []byte {
+	return interpretedVar.ReplaceAllFunc(b[varTagLength:found[0]], func(b []byte) []byte {
 		return []byte(parseVariables(b, req))
 	}), c.length
 }

esi/when.go

@@ -23,6 +23,7 @@ func validateTest(b []byte, req *http.Request) bool {
 	} else if r := comparison.FindSubmatch(b); r != nil {
 		r1 := strings.TrimSpace(parseVariables(r[1], req))
 		r2 := strings.TrimSpace(parseVariables(r[3], req))
+
 		switch string(r[2]) {
 		case "==":
 			return r1 == r2

go.mod

@@ -1,3 +1,3 @@
 module github.com/darkweak/go-esi
 
-go 1.19
+go 1.22

middleware/caddy/esi.go

@@ -12,7 +12,7 @@ import (
 	"github.com/darkweak/go-esi/writer"
 )
 
-var bufPool *sync.Pool = &sync.Pool{
+var bufPool = &sync.Pool{
 	New: func() any {
 		return &bytes.Buffer{}
 	},

middleware/caddy/go.mod

@@ -1,6 +1,8 @@
 module github.com/darkweak/go-esi/middleware/caddy
 
-go 1.19
+go 1.22
+
+toolchain go1.23.1
 
 require (
 	github.com/caddyserver/caddy/v2 v2.6.4