inetmock/api
1
0
Fork 0
Code Issues 1 Pull requests Packages Projects Releases Wiki Activity

Apply comments

Browse source
This commit is contained in:
Peter 2021-01-16 18:26:15 +01:00
parent 38293bb8c4
commit 37c87adaf3
Signed by: prskr
GPG key ID: C1DB5D2E8DB512F9
6 changed files with 21 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api/proto/internal/rpc/audit.proto
View file

@ -26,7 +26,7 @@ message RemoveFileSinkRequest {
}
message RemoveFileSinkResponse {
bool SinkGotRemoved = 1;
}
service Audit {

2
cmd/imctl/endpoints.go
View file

@ -51,7 +51,7 @@ func fromEndpoints(eps []*rpc.Endpoint) (out []*printableEndpoint) {
func runGetEndpoints(_ *cobra.Command, _ []string) (err error) {
endpointsClient := rpc.NewEndpointsClient(conn)
ctx, cancel := context.WithTimeout(context.Background(), grpcTimeout)
ctx, cancel := context.WithTimeout(appCtx, grpcTimeout)
defer cancel()
var endpointsResp *rpc.GetEndpointsResponse
if endpointsResp, err = endpointsClient.GetEndpoints(ctx, &rpc.GetEndpointsRequest{}); err != nil {

19
internal/rpc/audit_server.go
View file

@ -5,20 +5,21 @@ import (
"io"
"os"
"gitlab.com/inetmock/inetmock/internal/app"
"gitlab.com/inetmock/inetmock/pkg/audit"
"gitlab.com/inetmock/inetmock/pkg/audit/sink"
"gitlab.com/inetmock/inetmock/pkg/logging"
"go.uber.org/zap"
)
type auditServer struct {
UnimplementedAuditServer
app app.App
logger logging.Logger
eventStream audit.EventStream
}
func (a *auditServer) WatchEvents(req *WatchEventsRequest, srv Audit_WatchEventsServer) (err error) {
a.app.Logger().Info("watcher attached", zap.String("name", req.WatcherName))
err = a.app.EventStream().RegisterSink(sink.NewGRPCSink(srv.Context(), req.WatcherName, func(ev audit.Event) {
a.logger.Info("watcher attached", zap.String("name", req.WatcherName))
err = a.eventStream.RegisterSink(sink.NewGRPCSink(srv.Context(), req.WatcherName, func(ev audit.Event) {
if err = srv.Send(ev.ProtoMessage()); err != nil {
return
}
@ -29,7 +30,7 @@ func (a *auditServer) WatchEvents(req *WatchEventsRequest, srv Audit_WatchEvents
}
<-srv.Context().Done()
a.app.Logger().Info("Watcher detached", zap.String("name", req.WatcherName))
a.logger.Info("Watcher detached", zap.String("name", req.WatcherName))
return
}
@ -38,7 +39,7 @@ func (a *auditServer) RegisterFileSink(_ context.Context, req *RegisterFileSinkR
if writer, err = os.OpenFile(req.TargetPath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644); err != nil {
return
}
if err = a.app.EventStream().RegisterSink(sink.NewWriterSink(req.TargetPath, audit.NewEventWriter(writer))); err != nil {
if err = a.eventStream.RegisterSink(sink.NewWriterSink(req.TargetPath, audit.NewEventWriter(writer))); err != nil {
return
}
resp = &RegisterFileSinkResponse{}
@ -46,6 +47,8 @@ func (a *auditServer) RegisterFileSink(_ context.Context, req *RegisterFileSinkR
}
func (a *auditServer) RemoveFileSink(_ context.Context, req *RemoveFileSinkRequest) (*RemoveFileSinkResponse, error) {
a.app.EventStream().RemoveSink(req.TargetPath)
return &RemoveFileSinkResponse{}, nil
gotRemoved := a.eventStream.RemoveSink(req.TargetPath)
return &RemoveFileSinkResponse{
SinkGotRemoved: gotRemoved,
}, nil
}

3
internal/rpc/grpc_api.go
View file

@ -53,7 +53,8 @@ func (i *inetmockAPI) StartServer() (err error) {
})
RegisterAuditServer(i.server, &auditServer{
app: i.app,
logger: i.app.Logger(),
eventStream: i.app.EventStream(),
})
go i.startServerAsync(lis)

2
pkg/audit/api.go
View file

@ -27,5 +27,5 @@ type EventStream interface {
Emitter
RegisterSink(s Sink) error
Sinks() []string
RemoveSink(name string)
RemoveSink(name string) (exists bool)
}

7
pkg/audit/event_stream.go
View file

@ -72,11 +72,12 @@ func (e *eventStream) Emit(ev Event) {
}
}
func (e *eventStream) RemoveSink(name string) {
func (e *eventStream) RemoveSink(name string) (exists bool) {
e.lock.Lock()
defer e.lock.Unlock()
sink, exists := e.sinks[name]
var sink *registeredSink
sink, exists = e.sinks[name]
if !exists {
return
}
@ -84,6 +85,8 @@ func (e *eventStream) RemoveSink(name string) {
defer sink.lock.Unlock()
delete(e.sinks, name)
close(sink.downstream)
return
}
func (e *eventStream) RegisterSink(s Sink) error {