From 167e086161ab2941779cc6c1fef4acbade8f29a4 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 20 十二月 2019 10:57:10 +0800
Subject: [PATCH] bakBeforeUpgrade

---
 service/SysService.go   |   33 +++++++++++++++++++++++++++++++++
 extend/config/config.go |    1 +
 2 files changed, 34 insertions(+), 0 deletions(-)

diff --git a/extend/config/config.go b/extend/config/config.go
index 41cb8bb..48aa484 100644
--- a/extend/config/config.go
+++ b/extend/config/config.go
@@ -30,6 +30,7 @@
 	SysThresholds []threshold `mapstructure: "sysThresholds"`
 	PTZSpeed      int         `mapstructure: "ptzSpeed"` // 浜戝彴绉诲姩閫熷害
 	PatchPath       string `mapstructure:"patchPath"`//绯荤粺鏇存柊鍖呰矾寰�
+	BakPath         string `mapstructure:"bakPath"`//绯荤粺鏇存柊鍖呰矾寰�
 }
 
 var Server = &server{}
diff --git a/service/SysService.go b/service/SysService.go
index 5ff8a95..b2acfb7 100644
--- a/service/SysService.go
+++ b/service/SysService.go
@@ -152,6 +152,9 @@
 
 //upgrade
 func (sv SysService) Upgrade(identifier string,filename string) (bool,error) {
+	if !bakBeforeUpgrade() {
+		return false,errors.New("鏇存柊鍓嶅浠藉け璐�")
+	}
 	configPatchPath := ""
 	if config.Server.PatchPath != "" {
 		configPatchPath = config.Server.PatchPath
@@ -187,6 +190,30 @@
 	} else {
 		return false,errors.New("鍗囩骇鏂囦欢宸蹭涪澶憋紝璇烽噸鏂颁笂浼�")
 	}
+}
+
+func bakBeforeUpgrade() bool {
+	configBakPath := ""
+	if config.Server.BakPath != "" {
+		configBakPath = config.Server.BakPath
+	} else {
+		configBakPath = "/opt/vasystem/bak"
+	}
+	if util.Exists(configBakPath) {
+		//鍙繚鐣欐渶鏂扮殑鐗堟湰
+		if err := os.RemoveAll(configBakPath);err != nil {
+			return false
+		}
+	}
+	if !util.CreateDirectory(configBakPath) {
+		return false
+	}
+	b, err := ExecCmd("cp /opt/vasystem/bin /opt/vasystem/bak")
+	logger.Debug("bakBeforeUpgrade result:",string(b))
+	if err != nil {
+		return false
+	}
+	return true
 }
 
 //鏇存柊绯荤粺绋嬪簭
@@ -244,4 +271,10 @@
 		logger.Debug("mergeChunks result:",string(b),"cmd: ./mergeAll.sh ", chunkPath, storePath)
 		return true
 	}
+}
+
+func ExecCmd(cmdStr string) ([]byte,error) {
+	var cmd *exec.Cmd
+	cmd = exec.Command("/bin/sh", "-c",cmdStr)
+	return cmd.Output()
 }
\ No newline at end of file

--
Gitblit v1.8.0