OLD | NEW |
1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
4 | 4 |
5 package featureBreaker | 5 package featureBreaker |
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 16 matching lines...) Expand all Loading... |
27 } | 27 } |
28 | 28 |
29 func (t *tqState) Purge(queueName string) error { | 29 func (t *tqState) Purge(queueName string) error { |
30 return t.run(func() error { return t.tq.Purge(queueName) }) | 30 return t.run(func() error { return t.tq.Purge(queueName) }) |
31 } | 31 } |
32 | 32 |
33 func (t *tqState) Stats(queueNames []string, cb tq.RawStatsCB) error { | 33 func (t *tqState) Stats(queueNames []string, cb tq.RawStatsCB) error { |
34 return t.run(func() error { return t.tq.Stats(queueNames, cb) }) | 34 return t.run(func() error { return t.tq.Stats(queueNames, cb) }) |
35 } | 35 } |
36 | 36 |
37 func (t *tqState) Testable() tq.Testable { | 37 func (t *tqState) GetTestable() tq.Testable { |
38 » return t.tq.Testable() | 38 » return t.tq.GetTestable() |
39 } | 39 } |
40 | 40 |
41 // FilterTQ installs a featureBreaker TaskQueue filter in the context. | 41 // FilterTQ installs a featureBreaker TaskQueue filter in the context. |
42 func FilterTQ(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { | 42 func FilterTQ(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { |
43 state := newState(defaultError) | 43 state := newState(defaultError) |
44 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { | 44 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { |
45 return &tqState{state, tq} | 45 return &tqState{state, tq} |
46 }), state | 46 }), state |
47 } | 47 } |
OLD | NEW |