ByteSize

Author: gbbr

diff/diff.go

@@ -21,7 +21,7 @@ func (d *FileDiff) Stat() Stat {
 func (h *Hunk) Stat() Stat {
 	lines := bytes.Split(h.Body, []byte{'\n'})
 	var last byte
-	st := Stat{}
+	st := Stat{ByteSize: int32(len(h.Body))}
 	for _, line := range lines {
 		if len(line) == 0 {
 			last = 0
@@ -68,4 +68,5 @@ func (s *Stat) add(o Stat) {
 	s.Added += o.Added
 	s.Changed += o.Changed
 	s.Deleted += o.Deleted
+	s.ByteSize += o.ByteSize
 }

diff/diff.pb.go

@@ -72,5 +72,8 @@ message Stat {
 
 	// number of lines deleted
 	int32 deleted = 3;
+
+	// number of bytes contained in body
+	int32 byte_size = 4;
 }
 

diff/diff.proto

@@ -246,7 +246,7 @@ func TestFileDiff_Stat(t *testing.T) {
  b
 `)},
 			},
-			want: Stat{},
+			want: Stat{ByteSize: 19},
 		},
 		"added/deleted": {
 			hunks: []*Hunk{
@@ -257,7 +257,7 @@ func TestFileDiff_Stat(t *testing.T) {
  d
 `)},
 			},
-			want: Stat{Added: 1, Deleted: 1},
+			want: Stat{Added: 1, Deleted: 1, ByteSize: 25},
 		},
 		"changed": {
 			hunks: []*Hunk{
@@ -269,7 +269,7 @@ func TestFileDiff_Stat(t *testing.T) {
  e
 `)},
 			},
-			want: Stat{Added: 1, Changed: 1, Deleted: 1},
+			want: Stat{Added: 1, Changed: 1, Deleted: 1, ByteSize: 28},
 		},
 		"many changes": {
 			hunks: []*Hunk{
@@ -281,7 +281,7 @@ func TestFileDiff_Stat(t *testing.T) {
  e
 `)},
 			},
-			want: Stat{Added: 0, Changed: 2, Deleted: 0},
+			want: Stat{Added: 0, Changed: 2, Deleted: 0, ByteSize: 28},
 		},
 	}
 	for label, test := range tests {