-
Notifications
You must be signed in to change notification settings - Fork 110
/
mygizmo.go
100 lines (85 loc) · 2.4 KB
/
mygizmo.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
89
90
91
92
93
94
95
96
97
98
99
100
// Package mygizmo implements an acme:component:gizmo, a demonstration component that simply shows the various methods available in grpc.
package mygizmo
import (
"context"
"fmt"
"sync"
"github.com/edaniels/golog"
"go.viam.com/rdk/examples/customresources/apis/gizmoapi"
"go.viam.com/rdk/resource"
)
var Model = resource.NewModel("acme", "demo", "mygizmo")
func init() {
resource.RegisterComponent(gizmoapi.API, Model, resource.Registration[gizmoapi.Gizmo, resource.NoNativeConfig]{
Constructor: func(
ctx context.Context,
deps resource.Dependencies,
conf resource.Config,
logger golog.Logger,
) (gizmoapi.Gizmo, error) {
return NewMyGizmo(deps, conf, logger)
},
})
}
type myActualGizmo struct {
resource.Named
resource.TriviallyCloseable
myArgMu sync.Mutex
myArg string
}
func NewMyGizmo(
deps resource.Dependencies,
conf resource.Config,
logger golog.Logger,
) (gizmoapi.Gizmo, error) {
g := &myActualGizmo{
Named: conf.ResourceName().AsNamed(),
}
if err := g.Reconfigure(context.Background(), deps, conf); err != nil {
return nil, err
}
return g, nil
}
func (g *myActualGizmo) Reconfigure(ctx context.Context, deps resource.Dependencies, conf resource.Config) error {
g.myArgMu.Lock()
g.myArg = conf.Attributes.String("arg1")
g.myArgMu.Unlock()
return nil
}
func (g *myActualGizmo) DoOne(ctx context.Context, arg1 string) (bool, error) {
g.myArgMu.Lock()
defer g.myArgMu.Unlock()
return arg1 == g.myArg, nil
}
func (g *myActualGizmo) DoOneClientStream(ctx context.Context, arg1 []string) (bool, error) {
g.myArgMu.Lock()
defer g.myArgMu.Unlock()
if len(arg1) == 0 {
return false, nil
}
ret := true
for _, arg := range arg1 {
ret = ret && arg == g.myArg
}
return ret, nil
}
func (g *myActualGizmo) DoOneServerStream(ctx context.Context, arg1 string) ([]bool, error) {
g.myArgMu.Lock()
defer g.myArgMu.Unlock()
return []bool{arg1 == g.myArg, false, true, false}, nil
}
func (g *myActualGizmo) DoOneBiDiStream(ctx context.Context, arg1 []string) ([]bool, error) {
g.myArgMu.Lock()
defer g.myArgMu.Unlock()
var rets []bool
for _, arg := range arg1 {
rets = append(rets, arg == g.myArg)
}
return rets, nil
}
func (g *myActualGizmo) DoTwo(ctx context.Context, arg1 bool) (string, error) {
return fmt.Sprintf("arg1=%t", arg1), nil
}
func (g *myActualGizmo) DoCommand(ctx context.Context, cmd map[string]interface{}) (map[string]interface{}, error) {
return cmd, nil
}