main.go 1.0 KB
Newer Older
1 2 3 4 5 6 7 8
package main

import (
	"context"
	"os"
	"os/signal"
	"syscall"

N
neza2017 已提交
9
	distributed "github.com/zilliztech/milvus-distributed/cmd/distributed/components"
N
neza2017 已提交
10 11
	"github.com/zilliztech/milvus-distributed/internal/log"
	"github.com/zilliztech/milvus-distributed/internal/masterservice"
G
groot 已提交
12
	"github.com/zilliztech/milvus-distributed/internal/msgstream/pulsarms"
N
neza2017 已提交
13
	"go.uber.org/zap"
14 15 16 17 18 19
)

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

N
neza2017 已提交
20 21 22 23 24 25 26 27
	masterservice.Params.Init()
	log.SetupLogger(&masterservice.Params.Log)
	defer func() {
		if err := log.Sync(); err != nil {
			panic(err)
		}
	}()

G
groot 已提交
28 29
	msFactory := pulsarms.NewFactory()
	ms, err := distributed.NewMasterService(ctx, msFactory)
30 31 32
	if err != nil {
		panic(err)
	}
N
neza2017 已提交
33
	if err = ms.Run(); err != nil {
34 35 36 37 38 39 40 41 42 43
		panic(err)
	}

	sc := make(chan os.Signal, 1)
	signal.Notify(sc,
		syscall.SIGHUP,
		syscall.SIGINT,
		syscall.SIGTERM,
		syscall.SIGQUIT)
	sig := <-sc
B
bigsheeper 已提交
44
	log.Debug("Get signal to exit", zap.String("signal", sig.String()))
N
neza2017 已提交
45 46 47 48
	err = ms.Stop()
	if err != nil {
		panic(err)
	}
49
}