| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The LUCI 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 under the Apache License, Version 2.0 |
| 3 // found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
| 4 | 4 |
| 5 package dsQueryBatch | 5 package dsQueryBatch |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 ds "github.com/luci/gae/service/datastore" | 8 ds "github.com/luci/gae/service/datastore" |
| 9 "golang.org/x/net/context" | 9 "golang.org/x/net/context" |
| 10 ) | 10 ) |
| 11 | 11 |
| 12 // BatchQueries installs a datastore filter that causes all queries to be broken | 12 // BatchQueries installs a datastore filter that causes all queries to be broken |
| 13 // into a series of iterative fixed-size queries. The batching uses cursors to | 13 // into a series of iterative fixed-size queries. The batching uses cursors to |
| 14 // chain the iterations together. | 14 // chain the iterations together. |
| 15 // | 15 // |
| 16 // This helps accommodate query size or time limits enforced by the backing | 16 // This helps accommodate query size or time limits enforced by the backing |
| 17 // datastore implementation. | 17 // datastore implementation. |
| 18 // | 18 // |
| 19 // Note that this expands a single query into a series of queries, which may | 19 // Note that this expands a single query into a series of queries, which may |
| 20 // lose additional single-query consistency guarantees. | 20 // lose additional single-query consistency guarantees. |
| 21 func BatchQueries(c context.Context, batchSize int32) context.Context { | 21 func BatchQueries(c context.Context, batchSize int32) context.Context { |
| 22 return ds.AddRawFilters(c, func(ic context.Context, ri ds.RawInterface)
ds.RawInterface { | 22 return ds.AddRawFilters(c, func(ic context.Context, ri ds.RawInterface)
ds.RawInterface { |
| 23 return &iterQueryFilter{ | 23 return &iterQueryFilter{ |
| 24 RawInterface: ri, | 24 RawInterface: ri, |
| 25 batchSize: batchSize, | 25 batchSize: batchSize, |
| 26 } | 26 } |
| 27 }) | 27 }) |
| 28 } | 28 } |
| OLD | NEW |