书籍分类统计

Author: TruthHun88

controllers/BookController.go

@@ -185,7 +185,7 @@ func (this *BookController) Setting() {
 	this.TplName = "book/setting.html"
 }
 
-//保存书籍信息
+// SaveBook 保存书籍信息
 func (this *BookController) SaveBook() {
 
 	bookResult, err := this.IsPermission()
@@ -260,7 +260,7 @@ func (this *BookController) SaveBook() {
 			beego.Error(errSearch.Error())
 		}
 	}()
-
+	go models.CountCategory()
 	this.JsonResult(0, "ok", bookResult)
 }
 
@@ -687,7 +687,7 @@ func (this *BookController) Delete() {
 			beego.Error(errDel.Error())
 		}
 	}()
-
+	go models.CountCategory()
 	this.JsonResult(0, "ok")
 }
 

controllers/ManagerController.go

@@ -516,7 +516,7 @@ func (this *ManagerController) DeleteBook() {
 			beego.Error(errDel.Error())
 		}
 	}
-
+	go models.CountCategory()
 	this.JsonResult(0, "书籍删除成功")
 }
 

models/base.go

@@ -189,27 +189,17 @@ func SitemapUpdate(domain string) {
 	Sitemap.CreateSitemapIndex(si, "sitemap.xml")
 }
 
-// 统计书籍分类
-var counting = false
-
 type Count struct {
 	Cnt        int
 	CategoryId int
 }
 
+// CountCategory 统计书籍分类
 func CountCategory() {
-	if counting {
-		return
-	}
-	counting = true
-	defer func() {
-		counting = false
-	}()
-
 	var count []Count
 
 	o := orm.NewOrm()
-	sql := "select count(bc.id) cnt, bc.category_id from md_book_category bc left join md_books b on b.book_id=bc.book_id where b.privately_owned=0 group by bc.category_id"
+	sql := "select count(bc.id) cnt, bc.category_id from md_book_category bc left join md_books b on b.book_id=bc.book_id where b.privately_owned=0 and bc.category_id>0  group by bc.category_id"
 	o.Raw(sql).QueryRows(&count)
 	if len(count) == 0 {
 		return
@@ -227,21 +217,23 @@ func CountCategory() {
 	o.Begin()
 	defer func() {
 		if err != nil {
+			beego.Error(err)
 			o.Rollback()
 		} else {
 			o.Commit()
 		}
 	}()
 
-	o.QueryTable(tableCate).Update(orm.Params{"cnt": 0})
 	cateChild := make(map[int]int)
+	if _, err = o.QueryTable(tableCate).Filter("id__gt", 0).Update(orm.Params{"cnt": 0}); err != nil {
+		return
+	}
+
 	for _, item := range count {
-		if item.Cnt > 0 {
-			cateChild[item.CategoryId] = item.Cnt
-			_, err = o.QueryTable(tableCate).Filter("id", item.CategoryId).Update(orm.Params{"cnt": item.Cnt})
-			if err != nil {
-				return
-			}
+		cateChild[item.CategoryId] = item.Cnt
+		_, err = o.QueryTable(tableCate).Filter("id", item.CategoryId).Update(orm.Params{"cnt": item.Cnt})
+		if err != nil {
+			return
 		}
 	}
 }

views/book/setting.html

@@ -127,11 +127,11 @@
                                     <span class="input-group-btn">
                                         <button class="btn btn-default" type="button">广告标题</button>
                                     </span>
-                                    <input type="text" placeholder="选填" value="{{.Model.AdTitle}}" name="ad_title" id="author" class="form-control">
+                                    <input type="text" placeholder="选填" value="{{.Model.AdTitle}}" name="ad_title" class="form-control">
                                     <span class="input-group-btn">
                                         <button class="btn btn-default" style="border-left: 0px;border-right: 0px;border-radius: 0px;" type="button">广告链接</button>
                                     </span>
-                                    <input type="text" placeholder="选填" value="{{.Model.AdLink}}" name="ad_link" id="author_url" class="form-control">
+                                    <input type="text" placeholder="选填" value="{{.Model.AdLink}}" name="ad_link" class="form-control">
                                 </div>
                                 <p class="text help-block">文字广告，可以放原作者的打赏和赞助二维码链接，也可以放开源书籍的购买链接以支持作者</p>
                             </div>