/
server.go
76 lines (70 loc) · 2.07 KB
/
server.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
package repo
import (
"context"
"github.com/infinimesh/infinimesh/pkg/repo/repopb"
"go.uber.org/zap"
)
type Server struct {
Repo Repo
Log *zap.Logger
}
func (s *Server) Get(context context.Context, request *repopb.GetRequest) (response *repopb.GetResponse, err error) {
log := s.Log.Named("Get Device Status Controller")
log.Info("Function Invoked", zap.String("Device", request.Id))
response = &repopb.GetResponse{
Repo: &repopb.Repo{},
}
deviceStatus, err := s.Repo.GetDeviceStatus(request.Id)
if err != nil {
deviceStatus.ID = request.Id
deviceStatus.Status = DeviceStatus{
Fingerprint: []byte("{}"),
Enabled: false,
NamespaceID: "",
}
}
response.Repo = &repopb.Repo{
Enabled: deviceStatus.Status.Enabled,
NamespaceID: deviceStatus.Status.NamespaceID,
FingerPrint: deviceStatus.Status.Fingerprint,
}
return response, nil
}
func (s *Server) SetDeviceState(context context.Context, request *repopb.SetDeviceStateRequest) (response *repopb.SetDeviceStateResponse, err error) {
log := s.Log.Named("Set Device Status Controller")
log.Info("Function Invoked", zap.String("Device", request.Id))
deviceState := DeviceState{
ID: request.Id,
Status: DeviceStatus{
Fingerprint: request.Repo.FingerPrint,
Enabled: request.Repo.Enabled,
NamespaceID: request.Repo.NamespaceID,
},
}
err = s.Repo.SetDeviceStatus(deviceState)
if err != nil {
response := &repopb.SetDeviceStateResponse{
Status: false,
}
return response, err
}
response = &repopb.SetDeviceStateResponse{
Status: true,
}
return response, nil
}
func (s *Server) DeleteDeviceState(context context.Context, request *repopb.DeleteDeviceStateRequest) (response *repopb.DeleteDeviceStateResponse, err error) {
log := s.Log.Named("Delete Device Status Controller")
log.Info("Function Invoked", zap.String("Device", request.Id))
err = s.Repo.DeleteDeviceStatus(request.Id)
if err != nil {
response := &repopb.DeleteDeviceStateResponse{
Status: false,
}
return response, err
}
response = &repopb.DeleteDeviceStateResponse{
Status: true,
}
return response, nil
}