fixed bug

cheekybits · Oct 31, 2014 · 8159883 · 8159883
1 parent b4494c1
commit 8159883
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/parse/parse.go b/parse/parse.go
@@ -37,7 +37,7 @@ var (
 	linefeed       = "\r\n"
 )
 var unwantedLinePrefixes = [][]byte{
-	[]byte("//go:generate genny"),
+	[]byte("//go:generate genny "),
 }
 
 func generateSpecific(filename string, in io.ReadSeeker, typeSet map[string]string) ([]byte, error) {
@@ -177,12 +177,18 @@ func Generics(filename string, in io.ReadSeeker, typeSets []map[string]string) (
 		}
 
 		// check all unwantedLinePrefixes - and skip them
+		skipline := false
 		for _, prefix := range unwantedLinePrefixes {
 			if bytes.HasPrefix(scanner.Bytes(), prefix) {
+				skipline = true
 				continue
 			}
 		}
 
+		if skipline {
+			continue
+		}
+
 		cleanOutputLines = append(cleanOutputLines, line(scanner.Text()))
 	}