OLD | NEW |
1 // Copyright 2016 The LUCI Authors. | 1 // Copyright 2016 The LUCI Authors. |
2 // | 2 // |
3 // Licensed under the Apache License, Version 2.0 (the "License"); | 3 // Licensed under the Apache License, Version 2.0 (the "License"); |
4 // you may not use this file except in compliance with the License. | 4 // you may not use this file except in compliance with the License. |
5 // You may obtain a copy of the License at | 5 // You may obtain a copy of the License at |
6 // | 6 // |
7 // http://www.apache.org/licenses/LICENSE-2.0 | 7 // http://www.apache.org/licenses/LICENSE-2.0 |
8 // | 8 // |
9 // Unless required by applicable law or agreed to in writing, software | 9 // Unless required by applicable law or agreed to in writing, software |
10 // distributed under the License is distributed on an "AS IS" BASIS, | 10 // distributed under the License is distributed on an "AS IS" BASIS, |
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 // See the License for the specific language governing permissions and | 12 // See the License for the specific language governing permissions and |
13 // limitations under the License. | 13 // limitations under the License. |
14 | 14 |
15 package rawpresentation | 15 package rawpresentation |
16 | 16 |
17 import ( | 17 import ( |
18 » "errors" | 18 » "strings" |
19 » "fmt" | |
20 "time" | 19 "time" |
21 | 20 |
| 21 "github.com/golang/protobuf/proto" |
| 22 "golang.org/x/net/context" |
| 23 |
| 24 "github.com/luci/luci-go/common/errors" |
22 log "github.com/luci/luci-go/common/logging" | 25 log "github.com/luci/luci-go/common/logging" |
23 "github.com/luci/luci-go/common/proto/google" | 26 "github.com/luci/luci-go/common/proto/google" |
24 miloProto "github.com/luci/luci-go/common/proto/milo" | 27 miloProto "github.com/luci/luci-go/common/proto/milo" |
25 "github.com/luci/luci-go/logdog/api/logpb" | 28 "github.com/luci/luci-go/logdog/api/logpb" |
26 "github.com/luci/luci-go/logdog/client/coordinator" | 29 "github.com/luci/luci-go/logdog/client/coordinator" |
27 "github.com/luci/luci-go/logdog/common/types" | 30 "github.com/luci/luci-go/logdog/common/types" |
28 "github.com/luci/luci-go/logdog/common/viewer" | 31 "github.com/luci/luci-go/logdog/common/viewer" |
29 "github.com/luci/luci-go/luci_config/common/cfgtypes" | 32 "github.com/luci/luci-go/luci_config/common/cfgtypes" |
30 "github.com/luci/luci-go/milo/api/resp" | 33 "github.com/luci/luci-go/milo/api/resp" |
31 "github.com/luci/luci-go/milo/buildsource/rawpresentation/internal" | 34 "github.com/luci/luci-go/milo/buildsource/rawpresentation/internal" |
32 | 35 » "github.com/luci/luci-go/milo/common" |
33 » "github.com/golang/protobuf/proto" | |
34 » "golang.org/x/net/context" | |
35 ) | 36 ) |
36 | 37 |
37 const ( | 38 const ( |
38 // intermediateCacheLifetime is the amount of time to cache intermediate
(non- | 39 // intermediateCacheLifetime is the amount of time to cache intermediate
(non- |
39 // terminal) annotation streams. Terminal annotation streams are cached | 40 // terminal) annotation streams. Terminal annotation streams are cached |
40 // indefinitely. | 41 // indefinitely. |
41 intermediateCacheLifetime = 10 * time.Second | 42 intermediateCacheLifetime = 10 * time.Second |
42 | 43 |
43 // defaultLogDogHost is the default LogDog host, if one isn't specified
via | 44 // defaultLogDogHost is the default LogDog host, if one isn't specified
via |
44 // query string. | 45 // query string. |
45 defaultLogDogHost = "luci-logdog.appspot.com" | 46 defaultLogDogHost = "luci-logdog.appspot.com" |
46 ) | 47 ) |
47 | 48 |
48 // AnnotationStream represents a LogDog annotation protobuf stream. | 49 // AnnotationStream represents a LogDog annotation protobuf stream. |
49 type AnnotationStream struct { | 50 type AnnotationStream struct { |
50 Project cfgtypes.ProjectName | 51 Project cfgtypes.ProjectName |
51 Path types.StreamPath | 52 Path types.StreamPath |
52 | 53 |
53 // Client is the HTTP client to use for LogDog communication. | 54 // Client is the HTTP client to use for LogDog communication. |
54 Client *coordinator.Client | 55 Client *coordinator.Client |
55 | 56 |
56 // cs is the unmarshalled annotation stream Step and associated data. | 57 // cs is the unmarshalled annotation stream Step and associated data. |
57 cs internal.CachedStep | 58 cs internal.CachedStep |
58 } | 59 } |
59 | 60 |
60 // Normalize validates and normalizes the stream's parameters. | 61 // Normalize validates and normalizes the stream's parameters. |
61 func (as *AnnotationStream) Normalize() error { | 62 func (as *AnnotationStream) Normalize() error { |
62 if err := as.Project.Validate(); err != nil { | 63 if err := as.Project.Validate(); err != nil { |
63 » » return fmt.Errorf("Invalid project name: %s", as.Project) | 64 » » return errors.Annotate(err, "Invalid project name: %s", as.Proje
ct).Tag(common.CodeParameterError).Err() |
64 } | 65 } |
65 | 66 |
66 if err := as.Path.Validate(); err != nil { | 67 if err := as.Path.Validate(); err != nil { |
67 » » return fmt.Errorf("Invalid log stream path %q: %s", as.Path, err
) | 68 » » return errors.Annotate(err, "Invalid log stream path %q", as.Pat
h).Tag(common.CodeParameterError).Err() |
68 } | 69 } |
69 | 70 |
70 return nil | 71 return nil |
71 } | 72 } |
72 | 73 |
73 var errNotMilo = errors.New("Requested stream is not a Milo annotation protobuf"
) | 74 var errNotMilo = errors.New("Requested stream is not a Milo annotation protobuf"
) |
74 var errNotDatagram = errors.New("Requested stream is not a datagram stream") | 75 var errNotDatagram = errors.New("Requested stream is not a datagram stream") |
75 var errNoEntries = errors.New("Log stream has no annotation entries") | 76 var errNoEntries = errors.New("Log stream has no annotation entries") |
76 | 77 |
77 // Fetch loads the annotation stream from LogDog. | 78 // Fetch loads the annotation stream from LogDog. |
(...skipping 146 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
224 if link.Label == "" { | 225 if link.Label == "" { |
225 link.Label = "unnamed" | 226 link.Label = "unnamed" |
226 } | 227 } |
227 return link | 228 return link |
228 | 229 |
229 default: | 230 default: |
230 // Don't know how to render. | 231 // Don't know how to render. |
231 return nil | 232 return nil |
232 } | 233 } |
233 } | 234 } |
| 235 |
| 236 // BuildID implements buildsource.ID. |
| 237 type BuildID struct { |
| 238 Host string |
| 239 Project cfgtypes.ProjectName |
| 240 Path types.StreamPath |
| 241 } |
| 242 |
| 243 // GetLog implements buildsource.ID. |
| 244 func (b *BuildID) GetLog(context.Context, string) (string, bool, error) { panic(
"not implemented") } |
| 245 |
| 246 // Get implements buildsource.ID. |
| 247 func (b *BuildID) Get(c context.Context) (*resp.MiloBuild, error) { |
| 248 as := AnnotationStream{ |
| 249 Project: b.Project, |
| 250 Path: b.Path, |
| 251 } |
| 252 if err := as.Normalize(); err != nil { |
| 253 return nil, err |
| 254 } |
| 255 |
| 256 // Setup our LogDog client. |
| 257 var err error |
| 258 if as.Client, err = NewClient(c, b.Host); err != nil { |
| 259 return nil, errors.Annotate(err, "generating LogDog Client").Err
() |
| 260 } |
| 261 |
| 262 // Load the Milo annotation protobuf from the annotation stream. |
| 263 switch _, err := as.Fetch(c); errors.Unwrap(err) { |
| 264 case nil, errNoEntries: |
| 265 |
| 266 case coordinator.ErrNoSuchStream: |
| 267 return nil, common.CodeNotFound.Tag().Apply(err) |
| 268 |
| 269 case coordinator.ErrNoAccess: |
| 270 return nil, common.CodeNoAccess.Tag().Apply(err) |
| 271 |
| 272 case errNotMilo, errNotDatagram: |
| 273 // The user requested a LogDog url that isn't a Milo annotation. |
| 274 return nil, common.CodeParameterError.Tag().Apply(err) |
| 275 |
| 276 default: |
| 277 return nil, errors.Annotate(err, "failed to load stream").Err() |
| 278 } |
| 279 |
| 280 return as.toMiloBuild(c), nil |
| 281 } |
| 282 |
| 283 // NewBuildID generates a new un-validated BuildID. |
| 284 func NewBuildID(host, project, path string) *BuildID { |
| 285 return &BuildID{ |
| 286 strings.TrimSpace(host), |
| 287 cfgtypes.ProjectName(project), |
| 288 types.StreamPath(strings.Trim(path, "/")), |
| 289 } |
| 290 } |
OLD | NEW |