Method SetBookCates

models/book_category.go:32–71

处理书籍分类

(bookId int, cids []string)

//处理书籍分类
func (this *BookCategory) SetBookCates(bookId int, cids []string) {

	if len(cids) == 0 {
		return
	}

	var (
		cates             []Category
		tableCategory     = "md_category"
		tableBookCategory = "md_book_category"
	)

	o := orm.NewOrm()
	o.QueryTable(tableCategory).Filter("id__in", cids).All(&cates, "id", "pid")

	cidMap := make(map[string]bool)
	for _, cate := range cates {
		cidMap[strconv.Itoa(cate.Pid)] = true
		cidMap[strconv.Itoa(cate.Id)] = true
	}
	cids = []string{}
	for cid, _ := range cidMap {
		cids = append(cids, cid)
	}

	o.QueryTable(tableBookCategory).Filter("book_id", bookId).Delete()
	var bcs []BookCategory
	for _, cid := range cids {
		cidNum, _ := strconv.Atoi(cid)
		bookCate := BookCategory{
			CategoryId: cidNum,
			BookId:     bookId,
		}
		bcs = append(bcs, bookCate)
	}
	if l := len(bcs); l > 0 {
		o.InsertMulti(l, &bcs)
	}
	go CountCategory()
}