blob: 91b76a3e004d7ea2dae517f2feda0ad37897c3cf [file] [log] [blame]
// 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/internal/permissions"
"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 := permissions.VerifyInvNames(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
}