瀏覽代碼

fix(cli): don't forget to apply changes to ovpmd cmd

Mustafa Arici 8 年之前
父節點
當前提交
a83447d466
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      cmd/ovpmd/main.go

+ 3 - 2
cmd/ovpmd/main.go

@@ -24,6 +24,7 @@ import (
 )
 
 var action string
+var db *ovpm.DB
 
 func main() {
 	app := cli.NewApp()
@@ -45,11 +46,11 @@ func main() {
 		if c.GlobalBool("verbose") {
 			logrus.SetLevel(logrus.DebugLevel)
 		}
-		ovpm.SetupDB("sqlite3", "")
+		db = ovpm.CreateDB("sqlite3", "")
 		return nil
 	}
 	app.After = func(c *cli.Context) error {
-		ovpm.CeaseDB()
+		db.Cease()
 		return nil
 	}
 	app.Action = func(c *cli.Context) error {