goparsify/debugon.go

// +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)
	}
}