broker: add profiling

This commit is contained in:
Chris Lu
2020-04-19 03:03:40 -07:00
parent d278b4c228
commit 71ffb98475
8 changed files with 83 additions and 67 deletions

View File

@@ -23,12 +23,12 @@ func NewMessagingClient(bootstrapBrokers []string) (*MessagingClient, error) {
return nil, err
}
util.OnInterrupt(func() {
grpcConnection.Close()
})
return &MessagingClient{
bootstrapBrokers: bootstrapBrokers,
grpcConnection: grpcConnection,
}, nil
}
func (mc *MessagingClient) Shutdown() {
mc.grpcConnection.Close()
}