mirror of https://github.com/mindoc-org/mindoc.git
162 lines
3.7 KiB
Go
162 lines
3.7 KiB
Go
package models
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/beego/beego/v2/client/orm"
|
|
"github.com/beego/beego/v2/core/logs"
|
|
"github.com/mindoc-org/mindoc/conf"
|
|
)
|
|
|
|
//团队.
|
|
type Team struct {
|
|
TeamId int `orm:"column(team_id);pk;auto;unique;" json:"team_id"`
|
|
TeamName string `orm:"column(team_name);size(255)" json:"team_name"`
|
|
MemberId int `orm:"column(member_id);type(int);" json:"member_id"`
|
|
IsDelete bool `orm:"column(is_delete);default(0)" json:"is_delete"`
|
|
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
|
|
MemberCount int `orm:"-" json:"member_count"`
|
|
BookCount int `orm:"-" json:"book_count"`
|
|
MemberName string `orm:"-" json:"member_name"`
|
|
}
|
|
|
|
// TableName 获取对应数据库表名.
|
|
func (t *Team) TableName() string {
|
|
return "teams"
|
|
}
|
|
|
|
// TableEngine 获取数据使用的引擎.
|
|
func (t *Team) TableEngine() string {
|
|
return "INNODB"
|
|
}
|
|
|
|
func (t *Team) TableNameWithPrefix() string {
|
|
return conf.GetDatabasePrefix() + t.TableName()
|
|
}
|
|
|
|
func NewTeam() *Team {
|
|
return &Team{}
|
|
}
|
|
|
|
// 查询一个团队.
|
|
func (t *Team) First(id int, cols ...string) (*Team, error) {
|
|
if id <= 0 {
|
|
return nil, orm.ErrNoRows
|
|
}
|
|
o := orm.NewOrm()
|
|
err := o.QueryTable(t.TableNameWithPrefix()).Filter("team_id", id).One(t, cols...)
|
|
|
|
if err != nil {
|
|
logs.Error("查询团队失败 ->", id, err)
|
|
return nil, err
|
|
}
|
|
t.Include()
|
|
return t, err
|
|
}
|
|
|
|
func (t *Team) Delete(id int) (err error) {
|
|
if id <= 0 {
|
|
return ErrInvalidParameter
|
|
}
|
|
ormer := orm.NewOrm()
|
|
|
|
o, err := ormer.Begin()
|
|
|
|
if err != nil {
|
|
logs.Error("开启事物时出错 ->", err)
|
|
return
|
|
}
|
|
_, err = o.QueryTable(t.TableNameWithPrefix()).Filter("team_id", id).Delete()
|
|
|
|
if err != nil {
|
|
logs.Error("删除团队时出错 ->", err)
|
|
o.Rollback()
|
|
return
|
|
}
|
|
|
|
_, err = o.Raw("delete from md_team_member where team_id=?;", id).Exec()
|
|
|
|
if err != nil {
|
|
logs.Error("删除团队成员时出错 ->", err)
|
|
o.Rollback()
|
|
return
|
|
}
|
|
|
|
_, err = o.Raw("delete from md_team_relationship where team_id=?;", id).Exec()
|
|
|
|
if err != nil {
|
|
logs.Error("删除团队项目时出错 ->", err)
|
|
o.Rollback()
|
|
return err
|
|
}
|
|
|
|
err = o.Commit()
|
|
return
|
|
}
|
|
|
|
//分页查询团队.
|
|
func (t *Team) FindToPager(pageIndex, pageSize int) (list []*Team, totalCount int, err error) {
|
|
o := orm.NewOrm()
|
|
|
|
offset := (pageIndex - 1) * pageSize
|
|
|
|
_, err = o.QueryTable(t.TableNameWithPrefix()).OrderBy("-team_id").Offset(offset).Limit(pageSize).All(&list)
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
c, err := o.QueryTable(t.TableNameWithPrefix()).Count()
|
|
if err != nil {
|
|
return
|
|
}
|
|
totalCount = int(c)
|
|
|
|
for _, item := range list {
|
|
item.Include()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (t *Team) Include() {
|
|
|
|
o := orm.NewOrm()
|
|
|
|
if member, err := NewMember().Find(t.MemberId, "account", "real_name"); err == nil {
|
|
if member.RealName != "" {
|
|
t.MemberName = member.RealName
|
|
} else {
|
|
t.MemberName = member.Account
|
|
}
|
|
}
|
|
if c, err := o.QueryTable(NewTeamRelationship().TableNameWithPrefix()).Filter("team_id", t.TeamId).Count(); err == nil {
|
|
t.BookCount = int(c)
|
|
}
|
|
if c, err := o.QueryTable(NewTeamMember().TableNameWithPrefix()).Filter("team_id", t.TeamId).Count(); err == nil {
|
|
t.MemberCount = int(c)
|
|
}
|
|
}
|
|
|
|
//更新或添加一个团队.
|
|
func (t *Team) Save(cols ...string) (err error) {
|
|
if t.TeamName == "" {
|
|
return NewError(5001, "团队名称不能为空")
|
|
}
|
|
|
|
o := orm.NewOrm()
|
|
|
|
if t.TeamId <= 0 && o.QueryTable(t.TableNameWithPrefix()).Filter("team_name", t.TeamName).Exist() {
|
|
return errors.New("团队名称已存在")
|
|
}
|
|
if t.TeamId <= 0 {
|
|
_, err = o.Insert(t)
|
|
} else {
|
|
_, err = o.Update(t, cols...)
|
|
}
|
|
if err != nil {
|
|
logs.Error("在保存团队时出错 ->", err)
|
|
}
|
|
return
|
|
}
|