milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Proxy exit when disconnected to etcd (#13180) 

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
pull/13213/head
dragondriver 2021-12-10 22:09:17 +08:00 committed by GitHub
parent 3f0970153c
commit 175f3147d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
internal/proxy/proxy.go
View File

@ -22,6 +22,7 @@ import (
"math/rand"
"sync"
"sync/atomic"
"syscall"
"time"
"github.com/milvus-io/milvus/internal/util/metricsinfo"
@ -118,6 +119,7 @@ func (node *Proxy) Register() error {
if err := node.Stop(); err != nil {
log.Fatal("failed to stop server", zap.Error(err))
}
syscall.Kill(syscall.Getpid(), syscall.SIGINT)
})
// TODO Reset the logger
//Params.initLogCfg()
@ -347,18 +349,25 @@ func (node *Proxy) Stop() error {
if node.idAllocator != nil {
node.idAllocator.Close()
log.Info("close id allocator", zap.String("role", Params.RoleName))
}
if node.segAssigner != nil {
node.segAssigner.Close()
log.Info("close segment id assigner", zap.String("role", Params.RoleName))
}
if node.sched != nil {
node.sched.Close()
log.Info("close scheduler", zap.String("role", Params.RoleName))
}
if node.chTicker != nil {
err := node.chTicker.close()
if err != nil {
return err
}
log.Info("close channels time ticker", zap.String("role", Params.RoleName))
}
node.wg.Wait()