-
Notifications
You must be signed in to change notification settings - Fork 328
/
entity.go
69 lines (59 loc) · 1.98 KB
/
entity.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
// Copyright (C) 2017 Google Inc.
//
// 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 local
import (
"context"
"reflect"
"sync"
"github.com/google/gapid/core/event"
"github.com/google/gapid/core/log"
"github.com/google/gapid/test/robot/search"
"github.com/google/gapid/test/robot/search/eval"
"github.com/google/gapid/test/robot/stash"
)
var entityClass = reflect.TypeOf(&stash.Entity{})
type entityIndex struct {
mu sync.Mutex
entities []*stash.Entity
byID map[string]*stash.Entity
onAdd event.Broadcast
}
func (i *entityIndex) init() {
i.entities = []*stash.Entity{}
i.byID = map[string]*stash.Entity{}
}
func (i *entityIndex) lockedAddEntry(ctx context.Context, entity *stash.Entity) {
i.entities = append(i.entities, entity)
i.byID[entity.Upload.Id] = entity
if err := i.onAdd.Send(ctx, entity); err != nil {
log.E(ctx, "Stash notification failed. Error: %v", err)
}
}
func (e *entityIndex) Lookup(ctx context.Context, id string) (*stash.Entity, error) {
e.mu.Lock()
defer e.mu.Unlock()
entity, found := e.byID[id]
if !found {
return nil, stash.ErrEntityNotFound
}
return entity, nil
}
func (e *entityIndex) Search(ctx context.Context, query *search.Query, handler stash.EntityHandler) error {
filter := eval.Filter(ctx, query, entityClass, event.AsHandler(ctx, handler))
initial := event.AsProducer(ctx, e.entities)
if query.Monitor {
return event.Monitor(ctx, &e.mu, e.onAdd.Listen, initial, filter)
}
return event.Feed(ctx, filter, initial)
}