blob: ac43d15c3e12073b2edf72a8e61a0de61a881b2f [file] [log] [blame]
// Copyright 2020 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//go:build linux || darwin
// +build linux darwin
package dashboard
import (
"context"
"errors"
"log"
"cloud.google.com/go/datastore"
)
// getDatastoreResults populates result data on commits, fetched from Datastore.
func getDatastoreResults(ctx context.Context, cl *datastore.Client, commits []*commit, pkg string) {
var keys []*datastore.Key
for _, c := range commits {
pkey := datastore.NameKey("Package", pkg, nil)
pkey.Namespace = "Git"
key := datastore.NameKey("Commit", "|"+c.Hash, pkey)
key.Namespace = "Git"
keys = append(keys, key)
}
out := make([]*Commit, len(keys))
// datastore.ErrNoSuchEntity is returned when we ask for a commit that we do not yet have test data.
if err := cl.GetMulti(ctx, keys, out); err != nil && filterMultiError(err, ignoreNoSuchEntity) != nil {
log.Printf("getResults: error fetching %d results: %v", len(keys), err)
return
}
hashOut := make(map[string]*Commit)
for _, o := range out {
if o != nil && o.Hash != "" {
hashOut[o.Hash] = o
}
}
for _, c := range commits {
if result, ok := hashOut[c.Hash]; ok {
c.ResultData = result.ResultData
}
}
return
}
type ignoreFunc func(err error) error
// ignoreNoSuchEntity ignores datastore.ErrNoSuchEntity, which is returned when
// we ask for a commit that we do not yet have test data.
func ignoreNoSuchEntity(err error) error {
if !errors.Is(err, datastore.ErrNoSuchEntity) {
return err
}
return nil
}
// filterMultiError loops over datastore.MultiError, skipping errors ignored by
// the specified ignoreFuncs. Any unfiltered errors will be returned as a
// datastore.MultiError error. If no errors are left, nil will be returned.
// Errors that are not datastore.MultiError will be returned as-is.
func filterMultiError(err error, ignores ...ignoreFunc) error {
if err == nil {
return nil
}
me := datastore.MultiError{}
if ok := errors.As(err, &me); !ok {
return err
}
ret := datastore.MultiError{}
for _, err := range me {
var skip bool
for _, ignore := range ignores {
if err := ignore(err); err == nil {
skip = true
break
}
}
if !skip {
ret = append(ret, err)
}
}
if len(ret) > 0 {
return ret
}
return nil
}