diff options
| author | Chris Lu <chris.lu@gmail.com> | 2020-04-19 03:03:40 -0700 |
|---|---|---|
| committer | Chris Lu <chris.lu@gmail.com> | 2020-04-19 03:03:40 -0700 |
| commit | 71ffb98475d05c7e76e8c46a9525f29ef97f6228 (patch) | |
| tree | 133b4b4d5bbe685ac33334ec2e6ef759a619b71a /weed/messaging/client/client.go | |
| parent | d278b4c228905694a24cca2c9d08d8b8faa905bb (diff) | |
| download | seaweedfs-71ffb98475d05c7e76e8c46a9525f29ef97f6228.tar.xz seaweedfs-71ffb98475d05c7e76e8c46a9525f29ef97f6228.zip | |
broker: add profiling
Diffstat (limited to 'weed/messaging/client/client.go')
| -rw-r--r-- | weed/messaging/client/client.go | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/weed/messaging/client/client.go b/weed/messaging/client/client.go index 3f6d1ca53..910fd02e9 100644 --- a/weed/messaging/client/client.go +++ b/weed/messaging/client/client.go @@ -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() -} |