lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-25 08:15:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

cmd: graceful close the connection

Browse Source
This commit is contained in:
c9s 2021-12-23 14:21:26 +08:00
parent 562c287a4e
commit e2415857b0
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/cmd/userdatastream.go
View File

@ -58,7 +58,12 @@ var userDataStreamCmd = &cobra.Command{
} }
log.Infof("connected") log.Infof("connected")
defer s.Close() defer func() {
log.Infof("closing connection...")
if err := s.Close(); err != nil {
log.WithError(err).Errorf("connection close error")
}
}()
cmdutil.WaitForSignal(ctx, syscall.SIGINT, syscall.SIGTERM) cmdutil.WaitForSignal(ctx, syscall.SIGINT, syscall.SIGTERM)
return nil return nil