| 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 | 
|---|