mindoc/models/AttachmentModel.go

// 数据库模型.
package models

import (
	"time"

	"os"

	"strings"

	"github.com/beego/beego/v2/client/orm"
	"github.com/beego/beego/v2/core/logs"
	"github.com/mindoc-org/mindoc/conf"
	"github.com/mindoc-org/mindoc/utils/filetil"
	// "gorm.io/driver/sqlite"
	// "gorm.io/gorm"
	// "gorm.io/gorm/logger"
	// "gorm.io/gorm/schema"
)

// 定义全局的db对象，我们执行数据库操作主要通过他实现。
// var _db *gorm.DB

// Attachment struct .
type Attachment struct {
	AttachmentId int       `orm:"column(attachment_id);pk;auto;unique" json:"attachment_id"`
	BookId       int       `orm:"column(book_id);type(int);description(所属book id)" json:"book_id"`
	DocumentId   int       `orm:"column(document_id);type(int);null;description(所属文档id)" json:"doc_id"`
	FileName     string    `orm:"column(file_name);size(255);description(文件名称)" json:"file_name"`
	FilePath     string    `orm:"column(file_path);size(2000);description(文件路径)" json:"file_path"`
	FileSize     float64   `orm:"column(file_size);type(float);description(文件大小 字节)" json:"file_size"`
	HttpPath     string    `orm:"column(http_path);size(2000);description(文件路径)" json:"http_path"`
	FileExt      string    `orm:"column(file_ext);size(50);description(文件后缀)" json:"file_ext"`
	CreateTime   time.Time `orm:"type(datetime);column(create_time);auto_now_add;description(创建时间)" json:"create_time"`
	CreateAt     int       `orm:"column(create_at);type(int);description(创建人id)" json:"create_at"`
	ResourceType string    `orm:"-" json:"resource_type"`
}

// TableName 获取对应上传附件数据库表名.
func (m *Attachment) TableName() string {
	return "attachment"
}

// TableEngine 获取数据使用的引擎.
func (m *Attachment) TableEngine() string {
	return "INNODB"
}
func (m *Attachment) TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() + m.TableName()
}

func NewAttachment() *Attachment {
	return &Attachment{}
}

func (m *Attachment) Insert() error {
	o := orm.NewOrm()

	_, err := o.Insert(m)

	return err
}
func (m *Attachment) Update() error {
	o := orm.NewOrm()
	_, err := o.Update(m)
	return err
}

func (m *Attachment) Delete() error {
	o := orm.NewOrm()

	_, err := o.Delete(m)

	if err == nil {
		if err1 := os.Remove(m.FilePath); err1 != nil {
			logs.Error(err1)
		}
	}

	return err
}

func (m *Attachment) Find(id int) (*Attachment, error) {
	if id <= 0 {
		return m, ErrInvalidParameter
	}
	o := orm.NewOrm()

	err := o.QueryTable(m.TableNameWithPrefix()).Filter("attachment_id", id).One(m)

	return m, err
}

// 查询指定文档的附件列表
func (m *Attachment) FindListByDocumentId(docId int) (attaches []*Attachment, err error) {
	o := orm.NewOrm()

	_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("document_id", docId).Filter("book_id__gt", 0).OrderBy("-attachment_id").All(&attaches)
	return
}

// 分页查询附件
func (m *Attachment) FindToPager(pageIndex, pageSize int) (attachList []*AttachmentResult, totalCount int, err error) {
	o := orm.NewOrm()

	total, err := o.QueryTable(m.TableNameWithPrefix()).Count()

	if err != nil {

		return nil, 0, err
	}
	totalCount = int(total)

	var list []*Attachment

	offset := (pageIndex - 1) * pageSize
	if pageSize == 0 {
		_, err = o.QueryTable(m.TableNameWithPrefix()).OrderBy("-attachment_id").Offset(offset).Limit(pageSize).All(&list)
	} else {
		_, err = o.QueryTable(m.TableNameWithPrefix()).OrderBy("-attachment_id").All(&list)
	}

	if err != nil {
		if err == orm.ErrNoRows {
			logs.Info("没有查到附件 ->", err)
			err = nil
		}
		return
	}

	for _, item := range list {
		attach := &AttachmentResult{}
		attach.Attachment = *item
		attach.FileShortSize = filetil.FormatBytes(int64(attach.FileSize))
		//当项目ID为0标识是文章的附件
		if item.BookId == 0 && item.DocumentId > 0 {
			blog := NewBlog()
			if err := o.QueryTable(blog.TableNameWithPrefix()).Filter("blog_id", item.DocumentId).One(blog, "blog_title"); err == nil {
				attach.BookName = blog.BlogTitle
			} else {
				attach.BookName = "[文章不存在]"
			}
		} else {
			book := NewBook()

			if e := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", item.BookId).One(book, "book_name"); e == nil {
				attach.BookName = book.BookName

				doc := NewDocument()

				if e := o.QueryTable(doc.TableNameWithPrefix()).Filter("document_id", item.DocumentId).One(doc, "document_name"); e == nil {
					attach.DocumentName = doc.DocumentName
				} else {
					attach.DocumentName = "[文档不存在]"
				}

			} else {
				attach.BookName = "[项目不存在]"
			}
		}
		attach.LocalHttpPath = strings.Replace(item.FilePath, "\\", "/", -1)

		attachList = append(attachList, attach)
	}

	return
}