/
query_artifacts.go
88 lines (75 loc) · 2.64 KB
/
query_artifacts.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// Copyright 2020 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package resultdb
import (
"context"
"cloud.google.com/go/spanner"
"github.com/golang/protobuf/ptypes"
"go.chromium.org/luci/common/errors"
"go.chromium.org/luci/grpc/appstatus"
"go.chromium.org/luci/server/span"
"go.chromium.org/luci/resultdb/internal/artifacts"
"go.chromium.org/luci/resultdb/internal/invocations"
"go.chromium.org/luci/resultdb/internal/pagination"
"go.chromium.org/luci/resultdb/pbutil"
pb "go.chromium.org/luci/resultdb/proto/v1"
)
// validateQueryArtifactsRequest returns a non-nil error if req is determined
// to be invalid.
func validateQueryArtifactsRequest(req *pb.QueryArtifactsRequest) error {
if err := pbutil.ValidateTestResultPredicate(req.TestResultPredicate); err != nil {
return errors.Annotate(err, "test_result_predicate").Err()
}
return validateQueryRequest(req)
}
// QueryArtifacts implements pb.ResultDBServer.
func (s *resultDBServer) QueryArtifacts(ctx context.Context, in *pb.QueryArtifactsRequest) (*pb.QueryArtifactsResponse, error) {
if err := verifyPermissionInvNames(ctx, permListArtifacts, in.Invocations...); err != nil {
return nil, err
}
if err := validateQueryArtifactsRequest(in); err != nil {
return nil, appstatus.BadRequest(err)
}
// Open a transaction.
ctx, cancel := span.ReadOnlyTransaction(ctx)
defer cancel()
if in.MaxStaleness != nil {
st, _ := ptypes.Duration(in.MaxStaleness)
span.RO(ctx).WithTimestampBound(spanner.MaxStaleness(st))
}
// Get the transitive closure.
invs, err := invocations.Reachable(ctx, invocations.MustParseNames(in.Invocations))
if err != nil {
return nil, err
}
// Query artifacts.
q := artifacts.Query{
InvocationIDs: invs,
TestResultPredicate: in.TestResultPredicate,
PageSize: pagination.AdjustPageSize(in.PageSize),
PageToken: in.PageToken,
FollowEdges: in.FollowEdges,
}
arts, token, err := q.Fetch(ctx)
if err != nil {
return nil, err
}
if err := s.populateFetchURLs(ctx, arts...); err != nil {
return nil, err
}
return &pb.QueryArtifactsResponse{
Artifacts: arts,
NextPageToken: token,
}, nil
}