后台新增视频编号

api/controller/Article.go

@@ -290,6 +290,8 @@ func (r *Article) AddArticle(c *gin.Context) {
 		}
 	}()
 
+	articleDao := dao.ArticleDao{}
+
 	article := &model.Article{
 		ArticleTitle:   req.ArticleTitle,
 		ArticleStatus:  req.ArticleStatus,
@@ -300,8 +302,18 @@ func (r *Article) AddArticle(c *gin.Context) {
 		article.VoteNum = *req.VoteNum
 	}
 
-	articleDao := dao.ArticleDao{}
-	article, err := articleDao.AddArticle(tx, article)
+	// 生成文章编号
+	maps := make(map[string]interface{})
+	total, err := articleDao.GetArticleCount(maps)
+	if err != nil {
+		tx.Rollback()
+		responses.FailWithMessage(err.Error(), c)
+		return
+	}
+
+	article.ArticleNumber = fmt.Sprintf("%d", 1000+total)
+
+	article, err = articleDao.AddArticle(tx, article)
 	if err != nil {
 		tx.Rollback()
 		responses.FailWithMessage(err.Error(), c)

api/controller/Video.go

@@ -294,6 +294,8 @@ func (r *Video) AddVideo(c *gin.Context) {
 		}
 	}()
 
+	videoDao := dao.VideoDao{}
+
 	video := &model.Video{
 		VideoTitle:   req.VideoTitle,
 		VideoStatus:  req.VideoStatus,
@@ -305,8 +307,18 @@ func (r *Video) AddVideo(c *gin.Context) {
 		video.VoteNum = *req.VoteNum
 	}
 
-	videoDao := dao.VideoDao{}
-	video, err := videoDao.AddVideo(tx, video)
+	// 生成视频编号
+	maps := make(map[string]interface{})
+	total, err := videoDao.GetVideoCount(maps)
+	if err != nil {
+		tx.Rollback()
+		responses.FailWithMessage(err.Error(), c)
+		return
+	}
+
+	video.VideoNumber = fmt.Sprintf("%d", 1000+total)
+
+	video, err = videoDao.AddVideo(tx, video)
 	if err != nil {
 		tx.Rollback()
 		responses.FailWithMessage(err.Error(), c)