OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 package memory | 5 package memory |
6 | 6 |
7 import ( | 7 import ( |
8 "errors" | 8 "errors" |
9 "sync" | 9 "sync" |
10 | 10 |
(...skipping 11 matching lines...) Expand all Loading... |
22 data []byte | 22 data []byte |
23 } | 23 } |
24 | 24 |
25 // Storage is an implementation of the storage.Storage interface that stores | 25 // Storage is an implementation of the storage.Storage interface that stores |
26 // data in memory. | 26 // data in memory. |
27 // | 27 // |
28 // This is intended for testing, and not intended to be performant. | 28 // This is intended for testing, and not intended to be performant. |
29 type Storage struct { | 29 type Storage struct { |
30 // MaxGetCount, if not zero, is the maximum number of records to retriev
e from | 30 // MaxGetCount, if not zero, is the maximum number of records to retriev
e from |
31 // a single Get request. | 31 // a single Get request. |
32 » MaxGetCount int | 32 » MaxGetCount int64 |
33 | 33 |
34 stateMu sync.Mutex | 34 stateMu sync.Mutex |
35 streams map[types.StreamPath]*logStream | 35 streams map[types.StreamPath]*logStream |
36 closed bool | 36 closed bool |
37 } | 37 } |
38 | 38 |
39 var _ storage.Storage = (*Storage)(nil) | 39 var _ storage.Storage = (*Storage)(nil) |
40 | 40 |
41 // Close implements storage.Storage. | 41 // Close implements storage.Storage. |
42 func (s *Storage) Close() { | 42 func (s *Storage) Close() { |
(...skipping 22 matching lines...) Expand all Loading... |
65 | 65 |
66 // Get implements storage.Storage. | 66 // Get implements storage.Storage. |
67 func (s *Storage) Get(req *storage.GetRequest, cb storage.GetCallback) error { | 67 func (s *Storage) Get(req *storage.GetRequest, cb storage.GetCallback) error { |
68 recs := []*rec(nil) | 68 recs := []*rec(nil) |
69 err := s.run(func() error { | 69 err := s.run(func() error { |
70 ls := s.getLogStreamLocked(req.Path, false) | 70 ls := s.getLogStreamLocked(req.Path, false) |
71 if ls == nil { | 71 if ls == nil { |
72 return storage.ErrDoesNotExist | 72 return storage.ErrDoesNotExist |
73 } | 73 } |
74 | 74 |
75 » » limit := len(ls.logs) | 75 » » limit := int64(len(ls.logs)) |
76 if req.Limit > 0 && req.Limit < limit { | 76 if req.Limit > 0 && req.Limit < limit { |
77 limit = req.Limit | 77 limit = req.Limit |
78 } | 78 } |
79 if s.MaxGetCount > 0 && s.MaxGetCount < limit { | 79 if s.MaxGetCount > 0 && s.MaxGetCount < limit { |
80 limit = s.MaxGetCount | 80 limit = s.MaxGetCount |
81 } | 81 } |
82 | 82 |
83 // Grab all records starting from our start index. | 83 // Grab all records starting from our start index. |
84 for idx := req.Index; idx <= ls.latestIndex; idx++ { | 84 for idx := req.Index; idx <= ls.latestIndex; idx++ { |
85 if le, ok := ls.logs[idx]; ok { | 85 if le, ok := ls.logs[idx]; ok { |
86 recs = append(recs, &rec{ | 86 recs = append(recs, &rec{ |
87 index: idx, | 87 index: idx, |
88 data: le, | 88 data: le, |
89 }) | 89 }) |
90 } | 90 } |
91 | 91 |
92 » » » if len(recs) >= limit { | 92 » » » if int64(len(recs)) >= limit { |
93 break | 93 break |
94 } | 94 } |
95 } | 95 } |
96 return nil | 96 return nil |
97 }) | 97 }) |
98 if err != nil { | 98 if err != nil { |
99 return err | 99 return err |
100 } | 100 } |
101 | 101 |
102 // Punt all of the records upstream. We copy the data to prevent the | 102 // Punt all of the records upstream. We copy the data to prevent the |
(...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
178 } | 178 } |
179 | 179 |
180 if s.streams == nil { | 180 if s.streams == nil { |
181 s.streams = map[types.StreamPath]*logStream{} | 181 s.streams = map[types.StreamPath]*logStream{} |
182 } | 182 } |
183 s.streams[p] = ls | 183 s.streams[p] = ls |
184 } | 184 } |
185 | 185 |
186 return ls | 186 return ls |
187 } | 187 } |
OLD | NEW |