-
Notifications
You must be signed in to change notification settings - Fork 110
/
collectors.go
43 lines (35 loc) · 1.03 KB
/
collectors.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
package movementsensor
import (
"context"
"google.golang.org/protobuf/types/known/anypb"
"go.viam.com/rdk/data"
)
// TODO: add tests for this file.
func assertMovementSensor(resource interface{}) (MovementSensor, error) {
ms, ok := resource.(MovementSensor)
if !ok {
return nil, data.InvalidInterfaceErr(SubtypeName)
}
return ms, nil
}
type lowLevelCollector func(ctx context.Context, ms MovementSensor) (interface{}, error)
func registerCollector(name string, f lowLevelCollector) {
data.RegisterCollector(data.MethodMetadata{
Subtype: Subtype,
MethodName: name,
}, func(resource interface{}, params data.CollectorParams) (data.Collector, error) {
ms, err := assertMovementSensor(resource)
if err != nil {
return nil, err
}
cFunc := data.CaptureFunc(func(ctx context.Context, extra map[string]*anypb.Any) (interface{}, error) {
v, err := f(ctx, ms)
if err != nil {
return nil, data.FailedToReadErr(params.ComponentName, name, err)
}
return v, nil
})
return data.NewCollector(cFunc, params)
},
)
}