33package measure
44
55import (
6+ "context"
67 "io"
78 "time"
89
@@ -171,31 +172,31 @@ func recordLatency(h metrics.Histogram, start time.Time) {
171172 h .Observe (elapsed .Seconds ())
172173}
173174
174- func (m * measure ) Put (key datastore.Key , value []byte ) error {
175+ func (m * measure ) Put (ctx context. Context , key datastore.Key , value []byte ) error {
175176 defer recordLatency (m .putLatency , time .Now ())
176177 m .putNum .Inc ()
177178 m .putSize .Observe (float64 (len (value )))
178- err := m .backend .Put (key , value )
179+ err := m .backend .Put (ctx , key , value )
179180 if err != nil {
180181 m .putErr .Inc ()
181182 }
182183 return err
183184}
184185
185- func (m * measure ) Sync (prefix datastore.Key ) error {
186+ func (m * measure ) Sync (ctx context. Context , prefix datastore.Key ) error {
186187 defer recordLatency (m .syncLatency , time .Now ())
187188 m .syncNum .Inc ()
188- err := m .backend .Sync (prefix )
189+ err := m .backend .Sync (ctx , prefix )
189190 if err != nil {
190191 m .syncErr .Inc ()
191192 }
192193 return err
193194}
194195
195- func (m * measure ) Get (key datastore.Key ) (value []byte , err error ) {
196+ func (m * measure ) Get (ctx context. Context , key datastore.Key ) (value []byte , err error ) {
196197 defer recordLatency (m .getLatency , time .Now ())
197198 m .getNum .Inc ()
198- value , err = m .backend .Get (key )
199+ value , err = m .backend .Get (ctx , key )
199200 switch err {
200201 case nil :
201202 m .getSize .Observe (float64 (len (value )))
@@ -207,20 +208,20 @@ func (m *measure) Get(key datastore.Key) (value []byte, err error) {
207208 return value , err
208209}
209210
210- func (m * measure ) Has (key datastore.Key ) (exists bool , err error ) {
211+ func (m * measure ) Has (ctx context. Context , key datastore.Key ) (exists bool , err error ) {
211212 defer recordLatency (m .hasLatency , time .Now ())
212213 m .hasNum .Inc ()
213- exists , err = m .backend .Has (key )
214+ exists , err = m .backend .Has (ctx , key )
214215 if err != nil {
215216 m .hasErr .Inc ()
216217 }
217218 return exists , err
218219}
219220
220- func (m * measure ) GetSize (key datastore.Key ) (size int , err error ) {
221+ func (m * measure ) GetSize (ctx context. Context , key datastore.Key ) (size int , err error ) {
221222 defer recordLatency (m .getsizeLatency , time .Now ())
222223 m .getsizeNum .Inc ()
223- size , err = m .backend .GetSize (key )
224+ size , err = m .backend .GetSize (ctx , key )
224225 switch err {
225226 case nil , datastore .ErrNotFound :
226227 // Not really an error.
@@ -230,31 +231,31 @@ func (m *measure) GetSize(key datastore.Key) (size int, err error) {
230231 return size , err
231232}
232233
233- func (m * measure ) Delete (key datastore.Key ) error {
234+ func (m * measure ) Delete (ctx context. Context , key datastore.Key ) error {
234235 defer recordLatency (m .deleteLatency , time .Now ())
235236 m .deleteNum .Inc ()
236- err := m .backend .Delete (key )
237+ err := m .backend .Delete (ctx , key )
237238 if err != nil {
238239 m .deleteErr .Inc ()
239240 }
240241 return err
241242}
242243
243- func (m * measure ) Query (q query.Query ) (query.Results , error ) {
244+ func (m * measure ) Query (ctx context. Context , q query.Query ) (query.Results , error ) {
244245 defer recordLatency (m .queryLatency , time .Now ())
245246 m .queryNum .Inc ()
246- res , err := m .backend .Query (q )
247+ res , err := m .backend .Query (ctx , q )
247248 if err != nil {
248249 m .queryErr .Inc ()
249250 }
250251 return res , err
251252}
252253
253- func (m * measure ) Check () error {
254+ func (m * measure ) Check (ctx context. Context ) error {
254255 defer recordLatency (m .checkLatency , time .Now ())
255256 m .checkNum .Inc ()
256257 if c , ok := m .backend .(datastore.CheckedDatastore ); ok {
257- err := c .Check ()
258+ err := c .Check (ctx )
258259 if err != nil {
259260 m .checkErr .Inc ()
260261 }
@@ -263,11 +264,11 @@ func (m *measure) Check() error {
263264 return nil
264265}
265266
266- func (m * measure ) Scrub () error {
267+ func (m * measure ) Scrub (ctx context. Context ) error {
267268 defer recordLatency (m .scrubLatency , time .Now ())
268269 m .scrubNum .Inc ()
269270 if c , ok := m .backend .(datastore.ScrubbedDatastore ); ok {
270- err := c .Scrub ()
271+ err := c .Scrub (ctx )
271272 if err != nil {
272273 m .scrubErr .Inc ()
273274 }
@@ -276,11 +277,11 @@ func (m *measure) Scrub() error {
276277 return nil
277278}
278279
279- func (m * measure ) CollectGarbage () error {
280+ func (m * measure ) CollectGarbage (ctx context. Context ) error {
280281 defer recordLatency (m .gcLatency , time .Now ())
281282 m .gcNum .Inc ()
282283 if c , ok := m .backend .(datastore.GCDatastore ); ok {
283- err := c .CollectGarbage ()
284+ err := c .CollectGarbage (ctx )
284285 if err != nil {
285286 m .gcErr .Inc ()
286287 }
@@ -289,10 +290,10 @@ func (m *measure) CollectGarbage() error {
289290 return nil
290291}
291292
292- func (m * measure ) DiskUsage () (uint64 , error ) {
293+ func (m * measure ) DiskUsage (ctx context. Context ) (uint64 , error ) {
293294 defer recordLatency (m .duLatency , time .Now ())
294295 m .duNum .Inc ()
295- size , err := datastore .DiskUsage (m .backend )
296+ size , err := datastore .DiskUsage (ctx , m .backend )
296297 if err != nil {
297298 m .duErr .Inc ()
298299 }
@@ -304,12 +305,12 @@ type measuredBatch struct {
304305 m * measure
305306}
306307
307- func (m * measure ) Batch () (datastore.Batch , error ) {
308+ func (m * measure ) Batch (ctx context. Context ) (datastore.Batch , error ) {
308309 bds , ok := m .backend .(datastore.Batching )
309310 if ! ok {
310311 return nil , datastore .ErrBatchUnsupported
311312 }
312- batch , err := bds .Batch ()
313+ batch , err := bds .Batch (ctx )
313314 if err != nil {
314315 return nil , err
315316 }
@@ -320,31 +321,31 @@ func (m *measure) Batch() (datastore.Batch, error) {
320321 }, nil
321322}
322323
323- func (mt * measuredBatch ) Put (key datastore.Key , val []byte ) error {
324+ func (mt * measuredBatch ) Put (ctx context. Context , key datastore.Key , val []byte ) error {
324325 defer recordLatency (mt .m .batchPutLatency , time .Now ())
325326 mt .m .batchPutNum .Inc ()
326327 mt .m .batchPutSize .Observe (float64 (len (val )))
327- err := mt .b .Put (key , val )
328+ err := mt .b .Put (ctx , key , val )
328329 if err != nil {
329330 mt .m .batchPutErr .Inc ()
330331 }
331332 return err
332333}
333334
334- func (mt * measuredBatch ) Delete (key datastore.Key ) error {
335+ func (mt * measuredBatch ) Delete (ctx context. Context , key datastore.Key ) error {
335336 defer recordLatency (mt .m .batchDeleteLatency , time .Now ())
336337 mt .m .batchDeleteNum .Inc ()
337- err := mt .b .Delete (key )
338+ err := mt .b .Delete (ctx , key )
338339 if err != nil {
339340 mt .m .batchDeleteErr .Inc ()
340341 }
341342 return err
342343}
343344
344- func (mt * measuredBatch ) Commit () error {
345+ func (mt * measuredBatch ) Commit (ctx context. Context ) error {
345346 defer recordLatency (mt .m .batchCommitLatency , time .Now ())
346347 mt .m .batchCommitNum .Inc ()
347- err := mt .b .Commit ()
348+ err := mt .b .Commit (ctx )
348349 if err != nil {
349350 mt .m .batchCommitErr .Inc ()
350351 }
0 commit comments