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 |