marge: 合并代码

features/cm/ne.go

@@ -43,8 +43,28 @@ var (
 	CustomUriNeInstance = config.UriPrefix + "/systemManagement/{apiVersion}/elementType/{elementTypeValue}/objectType/instance/{action}"
 )
 
-func init() {
+const (
+	NEStatusActive   = 0
+	NEStatusOffline  = 1
+	NEStatusStandby  = 2
+	NEStatusMaintain = 3
+)
 
+var client = resty.New()
+
+func init() {
+	/*
+		client.
+			SetTimeout(10 * time.Second).
+			SetRetryCount(1).
+			SetRetryWaitTime(1 * time.Second).
+			SetRetryMaxWaitTime(2 * time.Second).
+			SetRetryAfter(func(client *resty.Client, resp *resty.Response) (time.Duration, error) {
+				return 0, errors.New("quota exceeded")
+			})
+	*/
+	client.
+		SetTimeout(time.Duration(1 * time.Second))
 }
 
 func GetNeInfo(w http.ResponseWriter, r *http.Request) {
@@ -69,7 +89,7 @@ func GetNeInfo(w http.ResponseWriter, r *http.Request) {
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
 	if err != nil {
 		log.Error("dborm.XormGetNeInfo is failed:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 
@@ -105,9 +125,10 @@ func PostNeInfo(w http.ResponseWriter, r *http.Request) {
 		services.ResponseNotFound404UriNotExist(w, r)
 		return
 	}
+	syncFlag := services.GetUriParamString(r, "sync2ne", ",", false, false)
 	body, err := io.ReadAll(io.LimitReader(r.Body, global.RequestBodyMaxLen))
 	if err != nil {
-		log.Error("io.ReadAll is failed:", err)
+		log.Error("Failed to o.ReadAll:", err)
 		services.ResponseNotFound404UriNotExist(w, r)
 		return
 	}
@@ -123,41 +144,13 @@ func PostNeInfo(w http.ResponseWriter, r *http.Request) {
 	neInfo.UpdateTime = time.Now().Format(time.DateTime)
 	log.Debug("NE info:", neInfo)
 
-	hostUri := global.CombineHostUri(neInfo.Ip, neInfo.Port)
-	//hostUri := fmt.Sprintf("http://%s:%v", neInfo.Ip, neInfo.Port)
-	apiUri := fmt.Sprintf(UriParamOmcNeConfig, strings.ToLower(neInfo.NeType))
-	requestURI2NF := fmt.Sprintf("%s%s", hostUri, apiUri)
-	log.Debug("requestURI2NF:", requestURI2NF)
-
-	omcNeConfig := &OmcNeConfig{
-		NeId:       neInfo.NeId,
-		RmUID:      neInfo.RmUID,
-		NeName:     neInfo.NeName,
-		PvFlag:     neInfo.PvFlag,
-		Province:   neInfo.Province,
-		VendorName: neInfo.VendorName,
-		Dn:         neInfo.Dn,
-	}
-	body, _ = json.Marshal(omcNeConfig)
-	client := resty.New()
-	response, err := client.R().
-		EnableTrace().
-		SetHeaders(map[string]string{"User-Agent": config.GetDefaultUserAgent()}).
-		SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}).
-		SetBody(body).
-		Put(requestURI2NF)
-	if err != nil {
-		log.Error("Failed to Put:", err)
-		services.ResponseInternalServerError500NFConnectRefused(w)
-		return
-	}
-	log.Info("StatusCode: ", response.StatusCode())
-
-	if !config.GetYamlConfig().OMC.Chk2Ne {
+	//if !config.GetYamlConfig().OMC.Chk2Ne {
+	if syncFlag == "false" {
+		neInfo.Status = NEStatusMaintain
 		affected, err := dborm.XormInsertNeInfo(neInfo)
 		if err != nil {
 			log.Error("Failed to insert Ne info:", err)
-			services.ResponseInternalServerError500DatabaseOperationFailed(w)
+			services.ResponseInternalServerError500ProcessError(w, err)
 			return
 		}
 
@@ -167,13 +160,43 @@ func PostNeInfo(w http.ResponseWriter, r *http.Request) {
 		services.ResponseWithJson(w, http.StatusOK, mapRow)
 		return
 	} else {
+		hostUri := global.CombineHostUri(neInfo.Ip, neInfo.Port)
+		//hostUri := fmt.Sprintf("http://%s:%v", neInfo.Ip, neInfo.Port)
+		apiUri := fmt.Sprintf(UriParamOmcNeConfig, strings.ToLower(neInfo.NeType))
+		requestURI2NF := fmt.Sprintf("%s%s", hostUri, apiUri)
+		log.Debug("requestURI2NF:", requestURI2NF)
+
+		omcNeConfig := &OmcNeConfig{
+			NeId:       neInfo.NeId,
+			RmUID:      neInfo.RmUID,
+			NeName:     neInfo.NeName,
+			PvFlag:     neInfo.PvFlag,
+			Province:   neInfo.Province,
+			VendorName: neInfo.VendorName,
+			Dn:         neInfo.Dn,
+		}
+		body, _ = json.Marshal(omcNeConfig)
+		response, err := client.R().
+			EnableTrace().
+			SetHeaders(map[string]string{"User-Agent": config.GetDefaultUserAgent()}).
+			SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}).
+			SetBody(body).
+			Put(requestURI2NF)
+		if err != nil {
+			log.Error("Failed to Put:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+		log.Info("StatusCode: ", response.StatusCode())
+
 		respMsg := make(map[string]interface{})
 		switch response.StatusCode() {
 		case http.StatusOK, http.StatusCreated, http.StatusNoContent, http.StatusAccepted:
+			neInfo.Status = NEStatusActive
 			affected, err := dborm.XormInsertNeInfo(neInfo)
 			if err != nil {
 				log.Error("Failed to dborm.XormInsertNeInfo:", err)
-				services.ResponseInternalServerError500DatabaseOperationFailed(w)
+				services.ResponseInternalServerError500ProcessError(w, err)
 				return
 			} else if affected <= 0 {
 				log.Infof("Not record affected to insert ne_info")
@@ -208,9 +231,10 @@ func PutNeInfo(w http.ResponseWriter, r *http.Request) {
 		services.ResponseNotFound404UriNotExist(w, r)
 		return
 	}
+	syncFlag := services.GetUriParamString(r, "sync2ne", ",", false, false)
 	body, err := io.ReadAll(io.LimitReader(r.Body, global.RequestBodyMaxLen))
 	if err != nil {
-		log.Error("io.ReadAll is failed:", err)
+		log.Error("Failed to io.ReadAll:", err)
 		services.ResponseNotFound404UriNotExist(w, r)
 		return
 	}
@@ -221,41 +245,13 @@ func PutNeInfo(w http.ResponseWriter, r *http.Request) {
 	neInfo.UpdateTime = time.Now().Format(time.DateTime)
 	log.Debug("NE info:", neInfo)
 
-	hostUri := global.CombineHostUri(neInfo.Ip, neInfo.Port)
-	//hostUri := fmt.Sprintf("http://%s:%v", neInfo.Ip, neInfo.Port)
-	apiUri := fmt.Sprintf(UriParamOmcNeConfig, strings.ToLower(neType))
-	requestURI2NF := fmt.Sprintf("%s%s", hostUri, apiUri)
-	log.Debug("requestURI2NF:", requestURI2NF)
-
-	omcNeConfig := &OmcNeConfig{
-		NeId:       neInfo.NeId,
-		RmUID:      neInfo.RmUID,
-		NeName:     neInfo.NeName,
-		PvFlag:     neInfo.PvFlag,
-		Province:   neInfo.Province,
-		VendorName: neInfo.VendorName,
-		Dn:         neInfo.Dn,
-	}
-	body, _ = json.Marshal(omcNeConfig)
-	client := resty.New()
-	response, err := client.R().
-		EnableTrace().
-		SetHeaders(map[string]string{"User-Agent": config.GetDefaultUserAgent()}).
-		SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}).
-		SetBody(body).
-		Put(requestURI2NF)
-	if err != nil {
-		log.Error("Failed to Put:", err)
-		services.ResponseInternalServerError500NFConnectRefused(w)
-		return
-	}
-	log.Info("StatusCode: ", response.StatusCode())
-
-	if !config.GetYamlConfig().OMC.Chk2Ne {
+	//if !config.GetYamlConfig().OMC.Chk2Ne {
+	if syncFlag == "false" {
+		neInfo.Status = NEStatusMaintain
 		affected, err := dborm.XormUpdateNeInfo(neInfo)
 		if err != nil {
 			log.Error("Failed to update Ne info:", err)
-			services.ResponseInternalServerError500DatabaseOperationFailed(w)
+			services.ResponseInternalServerError500ProcessError(w, err)
 			return
 		}
 
@@ -265,13 +261,43 @@ func PutNeInfo(w http.ResponseWriter, r *http.Request) {
 		services.ResponseWithJson(w, http.StatusOK, mapRow)
 		return
 	} else {
+		hostUri := global.CombineHostUri(neInfo.Ip, neInfo.Port)
+		//hostUri := fmt.Sprintf("http://%s:%v", neInfo.Ip, neInfo.Port)
+		apiUri := fmt.Sprintf(UriParamOmcNeConfig, strings.ToLower(neType))
+		requestURI2NF := fmt.Sprintf("%s%s", hostUri, apiUri)
+		log.Debug("requestURI2NF:", requestURI2NF)
+
+		omcNeConfig := &OmcNeConfig{
+			NeId:       neInfo.NeId,
+			RmUID:      neInfo.RmUID,
+			NeName:     neInfo.NeName,
+			PvFlag:     neInfo.PvFlag,
+			Province:   neInfo.Province,
+			VendorName: neInfo.VendorName,
+			Dn:         neInfo.Dn,
+		}
+		body, _ = json.Marshal(omcNeConfig)
+		response, err := client.R().
+			EnableTrace().
+			SetHeaders(map[string]string{"User-Agent": config.GetDefaultUserAgent()}).
+			SetHeaders(map[string]string{"Content-Type": "application/json;charset=UTF-8"}).
+			SetBody(body).
+			Put(requestURI2NF)
+		if err != nil {
+			log.Error("Failed to Put:", err)
+			services.ResponseInternalServerError500ProcessError(w, err)
+			return
+		}
+		log.Info("StatusCode: ", response.StatusCode())
+
 		respMsg := make(map[string]interface{})
 		switch response.StatusCode() {
 		case http.StatusOK, http.StatusCreated, http.StatusNoContent, http.StatusAccepted:
+			neInfo.Status = NEStatusActive
 			affected, err := dborm.XormUpdateNeInfo(neInfo)
 			if err != nil {
 				log.Error("Failed to dborm.XormUpdateNeInfo:", err)
-				services.ResponseInternalServerError500DatabaseOperationFailed(w)
+				services.ResponseInternalServerError500ProcessError(w, err)
 				return
 			} else if affected <= 0 {
 				log.Infof("Not record affected to insert ne_info")
@@ -321,7 +347,7 @@ func DeleteNeInfo(w http.ResponseWriter, r *http.Request) {
 	neInfo, err = dborm.XormGetNeInfo(neInfo.NeType, neInfo.NeId)
 	if err != nil || neInfo == nil {
 		log.Error("Failed to delete Ne info:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 	log.Debug("NE info:", neInfo)
@@ -331,7 +357,7 @@ func DeleteNeInfo(w http.ResponseWriter, r *http.Request) {
 		affected, err := dborm.XormDeleteNeInfo(neInfo)
 		if err != nil {
 			log.Error("Failed to delete Ne info:", err)
-			services.ResponseInternalServerError500DatabaseOperationFailed(w)
+			services.ResponseInternalServerError500ProcessError(w, err)
 			return
 		}
 
@@ -353,7 +379,6 @@ func IsActiveNF(neInfo *dborm.NeInfo) bool {
 	requestURI := fmt.Sprintf(config.UriPrefix+"/systemManagement/v1/elementType/%s/objectType/systemState",
 		strings.ToLower(neInfo.NeType))
 
-	client := resty.New()
 	response, err := client.R().
 		EnableTrace().
 		SetHeaders(map[string]string{"User-Agent": config.GetDefaultUserAgent()}).
@@ -361,6 +386,7 @@ func IsActiveNF(neInfo *dborm.NeInfo) bool {
 		Get(hostUri + requestURI)
 	if err != nil {
 		log.Error("Failed to Get:", err)
+		return false
 	}
 
 	switch response.StatusCode() {
@@ -395,7 +421,7 @@ func ExportCmFromNF(w http.ResponseWriter, r *http.Request) {
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
 	if err != nil {
 		log.Errorf("Failed to get ne_info:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 	log.Debug("neInfo:", neInfo)
@@ -561,7 +587,7 @@ func ImportCmToNF(w http.ResponseWriter, r *http.Request) {
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
 	if err != nil {
 		log.Errorf("Failed to get ne_info:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 	log.Debug("neInfo:", neInfo)
@@ -802,7 +828,7 @@ func PostNeServiceAction(w http.ResponseWriter, r *http.Request) {
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
 	if err != nil {
 		log.Error("Failed to get ne_info:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 	log.Debug("neInfo:", neInfo)
@@ -888,7 +914,7 @@ func PostNeInstanceAction(w http.ResponseWriter, r *http.Request) {
 	neInfo, err := dborm.XormGetNeInfo(neType, neId)
 	if err != nil {
 		log.Errorf("Failed to get ne_info:", err)
-		services.ResponseInternalServerError500DatabaseOperationFailed(w)
+		services.ResponseInternalServerError500ProcessError(w, err)
 		return
 	}
 	log.Debug("neInfo:", neInfo)