milvus/cmd/datanode/main.go
XuanYang-cn 10374f15d7 Update log in datanode
Signed-off-by: XuanYang-cn <xuan.yang@zilliz.com>
2021-02-26 10:13:36 +08:00

50 lines
945 B
Go

package main
import (
"context"
"os"
"os/signal"
"syscall"
"go.uber.org/zap"
dn "github.com/zilliztech/milvus-distributed/internal/datanode"
distributed "github.com/zilliztech/milvus-distributed/cmd/distributed/components"
"github.com/zilliztech/milvus-distributed/internal/log"
"github.com/zilliztech/milvus-distributed/internal/msgstream/pulsarms"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
msFactory := pulsarms.NewFactory()
dn.Params.Init()
log.SetupLogger(&dn.Params.Log)
dn, err := distributed.NewDataNode(ctx, msFactory)
if err != nil {
panic(err)
}
if err = dn.Run(); err != nil {
panic(err)
}
sc := make(chan os.Signal, 1)
signal.Notify(sc,
syscall.SIGHUP,
syscall.SIGINT,
syscall.SIGTERM,
syscall.SIGQUIT)
sig := <-sc
log.Debug("Got signal to exit signal", zap.String("signal", sig.String()))
err = dn.Stop()
if err != nil {
panic(err)
}
}