-
Notifications
You must be signed in to change notification settings - Fork 167
/
cells-log_grpc.enhanced.pb.go
124 lines (111 loc) · 4.12 KB
/
cells-log_grpc.enhanced.pb.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
// versions:
// - protoc-gen-go-grpc v1.1.0
// - protoc (unknown)
// source: cells-log.proto
package log
import (
context "context"
fmt "fmt"
grpc "google.golang.org/grpc"
codes "google.golang.org/grpc/codes"
metadata "google.golang.org/grpc/metadata"
status "google.golang.org/grpc/status"
sync "sync"
)
// This is a compile-time assertion to ensure that this generated file
// is compatible with the grpc package it is being compiled against.
// Requires gRPC-Go v1.32.0 or later.
const _ = grpc.SupportPackageIsVersion7
var (
enhancedLogRecorderServers = make(map[string]LogRecorderEnhancedServer)
enhancedLogRecorderServersLock = sync.RWMutex{}
)
type NamedLogRecorderServer interface {
LogRecorderServer
Name() string
}
type LogRecorderEnhancedServer map[string]NamedLogRecorderServer
// PutLog adds received log messages to the corresponding log repository.
func (m LogRecorderEnhancedServer) PutLog(s LogRecorder_PutLogServer) error {
md, ok := metadata.FromIncomingContext(s.Context())
if !ok || len(md.Get("targetname")) == 0 {
return status.Errorf(codes.FailedPrecondition, "method PutLog should have a context")
}
enhancedLogRecorderServersLock.RLock()
defer enhancedLogRecorderServersLock.RUnlock()
for _, mm := range m {
if mm.Name() == md.Get("targetname")[0] {
return mm.PutLog(s)
}
}
return status.Errorf(codes.Unimplemented, "method PutLog not implemented")
}
// ListLogs performs a paginated search query in the log repository.
func (m LogRecorderEnhancedServer) ListLogs(r *ListLogRequest, s LogRecorder_ListLogsServer) error {
md, ok := metadata.FromIncomingContext(s.Context())
if !ok || len(md.Get("targetname")) == 0 {
return status.Errorf(codes.FailedPrecondition, "method ListLogs should have a context")
}
enhancedLogRecorderServersLock.RLock()
defer enhancedLogRecorderServersLock.RUnlock()
for _, mm := range m {
if mm.Name() == md.Get("targetname")[0] {
return mm.ListLogs(r, s)
}
}
return status.Errorf(codes.Unimplemented, "method ListLogs not implemented")
}
// DeleteLogs deletes logs based on a request (cannot be empty)
func (m LogRecorderEnhancedServer) DeleteLogs(ctx context.Context, r *ListLogRequest) (*DeleteLogsResponse, error) {
md, ok := metadata.FromIncomingContext(ctx)
if !ok || len(md.Get("targetname")) == 0 {
return nil, status.Errorf(codes.FailedPrecondition, "method DeleteLogs should have a context")
}
enhancedLogRecorderServersLock.RLock()
defer enhancedLogRecorderServersLock.RUnlock()
for _, mm := range m {
if mm.Name() == md.Get("targetname")[0] {
return mm.DeleteLogs(ctx, r)
}
}
return nil, status.Errorf(codes.Unimplemented, "method DeleteLogs not implemented")
}
// AggregatedLogs performs a query to retrieve log events of the given type, faceted by time range.
func (m LogRecorderEnhancedServer) AggregatedLogs(r *TimeRangeRequest, s LogRecorder_AggregatedLogsServer) error {
md, ok := metadata.FromIncomingContext(s.Context())
if !ok || len(md.Get("targetname")) == 0 {
return status.Errorf(codes.FailedPrecondition, "method AggregatedLogs should have a context")
}
enhancedLogRecorderServersLock.RLock()
defer enhancedLogRecorderServersLock.RUnlock()
for _, mm := range m {
if mm.Name() == md.Get("targetname")[0] {
return mm.AggregatedLogs(r, s)
}
}
return status.Errorf(codes.Unimplemented, "method AggregatedLogs not implemented")
}
func (m LogRecorderEnhancedServer) mustEmbedUnimplementedLogRecorderServer() {}
func RegisterLogRecorderEnhancedServer(s grpc.ServiceRegistrar, srv NamedLogRecorderServer) {
enhancedLogRecorderServersLock.Lock()
defer enhancedLogRecorderServersLock.Unlock()
addr := fmt.Sprintf("%p", s)
m, ok := enhancedLogRecorderServers[addr]
if !ok {
m = LogRecorderEnhancedServer{}
enhancedLogRecorderServers[addr] = m
RegisterLogRecorderServer(s, m)
}
m[srv.Name()] = srv
}
func DeregisterLogRecorderEnhancedServer(s grpc.ServiceRegistrar, name string) {
enhancedLogRecorderServersLock.Lock()
defer enhancedLogRecorderServersLock.Unlock()
addr := fmt.Sprintf("%p", s)
m, ok := enhancedLogRecorderServers[addr]
if !ok {
return
}
delete(m, name)
}