| 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 "time" | 8 "time" |
| 9 | 9 |
| 10 "golang.org/x/net/context" | 10 "golang.org/x/net/context" |
| 11 | 11 |
| 12 "github.com/luci/gae/service/info" | 12 "github.com/luci/gae/service/info" |
| 13 ) | 13 ) |
| 14 | 14 |
| 15 type infoState struct { | 15 type infoState struct { |
| 16 *state | 16 *state |
| 17 | 17 |
| 18 » info.Interface | 18 » info.RawInterface |
| 19 } | 19 } |
| 20 | 20 |
| 21 func (g *infoState) ModuleHostname(module, version, instance string) (ret string
, err error) { | 21 func (g *infoState) ModuleHostname(module, version, instance string) (ret string
, err error) { |
| 22 err = g.run(func() (err error) { | 22 err = g.run(func() (err error) { |
| 23 » » ret, err = g.Interface.ModuleHostname(module, version, instance) | 23 » » ret, err = g.RawInterface.ModuleHostname(module, version, instan
ce) |
| 24 return | 24 return |
| 25 }) | 25 }) |
| 26 return | 26 return |
| 27 } | 27 } |
| 28 | 28 |
| 29 func (g *infoState) ServiceAccount() (ret string, err error) { | 29 func (g *infoState) ServiceAccount() (ret string, err error) { |
| 30 err = g.run(func() (err error) { | 30 err = g.run(func() (err error) { |
| 31 » » ret, err = g.Interface.ServiceAccount() | 31 » » ret, err = g.RawInterface.ServiceAccount() |
| 32 return | 32 return |
| 33 }) | 33 }) |
| 34 return | 34 return |
| 35 } | 35 } |
| 36 | 36 |
| 37 func (g *infoState) Namespace(namespace string) (ret context.Context, err error)
{ | 37 func (g *infoState) Namespace(namespace string) (ret context.Context, err error)
{ |
| 38 err = g.run(func() (err error) { | 38 err = g.run(func() (err error) { |
| 39 » » ret, err = g.Interface.Namespace(namespace) | 39 » » ret, err = g.RawInterface.Namespace(namespace) |
| 40 return | 40 return |
| 41 }) | 41 }) |
| 42 return | 42 return |
| 43 } | 43 } |
| 44 | 44 |
| 45 func (g *infoState) MustNamespace(namespace string) context.Context { | |
| 46 ret, err := g.Namespace(namespace) | |
| 47 if err != nil { | |
| 48 panic(err) | |
| 49 } | |
| 50 return ret | |
| 51 } | |
| 52 | |
| 53 func (g *infoState) AccessToken(scopes ...string) (token string, expiry time.Tim
e, err error) { | 45 func (g *infoState) AccessToken(scopes ...string) (token string, expiry time.Tim
e, err error) { |
| 54 err = g.run(func() (err error) { | 46 err = g.run(func() (err error) { |
| 55 » » token, expiry, err = g.Interface.AccessToken(scopes...) | 47 » » token, expiry, err = g.RawInterface.AccessToken(scopes...) |
| 56 return | 48 return |
| 57 }) | 49 }) |
| 58 return | 50 return |
| 59 } | 51 } |
| 60 | 52 |
| 61 func (g *infoState) PublicCertificates() (ret []info.Certificate, err error) { | 53 func (g *infoState) PublicCertificates() (ret []info.Certificate, err error) { |
| 62 err = g.run(func() (err error) { | 54 err = g.run(func() (err error) { |
| 63 » » ret, err = g.Interface.PublicCertificates() | 55 » » ret, err = g.RawInterface.PublicCertificates() |
| 64 return | 56 return |
| 65 }) | 57 }) |
| 66 return | 58 return |
| 67 } | 59 } |
| 68 | 60 |
| 69 func (g *infoState) SignBytes(bytes []byte) (keyName string, signature []byte, e
rr error) { | 61 func (g *infoState) SignBytes(bytes []byte) (keyName string, signature []byte, e
rr error) { |
| 70 err = g.run(func() (err error) { | 62 err = g.run(func() (err error) { |
| 71 » » keyName, signature, err = g.Interface.SignBytes(bytes) | 63 » » keyName, signature, err = g.RawInterface.SignBytes(bytes) |
| 72 return | 64 return |
| 73 }) | 65 }) |
| 74 return | 66 return |
| 75 } | 67 } |
| 76 | 68 |
| 77 // FilterGI installs a featureBreaker info filter in the context. | 69 // FilterGI installs a featureBreaker info filter in the context. |
| 78 func FilterGI(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { | 70 func FilterGI(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { |
| 79 state := newState(defaultError) | 71 state := newState(defaultError) |
| 80 » return info.AddFilters(c, func(ic context.Context, i info.Interface) inf
o.Interface { | 72 » return info.AddFilters(c, func(ic context.Context, i info.RawInterface)
info.RawInterface { |
| 81 return &infoState{state, i} | 73 return &infoState{state, i} |
| 82 }), state | 74 }), state |
| 83 } | 75 } |
| OLD | NEW |