mirror of
https://github.com/mindoc-org/mindoc.git
synced 2025-09-22 20:23:35 +08:00
优化单页打印
This commit is contained in:
Minho
221
models/DocumentModel.go
Normal file
221
models/DocumentModel.go
Normal file
@@ -0,0 +1,221 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"strconv"
|
||||
|
||||
"github.com/astaxie/beego"
|
||||
"github.com/astaxie/beego/orm"
|
||||
"github.com/lifei6671/mindoc/cache"
|
||||
"github.com/lifei6671/mindoc/conf"
|
||||
)
|
||||
|
||||
// Document struct.
|
||||
type Document struct {
|
||||
DocumentId int `orm:"pk;auto;unique;column(document_id)" json:"doc_id"`
|
||||
DocumentName string `orm:"column(document_name);size(500)" json:"doc_name"`
|
||||
// Identify 文档唯一标识
|
||||
Identify string `orm:"column(identify);size(100);index;null;default(null)" json:"identify"`
|
||||
BookId int `orm:"column(book_id);type(int);index" json:"book_id"`
|
||||
ParentId int `orm:"column(parent_id);type(int);index;default(0)" json:"parent_id"`
|
||||
OrderSort int `orm:"column(order_sort);default(0);type(int);index" json:"order_sort"`
|
||||
// Markdown markdown格式文档.
|
||||
Markdown string `orm:"column(markdown);type(text);null" json:"markdown"`
|
||||
// Release 发布后的Html格式内容.
|
||||
Release string `orm:"column(release);type(text);null" json:"release"`
|
||||
// Content 未发布的 Html 格式内容.
|
||||
Content string `orm:"column(content);type(text);null" json:"content"`
|
||||
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
|
||||
MemberId int `orm:"column(member_id);type(int)" json:"member_id"`
|
||||
ModifyTime time.Time `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
|
||||
ModifyAt int `orm:"column(modify_at);type(int)" json:"-"`
|
||||
Version int64 `orm:"type(bigint);column(version)" json:"version"`
|
||||
AttachList []*Attachment `orm:"-" json:"attach"`
|
||||
}
|
||||
|
||||
// 多字段唯一键
|
||||
func (m *Document) TableUnique() [][]string {
|
||||
return [][]string{
|
||||
[]string{"book_id", "identify"},
|
||||
}
|
||||
}
|
||||
|
||||
// TableName 获取对应数据库表名.
|
||||
func (m *Document) TableName() string {
|
||||
return "documents"
|
||||
}
|
||||
|
||||
// TableEngine 获取数据使用的引擎.
|
||||
func (m *Document) TableEngine() string {
|
||||
return "INNODB"
|
||||
}
|
||||
|
||||
func (m *Document) TableNameWithPrefix() string {
|
||||
return conf.GetDatabasePrefix() + m.TableName()
|
||||
}
|
||||
|
||||
func NewDocument() *Document {
|
||||
return &Document{
|
||||
Version: time.Now().Unix(),
|
||||
}
|
||||
}
|
||||
|
||||
//根据文档ID查询指定文档.
|
||||
func (m *Document) Find(id int) (*Document, error) {
|
||||
if id <= 0 {
|
||||
return m, ErrInvalidParameter
|
||||
}
|
||||
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", id).One(m)
|
||||
|
||||
if err == orm.ErrNoRows {
|
||||
return m, ErrDataNotExist
|
||||
}
|
||||
|
||||
return m, nil
|
||||
}
|
||||
|
||||
//插入和更新文档.
|
||||
func (m *Document) InsertOrUpdate(cols ...string) error {
|
||||
o := orm.NewOrm()
|
||||
var err error
|
||||
if m.DocumentId > 0 {
|
||||
_, err = o.Update(m, cols...)
|
||||
} else {
|
||||
if m.Identify == "" {
|
||||
book := NewBook()
|
||||
identify := "docs"
|
||||
if err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id",m.BookId).One(book,"identify");err == nil {
|
||||
identify = book.Identify
|
||||
}
|
||||
|
||||
m.Identify = fmt.Sprintf("%s-%s",identify,strconv.FormatInt(time.Now().UnixNano(), 32))
|
||||
}
|
||||
if m.OrderSort == 0{
|
||||
sort,_ := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id",m.BookId).Filter("parent_id",m.ParentId).Count()
|
||||
m.OrderSort = int(sort) + 1
|
||||
}
|
||||
_, err = o.Insert(m)
|
||||
NewBook().ResetDocumentNumber(m.BookId)
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
//根据文档识别编号和项目id获取一篇文档
|
||||
func (m *Document) FindByIdentityFirst(identify string, bookId int) (*Document, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", bookId).Filter("identify", identify).One(m)
|
||||
|
||||
return m, err
|
||||
}
|
||||
|
||||
//递归删除一个文档.
|
||||
func (m *Document) RecursiveDocument(docId int) error {
|
||||
|
||||
o := orm.NewOrm()
|
||||
|
||||
if doc, err := m.Find(docId); err == nil {
|
||||
o.Delete(doc)
|
||||
NewDocumentHistory().Clear(doc.DocumentId)
|
||||
}
|
||||
var maps []orm.Params
|
||||
|
||||
_, err := o.Raw("SELECT document_id FROM " + m.TableNameWithPrefix() + " WHERE parent_id=" + strconv.Itoa(docId)).Values(&maps)
|
||||
if err != nil {
|
||||
beego.Error("RecursiveDocument => ", err)
|
||||
return err
|
||||
}
|
||||
|
||||
for _, item := range maps {
|
||||
if docId, ok := item["document_id"].(string); ok {
|
||||
id, _ := strconv.Atoi(docId)
|
||||
o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", id).Delete()
|
||||
m.RecursiveDocument(id)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
//将文档写入缓存
|
||||
func (m *Document) PutToCache() {
|
||||
go func(m Document) {
|
||||
if v, err := json.Marshal(&m); err == nil {
|
||||
if m.Identify == "" {
|
||||
|
||||
if err := cache.Put("Document.Id."+strconv.Itoa(m.DocumentId), v, time.Second*3600); err != nil {
|
||||
beego.Info("文档缓存失败:", m.DocumentId)
|
||||
}
|
||||
} else {
|
||||
if err := cache.Put(fmt.Sprintf("Document.BookId.%d.Identify.%s", m.BookId, m.Identify), v, time.Second*3600); err != nil {
|
||||
beego.Info("文档缓存失败:", m.DocumentId)
|
||||
}
|
||||
}
|
||||
}
|
||||
}(*m)
|
||||
}
|
||||
|
||||
//清除缓存
|
||||
func (m *Document) RemoveCache() {
|
||||
go func(m Document) {
|
||||
cache.Put("Document.Id."+strconv.Itoa(m.DocumentId), m, time.Second*3600)
|
||||
|
||||
if m.Identify != "" {
|
||||
cache.Put(fmt.Sprintf("Document.BookId.%d.Identify.%s", m.BookId, m.Identify), m, time.Second*3600)
|
||||
}
|
||||
}(*m)
|
||||
}
|
||||
|
||||
//从缓存获取
|
||||
func (m *Document) FromCacheById(id int) (*Document, error) {
|
||||
b := cache.Get("Document.Id." + strconv.Itoa(id))
|
||||
if v, ok := b.([]byte); ok {
|
||||
|
||||
if err := json.Unmarshal(v, m); err == nil {
|
||||
beego.Info("从缓存中获取文档信息成功", m.DocumentId)
|
||||
return m, nil
|
||||
}
|
||||
}
|
||||
defer func() {
|
||||
if m.DocumentId > 0 {
|
||||
m.PutToCache()
|
||||
}
|
||||
}()
|
||||
return m.Find(id)
|
||||
}
|
||||
|
||||
//根据文档标识从缓存中查询文档
|
||||
func (m *Document) FromCacheByIdentify(identify string, bookId int) (*Document, error) {
|
||||
b := cache.Get(fmt.Sprintf("Document.BookId.%d.Identify.%s", bookId, identify))
|
||||
if v, ok := b.([]byte); ok {
|
||||
if err := json.Unmarshal(v, m); err == nil {
|
||||
beego.Info("从缓存中获取文档信息成功", m.DocumentId, identify)
|
||||
return m, nil
|
||||
}
|
||||
}
|
||||
defer func() {
|
||||
if m.DocumentId > 0 {
|
||||
m.PutToCache()
|
||||
}
|
||||
}()
|
||||
return m.FindByIdentityFirst(identify, bookId)
|
||||
}
|
||||
|
||||
//根据项目ID查询文档列表.
|
||||
func (m *Document) FindListByBookId(bookId int) (docs []*Document, err error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", bookId).OrderBy("order_sort").All(&docs)
|
||||
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user