变量保存功能

Author: M3chD09

.gitignore

@@ -2,4 +2,6 @@
 /Robot_Monitor_Web
 WebPageBuild
 bindata_assetfs.go
-DataAddr*
+DataAddr*
+VariablesToRead.json
+VariablesToMod.json

FileHandle/FileConvert.go

@@ -1,7 +1,6 @@
 package filehandle
 
 import (
-	"encoding/json"
 	"io/ioutil"
 	"log"
 	"os"
@@ -39,18 +38,6 @@ 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]})
 	}
-	jsonTxt, err := json.Marshal(ProjectVariables)
-	if err != nil {
-		log.Fatal(err)
-		return err
-	}
-	os.Remove("DataAddr.json")
-	f, err := os.OpenFile("DataAddr.json", os.O_WRONLY|os.O_CREATE, 0666)
-	if err != nil {
-		log.Fatal(err)
-		return err
-	}
-	defer f.Close()
-	f.Write(jsonTxt)
+	jsonSave("DataAddr.json", ProjectVariables)
 	return nil
 }

FileHandle/FileSave.go

@@ -0,0 +1,49 @@
+package filehandle
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"log"
+	"os"
+
+	datapack "www.scut-robotlab.cn/git/M3chD09/Robot_Monitor_Web/DataPack"
+)
+
+func jsonLoad(filename string, v interface{}) {
+	if _, err := os.Stat(filename); !os.IsNotExist(err) {
+		log.Println(filename, "Found")
+		data, err := ioutil.ReadFile(filename)
+		if err != nil {
+			log.Println(err)
+		}
+		err = json.Unmarshal(data, v)
+		if err != nil {
+			log.Println(err)
+		}
+	}
+}
+
+func jsonSave(filename string, v interface{}) {
+	if _, err := os.Stat(filename); !os.IsNotExist(err) {
+		os.Remove(filename)
+	}
+	jsonTxt, err := json.Marshal(v)
+	if err != nil {
+		log.Println(err)
+	}
+	err = ioutil.WriteFile(filename, jsonTxt, 0644)
+	if err != nil {
+		log.Println(err)
+	}
+}
+
+func SaveAll() {
+	jsonSave("VariablesToRead.json", datapack.CurrentVariables)
+	jsonSave("VariablesToMod.json", datapack.ModVariables)
+}
+
+func LoadSaves() {
+	jsonLoad("DataAddr.json", &ProjectVariables)
+	jsonLoad("VariablesToRead.json", &datapack.CurrentVariables)
+	jsonLoad("VariablesToMod.json", &datapack.ModVariables)
+}

main.go

@@ -2,13 +2,38 @@ package main
 
 import (
 	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
 
+	filehandle "www.scut-robotlab.cn/git/M3chD09/Robot_Monitor_Web/FileHandle"
 	serialhandle "www.scut-robotlab.cn/git/M3chD09/Robot_Monitor_Web/SerialHandle"
 	webhandle "www.scut-robotlab.cn/git/M3chD09/Robot_Monitor_Web/WebHandle"
 )
 
 func main() {
-	defer serialhandle.CloseSerialPort()
+	c := make(chan os.Signal)
+	signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
+	go func() {
+		for s := range c {
+			switch s {
+			case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT:
+				filehandle.SaveAll()
+				serialhandle.CloseSerialPort()
+				os.Exit(0)
+			default:
+				os.Exit(0)
+			}
+		}
+	}()
+	go func() {
+		for {
+			time.Sleep(5 * time.Second)
+			filehandle.SaveAll()
+		}
+	}()
+	filehandle.LoadSaves()
 	http.Handle("/", http.FileServer(http.Dir("./WebPage/")))
 	webhandle.Start()
 }