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 tq "github.com/luci/gae/service/taskqueue" | 10 tq "github.com/luci/gae/service/taskqueue" |
(...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
72 return ret, t.c.QueueStats.up(err) | 72 return ret, t.c.QueueStats.up(err) |
73 } | 73 } |
74 | 74 |
75 // FilterTQ installs a counter TaskQueue filter in the context. | 75 // FilterTQ installs a counter TaskQueue filter in the context. |
76 func FilterTQ(c context.Context) (context.Context, *TQCounter) { | 76 func FilterTQ(c context.Context) (context.Context, *TQCounter) { |
77 state := &TQCounter{} | 77 state := &TQCounter{} |
78 return tq.AddFilters(c, func(ic context.Context, tq tq.Interface) tq.Int
erface { | 78 return tq.AddFilters(c, func(ic context.Context, tq tq.Interface) tq.Int
erface { |
79 return &tqCounter{state, tq} | 79 return &tqCounter{state, tq} |
80 }), state | 80 }), state |
81 } | 81 } |
OLD | NEW |