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 count | 5 package count |
6 | 6 |
7 import ( | 7 import ( |
8 "golang.org/x/net/context" | 8 "golang.org/x/net/context" |
9 | 9 |
10 rds "github.com/luci/gae/service/rawdatastore" | 10 rds "github.com/luci/gae/service/rawdatastore" |
(...skipping 84 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
95 return ret, r.c.PutMulti.up(err) | 95 return ret, r.c.PutMulti.up(err) |
96 } | 96 } |
97 | 97 |
98 // FilterRDS installs a counter RawDatastore filter in the context. | 98 // FilterRDS installs a counter RawDatastore filter in the context. |
99 func FilterRDS(c context.Context) (context.Context, *RDSCounter) { | 99 func FilterRDS(c context.Context) (context.Context, *RDSCounter) { |
100 state := &RDSCounter{} | 100 state := &RDSCounter{} |
101 return rds.AddFilters(c, func(ic context.Context, rds rds.Interface) rds
.Interface { | 101 return rds.AddFilters(c, func(ic context.Context, rds rds.Interface) rds
.Interface { |
102 return &rdsCounter{state, rds} | 102 return &rdsCounter{state, rds} |
103 }), state | 103 }), state |
104 } | 104 } |
OLD | NEW |