Skip to content

Commit

Permalink
grpcproxy: fix memory leak
Browse files Browse the repository at this point in the history
use set instead of slice as interval value

fixes #10326
  • Loading branch information
Igor German authored and gyuho committed Dec 18, 2018
1 parent 3a7b8b3 commit 59cc0f9
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions proxy/grpcproxy/cache/store.go
Expand Up @@ -99,9 +99,12 @@ func (c *cache) Add(req *pb.RangeRequest, resp *pb.RangeResponse) {
iv = c.cachedRanges.Find(ivl)

if iv == nil {
c.cachedRanges.Insert(ivl, []string{key})
val := map[string]struct{}{key: {}}
c.cachedRanges.Insert(ivl, val)
} else {
iv.Val = append(iv.Val.([]string), key)
val := iv.Val.(map[string]struct{})
val[key] = struct{}{}
iv.Val = val
}
}

Expand Down Expand Up @@ -141,8 +144,8 @@ func (c *cache) Invalidate(key, endkey []byte) {

ivs = c.cachedRanges.Stab(ivl)
for _, iv := range ivs {
keys := iv.Val.([]string)
for _, key := range keys {
keys := iv.Val.(map[string]struct{})
for key := range keys {
c.lru.Remove(key)
}
}
Expand Down

0 comments on commit 59cc0f9

Please sign in to comment.