feat: 许可证和网元开关机

2023-10-12 09:49:34 +08:00
parent 3220f9d138
commit 55fe1d534d
5 changed files with 211 additions and 24 deletions
--- a/features/cm/license.go
+++ b/features/cm/license.go
@@ -1,8 +1,14 @@
 package cm

 import (
+	"fmt"
+	"io"
 	"net/http"
+	"os"
+	"os/exec"
+	"strings"

+	"ems.agt/lib/dborm"
 	"ems.agt/lib/log"
 	"ems.agt/lib/services"
 	"ems.agt/restagent/config"
@@ -11,12 +17,12 @@ import (
 )

 var (
-	// License
-	LicenseUri   = config.DefaultUriPrefix + "/systemManagement/{apiVersion}/{neType}/license"
-	NeLicenseUri = config.DefaultUriPrefix + "/systemManagement/{apiVersion}/{neType}/license/{neId}"
+	// General License URI
+	UriLicense    = config.DefaultUriPrefix + "/systemManagement/{apiVersion}/{elementTypeValue}/license"
+	UriLicenseExt = config.DefaultUriPrefix + "/systemManagement/{apiVersion}/elementType/{elementTypeValue}/objectType/license"

-	CustomLicenseUri   = config.UriPrefix + "/systemManagement/{apiVersion}/{neType}/license"
-	CustomNeLicenseUri = config.UriPrefix + "/systemManagement/{apiVersion}/{neType}/license/{neId}"
+	CustomUriLicense    = config.UriPrefix + "/systemManagement/{apiVersion}/{elementTypeValue}/license"
+	CustomUriLicenseExt = config.UriPrefix + "/systemManagement/{apiVersion}/elementType/{elementTypeValue}/objectType/license"
 )

 func UploadLicenseFile(w http.ResponseWriter, r *http.Request) {
@@ -39,7 +45,6 @@ func UploadLicenseFile(w http.ResponseWriter, r *http.Request) {
 	// neTypeLower := strings.ToLower(neType)

 	services.ResponseStatusOK204NoContent(w)
-	return
 }

 func DownloadLicenseFile(w http.ResponseWriter, r *http.Request) {
@@ -86,7 +91,6 @@ func DownloadLicenseFile(w http.ResponseWriter, r *http.Request) {
 	// md5Sum := (*neSoftware)[0]["md5_sum"]

 	// services.ResponseFileWithNameAndMD5(w, http.StatusOK, fileName, path, md5Sum)
-	return
 }

 func DeleteLcenseFile(w http.ResponseWriter, r *http.Request) {
@@ -147,5 +151,183 @@ func DeleteLcenseFile(w http.ResponseWriter, r *http.Request) {
 	// }

 	services.ResponseStatusOK204NoContent(w)
-	return
+}
+
+func UploadLicenseFileData(w http.ResponseWriter, r *http.Request) {
+	log.Info("UploadLicenseFileData processing... ")
+
+	_, err := services.CheckFrontValidRequest(w, r)
+	if err != nil {
+		log.Error("Http request error:", err)
+		return
+	}
+
+	vars := mux.Vars(r)
+	neType := vars["elementTypeValue"]
+	if neType == "" {
+		log.Error("elementTypeValue is empty")
+		services.ResponseNotFound404UriNotExist(w, r)
+		return
+	}
+	neTypeUpper := strings.ToUpper(neType)
+	neTypeLower := strings.ToLower(neType)
+
+	//md5Param := services.GetUriParamString(r, "md5Sum", ",", false, false)
+
+	neId := services.GetUriParamString(r, "neId", ",", false, false)
+
+	neInfo, err := dborm.XormGetNeInfo(neType, neId)
+	if err != nil {
+		log.Errorf("Failed to get ne_info:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+	log.Debug("neInfo:", neInfo)
+
+	licensePath := fmt.Sprintf("%s/%s", config.GetYamlConfig().OMC.License, neTypeLower)
+	err = os.MkdirAll(licensePath, os.ModePerm)
+	if err != nil {
+		log.Error("Failed to Mkdir:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+	//fileName, err := services.HandleUploadFile(r, softwarePath, "")
+
+	// 解析multipart/form-data请求
+	err = r.ParseMultipartForm(10 << 20) // 10MB
+	if err != nil {
+		log.Error("Faile to ParseMultipartForm:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+
+	// 获取文件和数据
+	licFile := r.MultipartForm.File["file"]
+	data := r.MultipartForm.Value["comment"]
+
+	var licenseFileName, comment string
+
+	// 处理软件rpm/deb文件
+	if len(licFile) > 0 {
+		file := licFile[0]
+		// 打开文件
+		f, err := file.Open()
+		if err != nil {
+			log.Error("Faile to Open:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+		defer f.Close()
+
+		// 创建本地文件
+		dst, err := os.Create(licensePath + "/" + file.Filename)
+		if err != nil {
+			log.Error("Faile to Create:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+		defer dst.Close()
+
+		licenseFileName = file.Filename
+		// 将文件内容拷贝到本地文件
+		_, err = io.Copy(dst, f)
+		if err != nil {
+			log.Error("Faile to Copy:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+	}
+
+	// 处理数据
+	if len(data) > 0 {
+		comment = data[0]
+	}
+
+	neLicensePath := strings.ReplaceAll(config.GetYamlConfig().NE.LicenseDir, "{neType}", neTypeLower)
+
+	srcFile := fmt.Sprintf("%s/%s", licensePath, licenseFileName)
+	scpDir := fmt.Sprintf("%s@%s:%s", config.GetYamlConfig().NE.User, neInfo.Ip, config.GetYamlConfig().NE.ScpDir)
+	cmd := exec.Command("scp", "-r", srcFile, scpDir)
+	out, err := cmd.CombinedOutput()
+	log.Debugf("Exec output: %v", string(out))
+	if err != nil {
+		log.Errorf("Faile to scp NF: neType=%s, neId=%s, ip=%s", neType, neId, neInfo.Ip)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+	// backup system.ini to system.ini.bak
+	sshHost := fmt.Sprintf("%s@%s", config.GetYamlConfig().NE.User, neInfo.Ip)
+	cpCmd := fmt.Sprintf("sudo cp -f %s/system.ini %s/system.ini.bak", neLicensePath, neLicensePath)
+	cmd = exec.Command("ssh", sshHost, cpCmd)
+	out, err = cmd.CombinedOutput()
+	log.Debugf("Exec output: %v", string(out))
+	if err != nil {
+		log.Error("Faile to execute cp command:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+
+	// replace system.ini
+	neFilePath := config.GetYamlConfig().NE.ScpDir + "/" + licenseFileName
+	cpCmd = fmt.Sprintf("sudo mv -f %s %s/system.ini", neFilePath, neLicensePath)
+	cmd = exec.Command("ssh", sshHost, cpCmd)
+	out, err = cmd.CombinedOutput()
+	log.Debugf("Exec output: %v", string(out))
+	if err != nil {
+		log.Error("Faile to execute cp command:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+
+	switch neTypeLower {
+	case "omc":
+		restartCmd := fmt.Sprintf("sudo %s/bin/omcsvc.sh restart", config.GetYamlConfig().NE.OmcDir)
+		cmd := exec.Command("ssh", sshHost, restartCmd)
+		out, err := cmd.CombinedOutput()
+		log.Debugf("Exec output: %v", string(out))
+		if err != nil {
+			log.Error("Faile to execute ssh restart omc:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+	case "ims":
+		restartCmd := "sudo ims-stop && sudo ims-start"
+		cmd := exec.Command("ssh", sshHost, restartCmd)
+		out, err := cmd.CombinedOutput()
+		log.Debugf("Exec output: %v", string(out))
+		if err != nil {
+			log.Error("Faile to execute ssh sudo systemctl command:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+	default:
+		restartCmd := fmt.Sprintf("sudo systemctl restart %s.service", neTypeLower)
+		cmd := exec.Command("ssh", sshHost, restartCmd)
+		out, err := cmd.CombinedOutput()
+		log.Debugf("Exec output: %v", string(out))
+		if err != nil {
+			log.Error("Faile to execute ssh sudo systemctl command:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+	}
+
+	neLicense := dborm.NeLicense{
+		NeType:   neTypeUpper,
+		NeID:     neId,
+		Status:   "ACTIVE",
+		Path:     licensePath,
+		FileName: licenseFileName,
+		Comment:  comment,
+	}
+
+	log.Debug("neLicense:", neLicense)
+	_, err = dborm.XormInsertTableOne("ne_license", neLicense)
+	if err != nil {
+		log.Error("Faile to XormInsertTableOne:", err)
+		services.ResponseInternalServerError500ProcessError(w, err)
+		return
+	}
+
+	services.ResponseStatusOK204NoContent(w)
 }
--- a/features/cm/ne.go
+++ b/features/cm/ne.go
@@ -766,7 +766,7 @@ func PostNeServiceAction(w http.ResponseWriter, r *http.Request) {
 	neTypeLower := strings.ToLower(neType)
 	action := vars["action"]

-	neId := services.GetUriParamString(r, "ne_id", ",", false, false)
+	neId := services.GetUriParamString(r, "neId", ",", false, false)

 	//	neInfo := new(dborm.NeInfo)
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
@@ -852,7 +852,7 @@ func PostNeInstanceAction(w http.ResponseWriter, r *http.Request) {
 	//neTypeLower := strings.ToLower(neType)
 	action := vars["action"]

-	neId := services.GetUriParamString(r, "ne_id", ",", false, false)
+	neId := services.GetUriParamString(r, "neId", ",", false, false)

 	//	neInfo := new(dborm.NeInfo)
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
--- a/features/cm/software.go
+++ b/features/cm/software.go
@@ -564,7 +564,7 @@ func DistributeSoftwareToNF(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	neFilePath := config.GetYamlConfig().NE.ScpDir + "/" + fileName
-	cpCmd := fmt.Sprintf("sudo cp -f %s %s/software/%s", neFilePath,
+	cpCmd := fmt.Sprintf("sudo mv -f %s %s/software/%s", neFilePath,
 		config.GetYamlConfig().NE.OmcDir, neTypeLower)
 	cmd = exec.Command("ssh", sshHost, cpCmd)
 	out, err = cmd.CombinedOutput()