fix exit issue

zc2638 · Apr 13, 2022 · 3252cef · 3252cef
1 parent f634ffa
commit 3252cef
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cmd/ddshop/app/app.go b/cmd/ddshop/app/app.go
@@ -62,8 +62,16 @@ func NewRootCommand() *cobra.Command {
 				return err
 			}
 			fmt.Println()
+
+			ctx, cancelFunc := context.WithCancel(context.Background())
 			go func() {
 				for {
+					select {
+					case <-ctx.Done():
+						logrus.Warningf("context done")
+						return
+					default:
+					}
 					if err := Start(session); err != nil {
 						switch err {
 						case core.ErrorNoValidProduct:
@@ -84,8 +92,10 @@ func NewRootCommand() *cobra.Command {
 
 			select {
 			case err := <-errCh:
+				cancelFunc()
 				return err
 			case <-successCh:
+				cancelFunc()
 				core.LoopRun(10, func() {
 					logrus.Info("抢菜成功，请尽快支付!")
 				})