From cc9d18219af9375ad89eaa8a23f1e0bcffa5734e Mon Sep 17 00:00:00 2001
From: Adam Scarr <adam@vektah.net>
Date: Mon, 7 Aug 2017 20:07:29 +1000
Subject: add debugging

---
 debugon.go | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 debugon.go

(limited to 'debugon.go')

diff --git a/debugon.go b/debugon.go
new file mode 100644
index 0000000..79acf0f
--- /dev/null
+++ b/debugon.go
@@ -0,0 +1,83 @@
+// +build debug
+
+package goparsify
+
+import (
+	"fmt"
+	"runtime"
+	"sort"
+	"strings"
+	"time"
+)
+
+var parsers []*DebugParser
+
+type DebugParser struct {
+	Description string
+	Caller      string
+	Next        Parser
+	Time        time.Duration
+	Calls       int
+}
+
+func (dp *DebugParser) Parse(ps *State) *Node {
+	start := time.Now()
+
+	ret := dp.Next(ps)
+
+	dp.Time = dp.Time + time.Since(start)
+	dp.Calls++
+
+	return ret
+}
+
+func getPackageName(f *runtime.Func) string {
+	parts := strings.Split(f.Name(), ".")
+	pl := len(parts)
+
+	if parts[pl-2][0] == '(' {
+		return strings.Join(parts[0:pl-2], ".")
+	} else {
+		return strings.Join(parts[0:pl-1], ".")
+	}
+}
+
+func NewParser(description string, p Parser) Parser {
+	fpcs := make([]uintptr, 1)
+	caller := ""
+
+	for i := 1; i < 10; i++ {
+		n := runtime.Callers(i, fpcs)
+
+		if n != 0 {
+			frame := runtime.FuncForPC(fpcs[0] - 1)
+			pkg := getPackageName(frame)
+
+			if pkg != "github.com/vektah/goparsify" {
+				file, line := frame.FileLine(fpcs[0] - 1)
+				caller = fmt.Sprintf("%s:%d", file, line)
+				break
+			}
+		}
+	}
+
+	dp := &DebugParser{
+		Description: description,
+		Next:        p,
+		Caller:      caller,
+	}
+
+	parsers = append(parsers, dp)
+	return dp.Parse
+}
+
+func DumpDebugStats() {
+	sort.Slice(parsers, func(i, j int) bool {
+		return parsers[i].Time >= parsers[j].Time
+	})
+
+	fmt.Println("Parser stats:")
+	for _, parser := range parsers {
+		fmt.Printf("%20s\t%10s\t%10d\tcalls\t%s\n", parser.Description, parser.Time.String(), parser.Calls, parser.Caller)
+	}
+}
-- 
cgit v1.2.3