Template.go 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182
  1. package models
  2. import (
  3. "errors"
  4. "time"
  5. "github.com/beego/beego/v2/client/orm"
  6. "github.com/beego/beego/v2/core/logs"
  7. "github.com/mindoc-org/mindoc/conf"
  8. )
  9. type Template struct {
  10. TemplateId int `orm:"column(template_id);pk;auto;unique;" json:"template_id"`
  11. TemplateName string `orm:"column(template_name);size(500);" json:"template_name"`
  12. MemberId int `orm:"column(member_id);index" json:"member_id"`
  13. BookId int `orm:"column(book_id);index" json:"book_id"`
  14. BookName string `orm:"-" json:"book_name"`
  15. //是否是全局模板:0 否/1 是; 全局模板在所有项目中都可以使用;否则只能在创建模板的项目中使用
  16. IsGlobal int `orm:"column(is_global);default(0)" json:"is_global"`
  17. TemplateContent string `orm:"column(template_content);type(text);null" json:"template_content"`
  18. CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
  19. CreateName string `orm:"-" json:"create_name"`
  20. ModifyTime time.Time `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
  21. ModifyAt int `orm:"column(modify_at);type(int)" json:"-"`
  22. ModifyName string `orm:"-" json:"modify_name"`
  23. Version int64 `orm:"type(bigint);column(version)" json:"version"`
  24. }
  25. // TableName 获取对应数据库表名.
  26. func (m *Template) TableName() string {
  27. return "templates"
  28. }
  29. // TableEngine 获取数据使用的引擎.
  30. func (m *Template) TableEngine() string {
  31. return "INNODB"
  32. }
  33. func (m *Template) TableNameWithPrefix() string {
  34. return conf.GetDatabasePrefix() + m.TableName()
  35. }
  36. func NewTemplate() *Template {
  37. return &Template{}
  38. }
  39. //查询指定ID的模板
  40. func (t *Template) Find(templateId int) (*Template, error) {
  41. if templateId <= 0 {
  42. return t, ErrInvalidParameter
  43. }
  44. o := orm.NewOrm()
  45. err := o.QueryTable(t.TableNameWithPrefix()).Filter("template_id", templateId).One(t)
  46. if err != nil {
  47. logs.Error("查询模板时失败 ->%s", err)
  48. }
  49. return t, err
  50. }
  51. //查询属于指定项目的模板.
  52. func (t *Template) FindByBookId(bookId int) ([]*Template, error) {
  53. if bookId <= 0 {
  54. return nil, ErrInvalidParameter
  55. }
  56. o := orm.NewOrm()
  57. var templateList []*Template
  58. _, err := o.QueryTable(t.TableNameWithPrefix()).Filter("book_id", bookId).OrderBy("-template_id").All(&templateList)
  59. if err != nil {
  60. logs.Error("查询模板列表失败 ->", err)
  61. }
  62. return templateList, err
  63. }
  64. //查询指定项目所有可用模板列表.
  65. func (t *Template) FindAllByBookId(bookId int) ([]*Template, error) {
  66. if bookId <= 0 {
  67. return nil, ErrInvalidParameter
  68. }
  69. o := orm.NewOrm()
  70. cond := orm.NewCondition()
  71. cond1 := cond.And("book_id", bookId).Or("is_global", 1)
  72. qs := o.QueryTable(t.TableNameWithPrefix())
  73. var templateList []*Template
  74. _, err := qs.SetCond(cond1).OrderBy("-template_id").All(&templateList)
  75. if err != nil {
  76. logs.Error("查询模板列表失败 ->", err)
  77. }
  78. return templateList, err
  79. }
  80. //删除一个模板
  81. func (t *Template) Delete(templateId int, memberId int) error {
  82. if templateId <= 0 {
  83. return ErrInvalidParameter
  84. }
  85. o := orm.NewOrm()
  86. qs := o.QueryTable(t.TableNameWithPrefix()).Filter("template_id", templateId)
  87. if memberId > 0 {
  88. qs = qs.Filter("member_id", memberId)
  89. }
  90. _, err := qs.Delete()
  91. if err != nil {
  92. logs.Error("删除模板失败 ->", err)
  93. }
  94. return err
  95. }
  96. //添加或更新模板
  97. func (t *Template) Save(cols ...string) (err error) {
  98. if t.BookId <= 0 {
  99. return ErrInvalidParameter
  100. }
  101. o := orm.NewOrm()
  102. if !o.QueryTable(NewBook().TableNameWithPrefix()).Filter("book_id", t.BookId).Exist() {
  103. return errors.New("项目不存在")
  104. }
  105. if !o.QueryTable(NewMember().TableNameWithPrefix()).Filter("member_id", t.MemberId).Filter("status", 0).Exist() {
  106. return errors.New("用户已被禁用")
  107. }
  108. t.Version = time.Now().Unix()
  109. if t.TemplateId > 0 {
  110. t.ModifyTime = time.Now()
  111. _, err = o.Update(t, cols...)
  112. } else {
  113. t.CreateTime = time.Now()
  114. _, err = o.Insert(t)
  115. }
  116. return
  117. }
  118. //预加载一些数据
  119. func (t *Template) Preload() *Template {
  120. if t != nil {
  121. if t.MemberId > 0 {
  122. m, err := NewMember().Find(t.MemberId, "account", "real_name")
  123. if err == nil {
  124. if m.RealName != "" {
  125. t.CreateName = m.RealName
  126. } else {
  127. t.CreateName = m.Account
  128. }
  129. } else {
  130. logs.Error("加载模板所有者失败 ->", err)
  131. }
  132. }
  133. if t.ModifyAt > 0 {
  134. if m, err := NewMember().Find(t.ModifyAt, "account", "real_name"); err == nil {
  135. if m.RealName != "" {
  136. t.ModifyName = m.RealName
  137. } else {
  138. t.ModifyName = m.Account
  139. }
  140. }
  141. }
  142. if t.BookId > 0 {
  143. if b, err := NewBook().Find(t.BookId, "book_name"); err == nil {
  144. t.BookName = b.BookName
  145. }
  146. }
  147. }
  148. return t
  149. }