后端配置保存功能

Author: M3chD09

.gitignore

@@ -5,3 +5,4 @@ bindata_assetfs.go
 DataAddr*
 VariablesToRead.json
 VariablesToMod.json
+Config.json

FileHandle/FileConfig.go

@@ -0,0 +1,36 @@
+package filehandle
+
+import "os"
+
+type ConfigT struct {
+	IsSaveDataAddr        bool
+	IsSaveVariablesToMod  bool
+	IsSaveVariablesToRead bool
+}
+
+var Config ConfigT
+
+func LoadConfig() {
+	if _, err := os.Stat("Config.json"); os.IsNotExist(err) {
+		Config.IsSaveDataAddr = true
+		Config.IsSaveVariablesToMod = true
+		Config.IsSaveVariablesToRead = true
+	} else {
+		jsonLoad("Config.json", &Config)
+	}
+}
+
+func SaveConfig() {
+	jsonSave("Config.json", Config)
+	if !Config.IsSaveDataAddr {
+		os.Remove("DataAddr.json")
+	} else {
+		jsonSave("DataAddr.json", ProjectVariables)
+	}
+	if !Config.IsSaveVariablesToMod {
+		os.Remove("VariablesToMod.json")
+	}
+	if !Config.IsSaveVariablesToRead {
+		os.Remove("VariablesToRead.json")
+	}
+}

FileHandle/FileConvert.go

@@ -31,6 +31,8 @@ func Txt2json() error {
 	for _, v := range match {
 		ProjectVariables.Variables = append(ProjectVariables.Variables, projectVariablesT{Addr: v[1], Size: v[2], Name: v[3], Type: v[5]})
 	}
-	jsonSave("DataAddr.json", ProjectVariables)
+	if Config.IsSaveDataAddr {
+		jsonSave("DataAddr.json", ProjectVariables)
+	}
 	return nil
 }

FileHandle/FileSave.go

@@ -38,12 +38,24 @@ func jsonSave(filename string, v interface{}) {
 }
 
 func SaveAll() {
-	jsonSave("VariablesToRead.json", datapack.CurrentVariables)
-	jsonSave("VariablesToMod.json", datapack.ModVariables)
+	SaveConfig()
+	if Config.IsSaveVariablesToMod {
+		jsonSave("VariablesToMod.json", datapack.ModVariables)
+	}
+	if Config.IsSaveVariablesToRead {
+		jsonSave("VariablesToRead.json", datapack.CurrentVariables)
+	}
 }
 
 func LoadSaves() {
-	jsonLoad("DataAddr.json", &ProjectVariables)
-	jsonLoad("VariablesToRead.json", &datapack.CurrentVariables)
-	jsonLoad("VariablesToMod.json", &datapack.ModVariables)
+	LoadConfig()
+	if Config.IsSaveDataAddr {
+		jsonLoad("DataAddr.json", &ProjectVariables)
+	}
+	if Config.IsSaveVariablesToMod {
+		jsonLoad("VariablesToMod.json", &datapack.ModVariables)
+	}
+	if Config.IsSaveVariablesToRead {
+		jsonLoad("VariablesToRead.json", &datapack.CurrentVariables)
+	}
 }

WebHandle/HandleFunc.go

@@ -187,14 +187,14 @@ func variableModListWebHandler(w http.ResponseWriter, _ *http.Request) {
 
 func fileUploadWebHandler(w http.ResponseWriter, r *http.Request) {
 	defer r.Body.Close()
+	defer os.Remove("DataAddr")
 	r.ParseMultipartForm(32 << 20)
 	file, _, err := r.FormFile("file")
 	if err != nil {
 		io.WriteString(w, "{\"status\":31}")
 		return
 	}
 	defer file.Close()
-	os.Remove("DataAddr")
 	f, err := os.OpenFile("DataAddr", os.O_WRONLY|os.O_CREATE, 0666)
 	if err != nil {
 		io.WriteString(w, "{\"status\":32}")
@@ -215,6 +215,28 @@ func fileVariablesWebHandler(w http.ResponseWriter, _ *http.Request) {
 	io.WriteString(w, string(b))
 }
 
+func fileConfigWebHandler(w http.ResponseWriter, r *http.Request) {
+	defer r.Body.Close()
+	r.ParseForm()
+	if len(r.Form) == 0 {
+		b, _ := json.Marshal(filehandle.Config)
+		io.WriteString(w, string(b))
+	} else {
+		if _, ok := r.Form["sda"]; ok {
+			filehandle.Config.IsSaveDataAddr, _ = strconv.ParseBool(strings.Join(r.Form["sda"], ""))
+		}
+		if _, ok := r.Form["svm"]; ok {
+			filehandle.Config.IsSaveVariablesToMod, _ = strconv.ParseBool(strings.Join(r.Form["svm"], ""))
+		}
+		if _, ok := r.Form["svr"]; ok {
+			filehandle.Config.IsSaveVariablesToRead, _ = strconv.ParseBool(strings.Join(r.Form["svr"], ""))
+		}
+		filehandle.SaveConfig()
+		io.WriteString(w, "{\"status\":0}")
+	}
+
+}
+
 func Start() {
 	jsonWS := make(chan string, 10)
 	go serialhandle.SerialThread(jsonWS)
@@ -233,6 +255,7 @@ func Start() {
 	http.HandleFunc("/variable/modlist", variableModListWebHandler)
 	http.HandleFunc("/file/upload", fileUploadWebHandler)
 	http.HandleFunc("/file/variables", fileVariablesWebHandler)
+	http.HandleFunc("/file/config", fileConfigWebHandler)
 	http.HandleFunc("/ws", WebSocketHandler)
 	addr := ":8080"
 	log.Println("Listen on " + addr)

main.go

@@ -13,6 +13,7 @@ import (
 )
 
 func main() {
+	filehandle.LoadSaves()
 	c := make(chan os.Signal)
 	signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
 	go func() {
@@ -33,7 +34,6 @@ func main() {
 			filehandle.SaveAll()
 		}
 	}()
-	filehandle.LoadSaves()
 	http.Handle("/", http.FileServer(http.Dir("./WebPage/")))
 	webhandle.Start()
 }