Function main

IceFireDB-SQLite/main.go:18–67

()

)

func main() {
	app := cli.NewApp()

	app.Name = "github.com/IceFireDB/IceFireDB/IceFireDB-SQLite"

	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:     "config, c",
			Usage:    "config file path",
			Value:    "config/config.yaml",
			Required: false,
		},
		cli.StringFlag{
			Name:  "log,l",
			Usage: "log level: debug,info,warning,error",
			Value: "info",
		},
	}

	app.Before = func(c *cli.Context) error {
		log.SetFlags(log.Llongfile)
		// init log
		lv, err := logrus.ParseLevel(c.String("log"))
		if err != nil {
			return err
		}
		logrus.SetLevel(lv)
		// init config
		confPath := c.String("config")
		config.InitConfig(confPath)
		return nil
	}

	app.Action = func(c *cli.Context) error {
		ctx, cancel := context.WithCancel(context.Background())
		stop := make(chan struct{})

		go func() {
			ms := mysql.NewMysqlProxy()
			ms.Run(ctx)
			stop <- struct{}{}
		}()

		return exitSignal(cancel, stop)
	}
	err := app.Run(os.Args)
	if err != nil {
		panic(err)
	}
}

func exitSignal(cancel context.CancelFunc, stop chan struct{}) error {
	sigs := make(chan os.Signal, 1)