..

services/chart_theme.go

@@ -2,6 +2,7 @@ package services
 
 import (
 	"eta/eta_forum_hub/models"
+	"eta/eta_forum_hub/utils"
 	"fmt"
 	"time"
 )
@@ -10,51 +11,55 @@ func AddOrUpdateChartTheme(req models.ChartTheme) (err error) {
 	// 查询主题信息是否存在
 	chartTheme := new(models.ChartTheme)
 	if req.ChartThemeId > 0 {
-		chartTheme, err = models.GetChartThemeId(req.ChartThemeId)
-		if err != nil {
-			err = fmt.Errorf("查询主题信息失败 %v", err)
+		var e error
+		chartTheme, e = models.GetChartThemeId(req.ChartThemeId)
+		if e != nil && e.Error() != utils.ErrNoRow() {
+			err = fmt.Errorf("查询主题信息失败 %v", e)
 			return
 		}
-		if chartTheme.BusinessCode != chartTheme.BusinessCode {
-			err = fmt.Errorf("更新主题信息失败")
-		}
+		if e == nil {
+			if chartTheme.BusinessCode != chartTheme.BusinessCode {
+				err = fmt.Errorf("更新主题信息失败")
+			}
 
-		chartTheme.ChartThemeName = req.ChartThemeName
-		chartTheme.ChartThemeTypeId = req.ChartThemeTypeId
-		chartTheme.Config = req.Config
-		chartTheme.ChartImage = req.ChartImage
-		chartTheme.IsDelete = req.IsDelete
-		chartTheme.SysUserId = req.SysUserId
-		chartTheme.SysUserRealName = req.SysUserRealName
-		chartTheme.ModifyTime = time.Now()
+			chartTheme.ChartThemeName = req.ChartThemeName
+			chartTheme.ChartThemeTypeId = req.ChartThemeTypeId
+			chartTheme.Config = req.Config
+			chartTheme.ChartImage = req.ChartImage
+			chartTheme.IsDelete = req.IsDelete
+			chartTheme.SysUserId = req.SysUserId
+			chartTheme.SysUserRealName = req.SysUserRealName
+			chartTheme.ModifyTime = time.Now()
 
-		// 添加入库
-		err = chartTheme.Update([]string{})
-		if err != nil {
-			err = fmt.Errorf("添加或更新图表主题失败，%v", err)
-			return
-		}
-	} else {
-		chartTheme = &models.ChartTheme{
-			ChartThemeId:     req.ChartThemeId,
-			ChartThemeName:   req.ChartThemeName,
-			ChartThemeTypeId: req.ChartThemeTypeId,
-			Config:           req.Config,
-			ChartImage:       req.ChartImage,
-			IsDelete:         req.IsDelete,
-			SysUserId:        req.SysUserId,
-			SysUserRealName:  req.SysUserRealName,
-			ModifyTime:       time.Now(),
-			CreateTime:       time.Now(),
-			BusinessCode:     req.BusinessCode,
-		}
-		// 添加入库
-		err = chartTheme.Add()
-		if err != nil {
-			err = fmt.Errorf("添加或更新图表主题失败，%v", err)
+			// 添加入库
+			err = chartTheme.Update([]string{})
+			if err != nil {
+				err = fmt.Errorf("添加或更新图表主题失败，%v", err)
+				return
+			}
 			return
 		}
 	}
 
+	chartTheme = &models.ChartTheme{
+		ChartThemeId:     req.ChartThemeId,
+		ChartThemeName:   req.ChartThemeName,
+		ChartThemeTypeId: req.ChartThemeTypeId,
+		Config:           req.Config,
+		ChartImage:       req.ChartImage,
+		IsDelete:         req.IsDelete,
+		SysUserId:        req.SysUserId,
+		SysUserRealName:  req.SysUserRealName,
+		ModifyTime:       time.Now(),
+		CreateTime:       time.Now(),
+		BusinessCode:     req.BusinessCode,
+	}
+	// 添加入库
+	err = chartTheme.Add()
+	if err != nil {
+		err = fmt.Errorf("添加或更新图表主题失败，%v", err)
+		return
+	}
+
 	return
 }