| OLD | NEW |
| 1 // Copyright 2016 The LUCI Authors. All rights reserved. | 1 // Copyright 2016 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 "github.com/luci/gae/service/module" | 10 "github.com/luci/gae/service/module" |
| 11 ) | 11 ) |
| 12 | 12 |
| 13 type modState struct { | 13 type modState struct { |
| 14 *state | 14 *state |
| 15 | 15 |
| 16 » module.Interface | 16 » module.RawInterface |
| 17 } | 17 } |
| 18 | 18 |
| 19 func (m *modState) List() (ret []string, err error) { | 19 func (m *modState) List() (ret []string, err error) { |
| 20 err = m.run(func() (err error) { | 20 err = m.run(func() (err error) { |
| 21 » » ret, err = m.Interface.List() | 21 » » ret, err = m.RawInterface.List() |
| 22 return | 22 return |
| 23 }) | 23 }) |
| 24 return | 24 return |
| 25 } | 25 } |
| 26 | 26 |
| 27 func (m *modState) NumInstances(mod, ver string) (ret int, err error) { | 27 func (m *modState) NumInstances(mod, ver string) (ret int, err error) { |
| 28 err = m.run(func() (err error) { | 28 err = m.run(func() (err error) { |
| 29 » » ret, err = m.Interface.NumInstances(mod, ver) | 29 » » ret, err = m.RawInterface.NumInstances(mod, ver) |
| 30 return | 30 return |
| 31 }) | 31 }) |
| 32 return | 32 return |
| 33 } | 33 } |
| 34 | 34 |
| 35 func (m *modState) SetNumInstances(mod, ver string, instances int) error { | 35 func (m *modState) SetNumInstances(mod, ver string, instances int) error { |
| 36 return m.run(func() (err error) { | 36 return m.run(func() (err error) { |
| 37 » » return m.Interface.SetNumInstances(mod, ver, instances) | 37 » » return m.RawInterface.SetNumInstances(mod, ver, instances) |
| 38 }) | 38 }) |
| 39 } | 39 } |
| 40 | 40 |
| 41 func (m *modState) Versions(mod string) (ret []string, err error) { | 41 func (m *modState) Versions(mod string) (ret []string, err error) { |
| 42 err = m.run(func() (err error) { | 42 err = m.run(func() (err error) { |
| 43 » » ret, err = m.Interface.Versions(mod) | 43 » » ret, err = m.RawInterface.Versions(mod) |
| 44 return | 44 return |
| 45 }) | 45 }) |
| 46 return | 46 return |
| 47 } | 47 } |
| 48 | 48 |
| 49 func (m *modState) DefaultVersion(mod string) (ret string, err error) { | 49 func (m *modState) DefaultVersion(mod string) (ret string, err error) { |
| 50 err = m.run(func() (err error) { | 50 err = m.run(func() (err error) { |
| 51 » » ret, err = m.Interface.DefaultVersion(mod) | 51 » » ret, err = m.RawInterface.DefaultVersion(mod) |
| 52 return | 52 return |
| 53 }) | 53 }) |
| 54 return | 54 return |
| 55 } | 55 } |
| 56 | 56 |
| 57 func (m *modState) Start(mod, ver string) error { | 57 func (m *modState) Start(mod, ver string) error { |
| 58 return m.run(func() (err error) { | 58 return m.run(func() (err error) { |
| 59 » » return m.Interface.Start(mod, ver) | 59 » » return m.RawInterface.Start(mod, ver) |
| 60 }) | 60 }) |
| 61 } | 61 } |
| 62 | 62 |
| 63 func (m *modState) Stop(mod, ver string) error { | 63 func (m *modState) Stop(mod, ver string) error { |
| 64 return m.run(func() (err error) { | 64 return m.run(func() (err error) { |
| 65 » » return m.Interface.Stop(mod, ver) | 65 » » return m.RawInterface.Stop(mod, ver) |
| 66 }) | 66 }) |
| 67 } | 67 } |
| 68 | 68 |
| 69 // FilterModule installs a featureBreaker module filter in the context. | 69 // FilterModule installs a featureBreaker module filter in the context. |
| 70 func FilterModule(c context.Context, defaultError error) (context.Context, Featu
reBreaker) { | 70 func FilterModule(c context.Context, defaultError error) (context.Context, Featu
reBreaker) { |
| 71 state := newState(defaultError) | 71 state := newState(defaultError) |
| 72 » return module.AddFilters(c, func(ic context.Context, i module.Interface)
module.Interface { | 72 » return module.AddFilters(c, func(ic context.Context, i module.RawInterfa
ce) module.RawInterface { |
| 73 return &modState{state, i} | 73 return &modState{state, i} |
| 74 }), state | 74 }), state |
| 75 } | 75 } |
| OLD | NEW |