From d3349edf0901299efff9d1947457c88182bece7f Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Sat, 14 Jan 2023 18:02:12 +0800
Subject: [PATCH] [optimize] yao start command

---
 cmd/start.go     | 17 +++++++++++++++--
 share/session.go |  2 +-
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/cmd/start.go b/cmd/start.go
index 3eb504f3..454c4757 100644
--- a/cmd/start.go
+++ b/cmd/start.go
@@ -118,7 +118,13 @@ var startCmd = &cobra.Command{
 		if mode == "development" {
 
 			// Start Studio Server
-			go studio.Start(config.Conf)
+			go func() {
+				err := studio.Start(config.Conf)
+				if err != nil {
+					fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
+					os.Exit(1)
+				}
+			}()
 			defer studio.Stop()
 
 			printApis(false)
@@ -147,7 +153,14 @@ var startCmd = &cobra.Command{
 		}
 
 		// Start server
-		go service.Start()
+		go func() {
+			err := service.Start()
+			if err != nil {
+				fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
+				os.Exit(1)
+			}
+		}()
+
 		fmt.Println(color.GreenString(L("✨LISTENING✨")))
 
 		for {
diff --git a/share/session.go b/share/session.go
index 3534ef0f..023d9fa4 100644
--- a/share/session.go
+++ b/share/session.go
@@ -66,7 +66,7 @@ func SessionFile() error {
 
 	burndb, err := session.NewBuntDB(file)
 	if err != nil {
-		return err
+		return fmt.Errorf("Session Store File %s Error: %s. Try to remove the file then restart", file, err.Error())
 	}
 
 	session.Register("file", burndb)