mirror of
https://github.com/mindoc-org/mindoc.git
synced 2026-02-27 17:03:57 +08:00
1、实现标签功能
2、实现删除用户,当删除一个用户时,该用户的所有信息将转义到超级管理员账户上
This commit is contained in:
326
models/book.go
326
models/book.go
@@ -3,53 +3,55 @@ package models
|
||||
import (
|
||||
"time"
|
||||
|
||||
"strings"
|
||||
|
||||
"github.com/astaxie/beego"
|
||||
"github.com/astaxie/beego/logs"
|
||||
"github.com/astaxie/beego/orm"
|
||||
"github.com/lifei6671/mindoc/conf"
|
||||
"github.com/astaxie/beego/logs"
|
||||
"strings"
|
||||
"github.com/astaxie/beego"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// Book struct .
|
||||
type Book struct {
|
||||
BookId int `orm:"pk;auto;unique;column(book_id)" json:"book_id"`
|
||||
BookId int `orm:"pk;auto;unique;column(book_id)" json:"book_id"`
|
||||
// BookName 项目名称.
|
||||
BookName string `orm:"column(book_name);size(500)" json:"book_name"`
|
||||
BookName string `orm:"column(book_name);size(500)" json:"book_name"`
|
||||
// Identify 项目唯一标识.
|
||||
Identify string `orm:"column(identify);size(100);unique" json:"identify"`
|
||||
OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`
|
||||
Identify string `orm:"column(identify);size(100);unique" json:"identify"`
|
||||
OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`
|
||||
// Description 项目描述.
|
||||
Description string `orm:"column(description);size(2000)" json:"description"`
|
||||
Label string `orm:"column(label);size(500)" json:"label"`
|
||||
Description string `orm:"column(description);size(2000)" json:"description"`
|
||||
Label string `orm:"column(label);size(500)" json:"label"`
|
||||
// PrivatelyOwned 项目私有: 0 公开/ 1 私有
|
||||
PrivatelyOwned int `orm:"column(privately_owned);type(int);default(0)" json:"privately_owned"`
|
||||
PrivatelyOwned int `orm:"column(privately_owned);type(int);default(0)" json:"privately_owned"`
|
||||
// 当项目是私有时的访问Token.
|
||||
PrivateToken string `orm:"column(private_token);size(500);null" json:"private_token"`
|
||||
PrivateToken string `orm:"column(private_token);size(500);null" json:"private_token"`
|
||||
//状态:0 正常/1 已删除
|
||||
Status int `orm:"column(status);type(int);default(0)" json:"status"`
|
||||
Status int `orm:"column(status);type(int);default(0)" json:"status"`
|
||||
//默认的编辑器.
|
||||
Editor string `orm:"column(editor);size(50)" json:"editor"`
|
||||
Editor string `orm:"column(editor);size(50)" json:"editor"`
|
||||
// DocCount 包含文档数量.
|
||||
DocCount int `orm:"column(doc_count);type(int)" json:"doc_count"`
|
||||
DocCount int `orm:"column(doc_count);type(int)" json:"doc_count"`
|
||||
// CommentStatus 评论设置的状态:open 为允许所有人评论,closed 为不允许评论, group_only 仅允许参与者评论 ,registered_only 仅允许注册者评论.
|
||||
CommentStatus string `orm:"column(comment_status);size(20);default(open)" json:"comment_status"`
|
||||
CommentCount int `orm:"column(comment_count);type(int)" json:"comment_count"`
|
||||
CommentStatus string `orm:"column(comment_status);size(20);default(open)" json:"comment_status"`
|
||||
CommentCount int `orm:"column(comment_count);type(int)" json:"comment_count"`
|
||||
//封面地址
|
||||
Cover string `orm:"column(cover);size(1000)" json:"cover"`
|
||||
Cover string `orm:"column(cover);size(1000)" json:"cover"`
|
||||
//主题风格
|
||||
Theme string `orm:"column(theme);size(255);default(default)" json:"theme"`
|
||||
Theme string `orm:"column(theme);size(255);default(default)" json:"theme"`
|
||||
// CreateTime 创建时间 .
|
||||
CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
|
||||
MemberId int `orm:"column(member_id);size(100)" json:"member_id"`
|
||||
ModifyTime time.Time `orm:"type(datetime);column(modify_time);null;auto_now" json:"modify_time"`
|
||||
Version int64 `orm:"type(bigint);column(version)" json:"version"`
|
||||
CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
|
||||
MemberId int `orm:"column(member_id);size(100)" json:"member_id"`
|
||||
ModifyTime time.Time `orm:"type(datetime);column(modify_time);null;auto_now" json:"modify_time"`
|
||||
Version int64 `orm:"type(bigint);column(version)" json:"version"`
|
||||
}
|
||||
|
||||
|
||||
// TableName 获取对应数据库表名.
|
||||
func (m *Book) TableName() string {
|
||||
return "books"
|
||||
}
|
||||
|
||||
// TableEngine 获取数据使用的引擎.
|
||||
func (m *Book) TableEngine() string {
|
||||
return "INNODB"
|
||||
@@ -64,18 +66,22 @@ func NewBook() *Book {
|
||||
|
||||
func (m *Book) Insert() error {
|
||||
o := orm.NewOrm()
|
||||
// o.Begin()
|
||||
// o.Begin()
|
||||
|
||||
_,err := o.Insert(m)
|
||||
_, err := o.Insert(m)
|
||||
|
||||
if err == nil {
|
||||
if m.Label != "" {
|
||||
NewLabel().InsertOrUpdateMulti(m.Label)
|
||||
}
|
||||
|
||||
relationship := NewRelationship()
|
||||
relationship.BookId = m.BookId
|
||||
relationship.RoleId = 0
|
||||
relationship.MemberId = m.MemberId
|
||||
err = relationship.Insert()
|
||||
if err != nil {
|
||||
logs.Error("插入项目与用户关联 => ",err)
|
||||
logs.Error("插入项目与用户关联 => ", err)
|
||||
//o.Rollback()
|
||||
return err
|
||||
}
|
||||
@@ -84,7 +90,7 @@ func (m *Book) Insert() error {
|
||||
document.DocumentName = "空白文档"
|
||||
document.MemberId = m.MemberId
|
||||
err = document.InsertOrUpdate()
|
||||
if err != nil{
|
||||
if err != nil {
|
||||
//o.Rollback()
|
||||
return err
|
||||
}
|
||||
@@ -95,113 +101,113 @@ func (m *Book) Insert() error {
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *Book) Find(id int) (*Book,error) {
|
||||
func (m *Book) Find(id int) (*Book, error) {
|
||||
if id <= 0 {
|
||||
return m,ErrInvalidParameter
|
||||
return m, ErrInvalidParameter
|
||||
}
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", id).One(m)
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("book_id",id).One(m)
|
||||
|
||||
return m,err
|
||||
return m, err
|
||||
}
|
||||
|
||||
func (m *Book) Update(cols... string) error {
|
||||
func (m *Book) Update(cols ...string) error {
|
||||
o := orm.NewOrm()
|
||||
|
||||
_,err := o.Update(m,cols...)
|
||||
temp := NewBook()
|
||||
temp.BookId = m.BookId
|
||||
|
||||
if err := o.Read(temp);err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if (m.Label + temp.Label) != "" {
|
||||
|
||||
go NewLabel().InsertOrUpdateMulti(m.Label + "," + temp.Label)
|
||||
}
|
||||
|
||||
_, err := o.Update(m, cols...)
|
||||
return err
|
||||
}
|
||||
|
||||
//根据指定字段查询结果集.
|
||||
func (m *Book) FindByField(field string,value interface{}) ([]*Book,error) {
|
||||
func (m *Book) FindByField(field string, value interface{}) ([]*Book, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
var books []*Book
|
||||
_,err := o.QueryTable(m.TableNameWithPrefix()).Filter(field,value).All(&books)
|
||||
_, err := o.QueryTable(m.TableNameWithPrefix()).Filter(field, value).All(&books)
|
||||
|
||||
return books,err
|
||||
return books, err
|
||||
}
|
||||
|
||||
//根据指定字段查询一个结果.
|
||||
func (m *Book) FindByFieldFirst(field string,value interface{})(*Book,error) {
|
||||
func (m *Book) FindByFieldFirst(field string, value interface{}) (*Book, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter(field,value).One(m)
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter(field, value).One(m)
|
||||
|
||||
return m,err
|
||||
return m, err
|
||||
|
||||
}
|
||||
|
||||
func (m *Book) FindByIdentify(identify string) (*Book,error) {
|
||||
func (m *Book) FindByIdentify(identify string) (*Book, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("identify",identify).One(m)
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("identify", identify).One(m)
|
||||
|
||||
return m,err
|
||||
return m, err
|
||||
}
|
||||
|
||||
//分页查询指定用户的项目
|
||||
func (m *Book) FindToPager(pageIndex, pageSize ,memberId int) (books []*BookResult,totalCount int,err error){
|
||||
func (m *Book) FindToPager(pageIndex, pageSize, memberId int) (books []*BookResult, totalCount int, err error) {
|
||||
|
||||
relationship := NewRelationship()
|
||||
|
||||
o := orm.NewOrm()
|
||||
|
||||
qb, _ := orm.NewQueryBuilder("mysql")
|
||||
sql1 := "SELECT COUNT(book.book_id) AS total_count FROM " + m.TableNameWithPrefix() + " AS book LEFT JOIN " +
|
||||
relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ? WHERE rel.relationship_id > 0 "
|
||||
|
||||
qb.Select("COUNT(book.book_id) AS total_count").
|
||||
From(m.TableNameWithPrefix() + " AS book").
|
||||
LeftJoin(relationship.TableNameWithPrefix() + " AS rel").
|
||||
On("book.book_id=rel.book_id AND rel.member_id = ?").
|
||||
Where("rel.relationship_id > 0")
|
||||
|
||||
err = o.Raw(qb.String(),memberId).QueryRow(&totalCount)
|
||||
err = o.Raw(sql1, memberId).QueryRow(&totalCount)
|
||||
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
offset := (pageIndex - 1) * pageSize
|
||||
qb2,_ := orm.NewQueryBuilder("mysql")
|
||||
|
||||
qb2.Select("book.*,rel.member_id","rel.role_id","m.account as create_name").
|
||||
From(m.TableNameWithPrefix() + " AS book").
|
||||
LeftJoin(relationship.TableNameWithPrefix() + " AS rel").On("book.book_id=rel.book_id AND rel.member_id = ?").
|
||||
LeftJoin(relationship.TableNameWithPrefix() + " AS rel1").On("book.book_id=rel1.book_id AND rel1.role_id=0").
|
||||
LeftJoin(NewMember().TableNameWithPrefix() + " AS m").On("rel1.member_id=m.member_id").
|
||||
Where("rel.relationship_id > 0").
|
||||
OrderBy("book.order_index DESC ","book.book_id").Desc().
|
||||
Limit(pageSize).
|
||||
Offset(offset)
|
||||
sql2 := "SELECT book.*,rel.member_id,rel.role_id,m.account as create_name FROM " + m.TableNameWithPrefix() + " AS book" +
|
||||
" LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel ON book.book_id=rel.book_id AND rel.member_id = ?" +
|
||||
" LEFT JOIN " + relationship.TableNameWithPrefix() + " AS rel1 ON book.book_id=rel1.book_id AND rel1.role_id=0" +
|
||||
" LEFT JOIN " + NewMember().TableNameWithPrefix() + " AS m ON rel1.member_id=m.member_id " +
|
||||
" WHERE rel.relationship_id > 0 ORDER BY book.order_index DESC,book.book_id DESC LIMIT " + fmt.Sprintf("%d,%d",offset,pageSize)
|
||||
|
||||
_,err = o.Raw(qb2.String(),memberId).QueryRows(&books)
|
||||
_, err = o.Raw(sql2, memberId).QueryRows(&books)
|
||||
if err != nil {
|
||||
logs.Error("分页查询项目列表 => ",err)
|
||||
logs.Error("分页查询项目列表 => ", err)
|
||||
return
|
||||
}
|
||||
sql := "SELECT m.account,doc.modify_time FROM md_documents AS doc LEFT JOIN md_members AS m ON doc.modify_at=m.member_id WHERE book_id = ? LIMIT 1 ORDER BY doc.modify_time DESC"
|
||||
|
||||
if err == nil && len(books) > 0{
|
||||
for index,book := range books {
|
||||
var text struct{
|
||||
Account string
|
||||
if err == nil && len(books) > 0 {
|
||||
for index, book := range books {
|
||||
var text struct {
|
||||
Account string
|
||||
ModifyTime time.Time
|
||||
}
|
||||
|
||||
|
||||
err1 := o.Raw(sql,book.BookId).QueryRow(&text)
|
||||
err1 := o.Raw(sql, book.BookId).QueryRow(&text)
|
||||
if err1 == nil {
|
||||
books[index].LastModifyText = text.Account + " 于 " + text.ModifyTime.Format("2006-01-02 15:04:05")
|
||||
}
|
||||
if book.RoleId == 0{
|
||||
if book.RoleId == 0 {
|
||||
book.RoleName = "创始人"
|
||||
}else if book.RoleId == 1 {
|
||||
} else if book.RoleId == 1 {
|
||||
book.RoleName = "管理员"
|
||||
}else if book.RoleId == 2 {
|
||||
} else if book.RoleId == 2 {
|
||||
book.RoleName = "编辑者"
|
||||
}else if book.RoleId == 3 {
|
||||
} else if book.RoleId == 3 {
|
||||
book.RoleName = "观察者"
|
||||
}
|
||||
}
|
||||
@@ -211,7 +217,7 @@ func (m *Book) FindToPager(pageIndex, pageSize ,memberId int) (books []*BookResu
|
||||
|
||||
// 彻底删除项目.
|
||||
func (m *Book) ThoroughDeleteBook(id int) error {
|
||||
if id <= 0{
|
||||
if id <= 0 {
|
||||
return ErrInvalidParameter
|
||||
}
|
||||
o := orm.NewOrm()
|
||||
@@ -221,17 +227,10 @@ func (m *Book) ThoroughDeleteBook(id int) error {
|
||||
return err
|
||||
}
|
||||
o.Begin()
|
||||
//sql1 := "DELETE FROM " + NewComment().TableNameWithPrefix() + " WHERE book_id = ?"
|
||||
//
|
||||
//_,err := o.Raw(sql1,m.BookId).Exec()
|
||||
//
|
||||
//if err != nil {
|
||||
// o.Rollback()
|
||||
// return err
|
||||
//}
|
||||
|
||||
sql2 := "DELETE FROM " + NewDocument().TableNameWithPrefix() + " WHERE book_id = ?"
|
||||
|
||||
_,err := o.Raw(sql2,m.BookId).Exec()
|
||||
_, err := o.Raw(sql2, m.BookId).Exec()
|
||||
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
@@ -239,7 +238,7 @@ func (m *Book) ThoroughDeleteBook(id int) error {
|
||||
}
|
||||
sql3 := "DELETE FROM " + m.TableNameWithPrefix() + " WHERE book_id = ?"
|
||||
|
||||
_,err = o.Raw(sql3,m.BookId).Exec()
|
||||
_, err = o.Raw(sql3, m.BookId).Exec()
|
||||
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
@@ -247,19 +246,23 @@ func (m *Book) ThoroughDeleteBook(id int) error {
|
||||
}
|
||||
sql4 := "DELETE FROM " + NewRelationship().TableNameWithPrefix() + " WHERE book_id = ?"
|
||||
|
||||
_,err = o.Raw(sql4,m.BookId).Exec()
|
||||
_, err = o.Raw(sql4, m.BookId).Exec()
|
||||
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
if m.Label != "" {
|
||||
NewLabel().InsertOrUpdateMulti(m.Label)
|
||||
}
|
||||
|
||||
return o.Commit()
|
||||
|
||||
}
|
||||
|
||||
//分页查找系统首页数据.
|
||||
func (m *Book) FindForHomeToPager(pageIndex, pageSize ,member_id int) (books []*BookResult,totalCount int,err error) {
|
||||
func (m *Book) FindForHomeToPager(pageIndex, pageSize, member_id int) (books []*BookResult, totalCount int, err error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
offset := (pageIndex - 1) * pageSize
|
||||
@@ -267,7 +270,7 @@ func (m *Book) FindForHomeToPager(pageIndex, pageSize ,member_id int) (books []*
|
||||
if member_id > 0 {
|
||||
sql1 := "SELECT COUNT(*) FROM md_books AS book LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ? WHERE relationship_id > 0 OR book.privately_owned = 0"
|
||||
|
||||
err = o.Raw(sql1,member_id).QueryRow(&totalCount)
|
||||
err = o.Raw(sql1, member_id).QueryRow(&totalCount)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
@@ -277,12 +280,12 @@ func (m *Book) FindForHomeToPager(pageIndex, pageSize ,member_id int) (books []*
|
||||
LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
|
||||
WHERE rel.relationship_id > 0 OR book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
|
||||
|
||||
_,err = o.Raw(sql2,member_id,offset,pageSize).QueryRows(&books)
|
||||
_, err = o.Raw(sql2, member_id, offset, pageSize).QueryRows(&books)
|
||||
|
||||
return
|
||||
|
||||
}else{
|
||||
count,err1 := o.QueryTable(m.TableNameWithPrefix()).Filter("privately_owned",0).Count()
|
||||
} else {
|
||||
count, err1 := o.QueryTable(m.TableNameWithPrefix()).Filter("privately_owned", 0).Count()
|
||||
|
||||
if err1 != nil {
|
||||
err = err1
|
||||
@@ -295,7 +298,7 @@ func (m *Book) FindForHomeToPager(pageIndex, pageSize ,member_id int) (books []*
|
||||
LEFT JOIN md_members AS member ON rel.member_id = member.member_id
|
||||
WHERE book.privately_owned = 0 ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
|
||||
|
||||
_,err = o.Raw(sql,offset,pageSize).QueryRows(&books)
|
||||
_, err = o.Raw(sql, offset, pageSize).QueryRows(&books)
|
||||
|
||||
return
|
||||
|
||||
@@ -303,30 +306,75 @@ func (m *Book) FindForHomeToPager(pageIndex, pageSize ,member_id int) (books []*
|
||||
|
||||
}
|
||||
|
||||
//分页全局搜索.
|
||||
func (m *Book) FindForLabelToPager(keyword string, pageIndex, pageSize, member_id int) (books []*BookResult, totalCount int, err error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
keyword = "%" + keyword + "%"
|
||||
offset := (pageIndex - 1) * pageSize
|
||||
//如果是登录用户
|
||||
if member_id > 0 {
|
||||
sql1 := "SELECT COUNT(*) FROM md_books AS book LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ? WHERE (relationship_id > 0 OR book.privately_owned = 0) AND book.label LIKE ?"
|
||||
|
||||
err = o.Raw(sql1, member_id,keyword).QueryRow(&totalCount)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
sql2 := `SELECT book.*,rel1.*,member.account AS create_name FROM md_books AS book
|
||||
LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.member_id = ?
|
||||
LEFT JOIN md_relationship AS rel1 ON rel1.book_id = book.book_id AND rel1.role_id = 0
|
||||
LEFT JOIN md_members AS member ON rel1.member_id = member.member_id
|
||||
WHERE (rel.relationship_id > 0 OR book.privately_owned = 0) AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
|
||||
|
||||
_, err = o.Raw(sql2, member_id,keyword, offset, pageSize).QueryRows(&books)
|
||||
|
||||
return
|
||||
|
||||
} else {
|
||||
count, err1 := o.QueryTable(NewBook().TableNameWithPrefix()).Filter("privately_owned", 0).Filter("label__icontains",keyword).Count()
|
||||
|
||||
if err1 != nil {
|
||||
err = err1
|
||||
return
|
||||
}
|
||||
totalCount = int(count)
|
||||
|
||||
sql := `SELECT book.*,rel.*,member.account AS create_name FROM md_books AS book
|
||||
LEFT JOIN md_relationship AS rel ON rel.book_id = book.book_id AND rel.role_id = 0
|
||||
LEFT JOIN md_members AS member ON rel.member_id = member.member_id
|
||||
WHERE book.privately_owned = 0 AND book.label LIKE ? ORDER BY order_index DESC ,book.book_id DESC LIMIT ?,?`
|
||||
|
||||
_, err = o.Raw(sql,keyword, offset, pageSize).QueryRows(&books)
|
||||
|
||||
return
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
func (book *Book) ToBookResult() *BookResult {
|
||||
|
||||
m := NewBookResult()
|
||||
|
||||
m.BookId = book.BookId
|
||||
m.BookName = book.BookName
|
||||
m.Identify = book.Identify
|
||||
m.OrderIndex = book.OrderIndex
|
||||
m.Description = strings.Replace(book.Description, "\r\n", "<br/>", -1)
|
||||
m.PrivatelyOwned = book.PrivatelyOwned
|
||||
m.PrivateToken = book.PrivateToken
|
||||
m.DocCount = book.DocCount
|
||||
m.CommentStatus = book.CommentStatus
|
||||
m.CommentCount = book.CommentCount
|
||||
m.CreateTime = book.CreateTime
|
||||
m.ModifyTime = book.ModifyTime
|
||||
m.Cover = book.Cover
|
||||
m.Label = book.Label
|
||||
m.Status = book.Status
|
||||
m.Editor = book.Editor
|
||||
m.Theme = book.Theme
|
||||
m.BookId = book.BookId
|
||||
m.BookName = book.BookName
|
||||
m.Identify = book.Identify
|
||||
m.OrderIndex = book.OrderIndex
|
||||
m.Description = strings.Replace(book.Description, "\r\n", "<br/>", -1)
|
||||
m.PrivatelyOwned = book.PrivatelyOwned
|
||||
m.PrivateToken = book.PrivateToken
|
||||
m.DocCount = book.DocCount
|
||||
m.CommentStatus = book.CommentStatus
|
||||
m.CommentCount = book.CommentCount
|
||||
m.CreateTime = book.CreateTime
|
||||
m.ModifyTime = book.ModifyTime
|
||||
m.Cover = book.Cover
|
||||
m.Label = book.Label
|
||||
m.Status = book.Status
|
||||
m.Editor = book.Editor
|
||||
m.Theme = book.Theme
|
||||
|
||||
|
||||
if book.Theme == ""{
|
||||
if book.Theme == "" {
|
||||
m.Theme = "default"
|
||||
}
|
||||
if book.Editor == "" {
|
||||
@@ -336,61 +384,13 @@ func (book *Book) ToBookResult() *BookResult {
|
||||
}
|
||||
|
||||
//重置文档数量
|
||||
func (m *Book) ResetDocumentNumber(book_id int) {
|
||||
func (m *Book) ResetDocumentNumber(book_id int) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
totalCount,err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id",book_id).Count()
|
||||
totalCount, err := o.QueryTable(NewDocument().TableNameWithPrefix()).Filter("book_id", book_id).Count()
|
||||
if err == nil {
|
||||
o.Raw("UPDATE md_books SET doc_count = ? WHERE book_id = ?",int(totalCount),book_id).Exec()
|
||||
}else{
|
||||
o.Raw("UPDATE md_books SET doc_count = ? WHERE book_id = ?", int(totalCount), book_id).Exec()
|
||||
} else {
|
||||
beego.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -155,7 +155,7 @@ func (m *Document) ReleaseContent(book_id int) {
|
||||
func (m *Document) FindListByBookId(book_id int) (docs []*Document, err error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", book_id).All(&docs)
|
||||
_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("book_id", book_id).OrderBy("order_sort").All(&docs)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
92
models/label.go
Normal file
92
models/label.go
Normal file
@@ -0,0 +1,92 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"github.com/lifei6671/mindoc/conf"
|
||||
"github.com/astaxie/beego/orm"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Label struct {
|
||||
LabelId int `orm:"column(label_id);pk;auto;unique;" json:"label_id"`
|
||||
LabelName string `orm:"column(label_name);size(50);unique" json:"label_name"`
|
||||
BookNumber int `orm:"column(book_number)" json:"book_number"`
|
||||
}
|
||||
|
||||
// TableName 获取对应数据库表名.
|
||||
func (m *Label) TableName() string {
|
||||
return "label"
|
||||
}
|
||||
// TableEngine 获取数据使用的引擎.
|
||||
func (m *Label) TableEngine() string {
|
||||
return "INNODB"
|
||||
}
|
||||
|
||||
func (m *Label)TableNameWithPrefix() string {
|
||||
return conf.GetDatabasePrefix() + m.TableName()
|
||||
}
|
||||
|
||||
func NewLabel() *Label {
|
||||
return &Label{}
|
||||
}
|
||||
|
||||
func (m *Label) FindFirst(field string, value interface{}) (*Label,error){
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter(field, value).One(m)
|
||||
|
||||
return m, err
|
||||
}
|
||||
|
||||
//插入或更新标签.
|
||||
func (m *Label) InsertOrUpdate(labelName string) error {
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("label_name",labelName).One(m)
|
||||
if err != nil && err != orm.ErrNoRows {
|
||||
return err
|
||||
}
|
||||
count,_ := o.QueryTable(NewBook().TableNameWithPrefix()).Filter("label__icontains",labelName).Count()
|
||||
m.BookNumber = int(count)
|
||||
m.LabelName = labelName
|
||||
|
||||
if err == orm.ErrNoRows {
|
||||
err = nil
|
||||
m.LabelName = labelName
|
||||
_,err = o.Insert(m)
|
||||
}else{
|
||||
_,err = o.Update(m)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
//批量插入或更新标签.
|
||||
func (m *Label) InsertOrUpdateMulti(labels string) {
|
||||
if labels != "" {
|
||||
labelArray := strings.Split(labels, ",")
|
||||
|
||||
for _, label := range labelArray {
|
||||
if label != "" {
|
||||
NewLabel().InsertOrUpdate(label)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//分页查找标签.
|
||||
func (m *Label) FindToPager(pageIndex, pageSize int) (labels []*Label,totalCount int,err error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
count,err := o.QueryTable(m.TableNameWithPrefix()).Count()
|
||||
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
totalCount = int(count)
|
||||
|
||||
offset := (pageIndex - 1) * pageSize
|
||||
|
||||
_,err = o.QueryTable(m.TableNameWithPrefix()).OrderBy("-book_number").Offset(offset).Limit(pageSize).All(&labels)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
@@ -217,6 +217,7 @@ func (m *Member) ResolveRoleName() {
|
||||
}
|
||||
}
|
||||
|
||||
//根据账号查找用户.
|
||||
func (m *Member) FindByAccount(account string) (*Member, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
@@ -228,6 +229,7 @@ func (m *Member) FindByAccount(account string) (*Member, error) {
|
||||
return m, err
|
||||
}
|
||||
|
||||
//分页查找用户.
|
||||
func (m *Member) FindToPager(pageIndex, pageSize int) ([]*Member, int64, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
@@ -260,6 +262,7 @@ func (c *Member) IsAdministrator() bool {
|
||||
return c.Role == 0 || c.Role == 1
|
||||
}
|
||||
|
||||
//根据指定字段查找用户.
|
||||
func (m *Member) FindByFieldFirst(field string, value interface{}) (*Member, error) {
|
||||
o := orm.NewOrm()
|
||||
|
||||
@@ -268,6 +271,7 @@ func (m *Member) FindByFieldFirst(field string, value interface{}) (*Member, err
|
||||
return m, err
|
||||
}
|
||||
|
||||
//校验用户.
|
||||
func (m *Member) Valid(is_hash_password bool) error {
|
||||
|
||||
//邮箱不能为空
|
||||
@@ -324,6 +328,67 @@ func (m *Member) Valid(is_hash_password bool) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
//删除一个用户.
|
||||
|
||||
func (m *Member) Delete(oldId int,newId int) error {
|
||||
o := orm.NewOrm()
|
||||
|
||||
err := o.Begin()
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_,err = o.Raw("DELETE FROM md_members WHERE member_id = ?",oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_attachment SET `create_at` = ? WHERE `create_at` = ?",newId,oldId).Exec()
|
||||
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
|
||||
_,err = o.Raw("UPDATE md_books SET member_id = ? WHERE member_id = ?",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_document_history SET member_id=? WHERE member_id = ?",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_document_history SET modify_at=? WHERE modify_at = ?",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_documents SET member_id = ? WHERE member_id = ?;",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_documents SET modify_at = ? WHERE modify_at = ?",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
_,err = o.Raw("UPDATE md_relationship SET member_id = ? WHERE member_id = ?",newId,oldId).Exec()
|
||||
if err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
if err = o.Commit();err != nil {
|
||||
o.Rollback()
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user