Remove cpprof

cmd/cmd.go

@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"os"
 	"regexp"
-	"runtime/pprof"
 	"strings"
 
 	"github.com/howardjohn/kubectl-grep/pkg"
@@ -25,12 +24,6 @@ var rootCmd = &cobra.Command{
 	Use:   "kubectl-grep",
 	Short: "A plugin to grep Kubernetes resources.",
 	RunE: func(cmd *cobra.Command, args []string) error {
-		f, perr := os.Create("cpu.pprof")
-		if perr != nil {
-			return perr
-		}
-		pprof.StartCPUProfile(f)
-		defer pprof.StopCPUProfile()
 		dm := pkg.Full
 		if summary {
 			dm = pkg.Summary

pkg/decode.go

@@ -10,19 +10,24 @@ import (
 
 // Implementation borrowed from k8s.io/apimachinery to avoid large imports
 
-const separator = "---"
+const (
+	separator     = "---"
+	listSeparator = "- "
+)
 
 type Reader interface {
 	Read() ([]byte, error)
 }
 
 type YAMLReader struct {
 	reader Reader
+	isList bool
 }
 
-func NewYAMLReader(r *bufio.Reader) *YAMLReader {
+func NewYAMLReader(r *bufio.Reader, isList bool) *YAMLReader {
 	return &YAMLReader{
 		reader: &LineReader{reader: r},
+		isList: isList,
 	}
 }
 
@@ -35,6 +40,15 @@ func (r *YAMLReader) Read() ([]byte, error) {
 			return nil, err
 		}
 
+		if r.isList {
+			// TODO: optimize if we know its a list
+			//sep := len([]byte(listSeparator))
+			//if i := bytes.Index(line, []byte(listSeparator)); i == 0 {
+			//	fmt.Println(string(line))
+			//	i+=sep
+			//}
+		}
+
 		sep := len([]byte(separator))
 		if i := bytes.Index(line, []byte(separator)); i == 0 {
 			// We have a potential document terminator
@@ -82,4 +96,4 @@ func (r *LineReader) Read() ([]byte, error) {
 	}
 	buffer.WriteByte('\n')
 	return buffer.Bytes(), err
-}
+}

pkg/grep.go

@@ -109,8 +109,14 @@ const (
 	CleanStatus
 )
 
+var listFastHeader = `apiVersion: v1
+items:
+-`
+
 func GrepResources(sel Selector, in io.Reader, mode DisplayMode) (string, error) {
-	reader := NewYAMLReader(bufio.NewReader(in))
+	r := bufio.NewReader(in)
+	b, _ := r.Peek(len(listFastHeader))
+	reader := NewYAMLReader(r, string(b) == listFastHeader)
 	matches := []string{}
 	for {
 		text, err := reader.Read()