-
Notifications
You must be signed in to change notification settings - Fork 0
/
state.go
94 lines (76 loc) · 2.27 KB
/
state.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
/*
Copyright 2022 Nokia.
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 wireproxy
import (
"context"
"sync"
"github.com/go-logr/logr"
"github.com/henderiw-nephio/wire-connector/pkg/proto/wirepb"
"github.com/henderiw-nephio/wire-connector/pkg/wirer"
"google.golang.org/grpc/peer"
"k8s.io/apimachinery/pkg/types"
ctrl "sigs.k8s.io/controller-runtime"
)
type stateConfig struct {
be wirer.Node2NodeWirer
}
type clientContext struct {
stream wirepb.Wire_WireWatchServer
cancel context.CancelFunc
}
type s struct {
m sync.RWMutex
// key is clientName
clients map[string]*clientContext
be wirer.Node2NodeWirer
l logr.Logger
}
func NewProxyState(cfg *stateConfig) *s {
l := ctrl.Log.WithName("proxy-state")
return &s{
clients: map[string]*clientContext{},
be: cfg.be,
l: l,
}
}
func (r *s) AddWireCallBackFn(req *wirepb.WatchRequest, stream wirepb.Wire_WireWatchServer) {
p, _ := peer.FromContext(stream.Context())
r.m.Lock()
// cancelFn if a client adss another entry the client is misbehaving
if clientCtx, ok := r.clients[p.Addr.String()]; ok {
clientCtx.cancel()
}
ctx, cancel := context.WithCancel(stream.Context())
r.clients[p.Addr.String()] = &clientContext{
stream: stream,
cancel: cancel,
}
r.m.Unlock()
// we already validated the existance of the backend before calling this function
r.be.AddWireWatch(r.CreateCallBackFn(stream))
for range ctx.Done() {
r.DeleteWireCallBackFn(p.Addr.String())
r.l.Info("watch stopped", "address", p.Addr.String())
return
}
}
func (r *s) DeleteWireCallBackFn(clientName string) {
r.m.Lock()
defer r.m.Unlock()
r.be.DeleteWireWatch()
delete(r.clients, clientName)
}
func (r *s) CreateCallBackFn(stream wirepb.Wire_WireWatchServer) wirer.CallbackFn {
// TODO
return func([]types.NamespacedName, string) {}
}