feat:update

Author: gphper

cmd/run/root.go

@@ -1,8 +1,12 @@
 package run
 
 import (
+	"bytes"
 	"context"
+	"encoding/json"
+	"fmt"
 	"grm/common"
+	"grm/global"
 	"grm/router"
 	"grm/web"
 	"io/ioutil"
@@ -11,6 +15,7 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
+	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
@@ -24,17 +29,43 @@ var CmdRun = &cobra.Command{
 }
 
 var (
-	host string
-	port string
+	host    string
+	port    string
+	confirm string
 )
 
 func init() {
-	CmdRun.Flags().StringVarP(&host, "host", "H", "0.0.0.0", "input hostname")
-	CmdRun.Flags().StringVarP(&port, "port", "p", "8088", "input port")
+	confirm = "n"
+	CmdRun.Flags().StringVarP(&host, "host", "H", global.GlobalConf.Host, "input hostname")
+	CmdRun.Flags().StringVarP(&port, "port", "p", global.GlobalConf.Port, "input port")
 }
 
 func runFunction(cmd *cobra.Command, args []string) {
 
+	fmt.Printf("%c[%d;%d;%dm%s%c[0m \n", 0x1B, 0, 40, 33, "================", 0x1B)
+	fmt.Printf("%c[%d;%d;%dm  Host:%s%c[0m \n", 0x1B, 0, 40, 33, host, 0x1B)
+	fmt.Printf("%c[%d;%d;%dm  Port:%s%c[0m \n", 0x1B, 0, 40, 33, port, 0x1B)
+	fmt.Printf("%c[%d;%d;%dm%s%c[0m \n", 0x1B, 0, 40, 33, "================", 0x1B)
+	fmt.Printf("%c[%d;%d;%dm%s%c[0m \n", 0x1B, 0, 40, 33, "Do you want to run the app at this address? Y/N", 0x1B)
+	fmt.Scan(&confirm)
+
+	if strings.ToUpper(confirm) == "N" {
+		os.Exit(0)
+	}
+
+	//保存host和port信息
+	buffer := &bytes.Buffer{}
+	encoder := json.NewEncoder(buffer)
+	global.GlobalConf.Host = host
+	global.GlobalConf.Port = port
+	err := encoder.Encode(global.GlobalConf)
+	if err != nil {
+		fmt.Println(err)
+	}
+	if err = common.WriteData(buffer.Bytes()); err != nil {
+		fmt.Println(err)
+	}
+
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = ioutil.Discard
 

global/default.go

@@ -37,6 +37,8 @@ type GlobalConfig struct {
 	RedisServicesCp map[string]RedisService
 	RedisServices   sync.Map
 	Separator       string
+	Host            string
+	Port            string
 	Tree            bool
 }
 
@@ -51,6 +53,8 @@ var GlobalConf = GlobalConfig{
 	Accounts:      make(map[string]string),
 	RedisServices: sync.Map{},
 	Separator:     ":",
+	Host:          "0.0.0.0",
+	Port:          "8088",
 	Tree:          true,
 }