Compare commits

...

2 Commits

Author SHA1 Message Date
Kim "BKC" Carlbäcker 36db9e213f compile error 6 years ago
Kim "BKC" Carlbäcker 5b17f054c5 Make ParsePatch more performant 6 years ago
  1. 46
      models/git_diff.go

@ -247,30 +247,54 @@ func ParsePatch(maxLines, maxLineCharacters, maxFiles int, reader io.Reader) (*D
lineCount int lineCount int
curFileLinesCount int curFileLinesCount int
curFileLFSPrefix bool curFileLFSPrefix bool
input = bufio.NewReader(reader)
isEOF = false
) )
input := bufio.NewReader(reader)
isEOF := false
for !isEOF { for !isEOF {
var linebuf bytes.Buffer var linebuf bytes.Buffer
for { for {
b, err := input.ReadByte() peek, err := input.Peek(maxLineCharacters)
if err != nil && err != bufio.ErrBufferFull {
return nil, fmt.Errorf("PeekByte: %v", err)
}
newLine := bytes.IndexByte(peek, '\n');
if newLine == -1 {
// Instead of reading things, and copying memory around,
// we simply discard them (which doesn't allocate memory)
curFile.IsIncomplete = true
// We already know that we can read `len(peek)` amount of bytes,
// hence no error-checking
input.Discard(len(peek))
continue
}
if curFile.IsIncomplete {
// Since we get here without hiting the above case, we've found a newline
// and only discard that part.
input.Discard(newLine)
break
}
buff := make([]byte, newLine)
n, err := input.Read(buff)
if err != nil { if err != nil {
if err == io.EOF { if err == io.EOF {
isEOF = true isEOF = true
break break
} else {
return nil, fmt.Errorf("ReadByte: %v", err)
} }
return nil, fmt.Errorf("Read: %v", err)
} }
if b == '\n' { if n != newLine {
break return nil, fmt.Errorf("Read: could not read enough bytes %d != %d", n, newLine)
} }
if linebuf.Len() < maxLineCharacters { n, err = linebuf.Write(buff)
linebuf.WriteByte(b) if err != nil {
} else if linebuf.Len() == maxLineCharacters { return nil, fmt.Errorf("Write: %v", err)
curFile.IsIncomplete = true }
if n != newLine {
return nil, fmt.Errorf("Write: could not write enough bytes %d != %d", n, newLine)
} }
break
} }
line := linebuf.String() line := linebuf.String()

Loading…
Cancel
Save