将需要生成电子书的书籍加入到队列

Author: TruthHun88

commands/command.go

@@ -99,6 +99,7 @@ func RegisterModel() {
 		models.NewSign(),
 		models.NewBookCounter(),
 		models.NewDownloadCounter(),
+		models.NewEbook(),
 	)
 	migrate.RegisterMigration()
 }

controllers/BookController.go

@@ -729,8 +729,7 @@ func (this *BookController) Release() {
 	this.JsonResult(0, "发布任务已推送到任务队列，稍后将在后台执行。")
 }
 
-//生成下载文档
-//加锁，防止用户不停地点击生成下载文档造成服务器资源开销.
+// 生成电子书
 func (this *BookController) Generate() {
 	identify := this.GetString(":key")
 
@@ -743,15 +742,20 @@ func (this *BookController) Generate() {
 		beego.Error(err)
 		this.JsonResult(1, "书籍不存在")
 	}
-	//书籍正在生成离线文档
-	if isGenerating := utils.BooksGenerate.Exist(book.BookId); isGenerating {
-		this.JsonResult(1, "上一次下载文档生成任务正在后台执行，请您稍后再执行新的下载文档生成操作")
+
+	ebookModel := models.NewEbook()
+
+	// 电子书不是处于完成状态，不允许再添加到电子书生成队列中
+	if ok := ebookModel.IsFinish(book.BookId); !ok {
+		this.JsonResult(1, "电子书生成任务已在处理中，如需再次生成，请您稍后再试。")
 	}
 
-	baseUrl := "http://localhost:" + beego.AppConfig.String("httpport")
-	go new(models.Document).GenerateBook(book, baseUrl)
+	// 添加到电子书生成队列
+	if err = ebookModel.AddToGenerate(book.BookId); err != nil {
+		this.JsonResult(1, err.Error())
+	}
 
-	this.JsonResult(0, "下载文档生成任务已交由后台执行，请您耐心等待。")
+	this.JsonResult(0, "电子书生成任务已交由后台执行，请您耐心等待。")
 }
 
 //文档排序.

models/ebook.go

@@ -0,0 +1,136 @@
+package models
+
+import (
+	"errors"
+	"time"
+
+	"github.com/astaxie/beego"
+	"github.com/astaxie/beego/orm"
+)
+
+type Ebook struct {
+	Id            int
+	Title         string    // 电子书名称
+	Keywords      string    // 关键字
+	Description   string    // 摘要
+	Path          string    // 文件路径。如果是网站生成的电子书，则为电子书的路径，否则为URL地址
+	BookID        int       `orm:"default(0);column(book_id);index"` // 所属书籍ID
+	Ext           string    `orm:"size(8);index"`                    // 文件扩展名
+	Status        int       `orm:"default(0);index"`                 // 0：待处理； 1: 转换中；2: 转换完成
+	Size          int64     `orm:"default(0)"`                       // 电子书大小
+	DownloadCount int       `orm:"default(0)"`                       // 电子书被下载次数
+	CreatedAt     time.Time `orm:"auto_now_add;type(datetime)"`
+	UpdatedAt     time.Time `orm:"auto_now;type(datetime)"`
+}
+
+var convert2ebookRunning = false
+
+const (
+	EBookStatusPending     = 0 // 待处理
+	EBookStatusProccessing = 1 // 处理中
+	EBookStatusSuccess     = 2 // 转换成功
+	EBookStatusFailure     = 3 // 失败
+)
+
+func NewEbook() *Ebook {
+	return &Ebook{}
+}
+
+func (m *Ebook) GetEBookByBookID(bookID int) (books []Ebook) {
+	if bookID <= 0 {
+		return
+	}
+
+	if _, err := orm.NewOrm().QueryTable(m).Filter("book_id", bookID).All(&books); err != nil && err != orm.ErrNoRows {
+		beego.Error(err)
+	}
+	return
+}
+
+func (m *Ebook) GetEBook(id int) (book Ebook) {
+	if id <= 0 {
+		return
+	}
+	err := orm.NewOrm().QueryTable(m).Filter("id", id).One(&book)
+	if err != nil {
+		beego.Error(err)
+	}
+	return
+}
+
+// 添加书籍到电子书生成队列
+func (m *Ebook) AddToGenerate(bookID int) (err error) {
+	var (
+		ebooks []Ebook
+		exts   = []string{".pdf", ".mobi", ".epub"}
+	)
+
+	b, _ := NewBook().Find(bookID)
+	if b == nil || b.BookId == 0 {
+		return errors.New("书籍不存在")
+	}
+	for _, ext := range exts {
+		ebooks = append(ebooks, Ebook{
+			Title:       b.BookName,
+			Keywords:    b.Label,
+			Description: b.Description,
+			BookID:      bookID,
+			Ext:         ext,
+			Status:      EBookStatusPending,
+		})
+	}
+
+	if _, err = orm.NewOrm().InsertMulti(len(ebooks), &ebooks); err != nil {
+		beego.Error(err)
+	}
+	return
+}
+
+// 电子书状态（最新的状态）
+func (m *Ebook) Stats(bookID int) (stats map[string]Ebook) {
+	var (
+		ebooks []Ebook
+		limit  = 4 // 先默认为4，即四个扩展名：.pdf,.epub,.mobi,.docx
+	)
+	stats = make(map[string]Ebook)
+	o := orm.NewOrm()
+	o.QueryTable(m).Filter("book_id", bookID).OrderBy("-id").Limit(limit).All(&ebooks)
+	if len(ebooks) == 0 {
+		return
+	}
+
+	for _, ebook := range ebooks {
+		if _, ok := stats[ebook.Ext]; !ok {
+			stats[ebook.Ext] = ebook
+		}
+	}
+	return
+}
+
+// 查询书籍是否处于完成状态。失败也是完成状态的一种。
+func (m *Ebook) IsFinish(bookID int) (ok bool) {
+	count, err := orm.NewOrm().QueryTable(m).Filter("book_id", bookID).Filter("status__in", EBookStatusPending, EBookStatusProccessing).Count()
+	if err != nil {
+		beego.Error(err)
+		return
+	}
+	return count == 0
+}
+
+// 生成电子书
+func (m *Ebook) convert2ebook() {
+	if convert2ebookRunning {
+		return
+	}
+	convert2ebookRunning = true
+	o := orm.NewOrm()
+	o.QueryTable(m).Filter("book_id__gt", 0).Filter("status", EBookStatusProccessing).Update(orm.Params{"status": EBookStatusPending})
+	for {
+		var ebook Ebook
+		o.QueryTable(m).Filter("book_id__gt", 0).Filter("status", EBookStatusPending).OrderBy("id").One(&ebook)
+		if ebook.Id > 0 {
+			// 根据电子书的ID，查找现有的电子书的队列
+		}
+		time.Sleep(5 * time.Second)
+	}
+}