fixbug

pushzhang · Dec 14, 2021 · b6a74eb · b6a74eb
1 parent ce8e0d8
commit b6a74eb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/cmd/server/app/command.go b/cmd/server/app/command.go
@@ -32,12 +32,13 @@ func NewCommand() (cmd *cobra.Command) {
 			RunEvent()
 			var nodeCtx context.Context
 			var err error
-			if nodeCtx, err = RunNode(context.Background()); err != nil {
+			ctx := context.Background()
+			if nodeCtx, err = RunNode(ctx); err != nil {
 				fmt.Println(err.Error())
 				os.Exit(1)
 			}
 			event.Event(event2.NewInfoNode("Run node success"))
-			if err = RunConsole(context.Background()); err != nil {
+			if err = RunConsole(ctx); err != nil {
 				fmt.Println(err.Error())
 				os.Exit(1)
 			}

diff --git a/cmd/server/app/console.go b/cmd/server/app/console.go
@@ -7,7 +7,7 @@ import (
 )
 
 // RunConsole run gin
-func RunConsole() (err error) {
-	err = console.Run(context.Background())
+func RunConsole(c context.Context) (err error) {
+	err = console.Run(c)
 	return
 }