mindoc/models/TeamMember.go

266 lines
7.2 KiB
Go
Raw Normal View History

2018-11-05 18:50:01 +08:00
package models
import (
"errors"
"github.com/beego/beego/v2/client/orm"
"github.com/beego/beego/v2/core/logs"
"github.com/beego/i18n"
"github.com/mindoc-org/mindoc/conf"
2018-11-05 18:50:01 +08:00
)
type TeamMember struct {
TeamMemberId int `orm:"column(team_member_id);pk;auto;unique;" json:"team_member_id"`
TeamId int `orm:"column(team_id);type(int);description(团队id)" json:"team_id"`
MemberId int `orm:"column(member_id);type(int);description(成员id)" json:"member_id"`
2018-11-05 18:50:01 +08:00
// RoleId 角色0 创始人(创始人不能被移除) / 1 管理员/2 编辑者/3 观察者
RoleId conf.BookRole `orm:"column(role_id);type(int);description(RoleId 角色0 创始人-创始人不能被移除 / 1 管理员/2 编辑者/3 观察者)" json:"role_id"`
2018-11-05 18:50:01 +08:00
RoleName string `orm:"-" json:"role_name"`
Account string `orm:"-" json:"account"`
RealName string `orm:"-" json:"real_name"`
Avatar string `orm:"-" json:"avatar"`
Lang string `orm:"-"`
2018-11-05 18:50:01 +08:00
}
// TableName 获取对应数据库表名.
func (m *TeamMember) TableName() string {
return "team_member"
}
func (m *TeamMember) TableNameWithPrefix() string {
return conf.GetDatabasePrefix() + m.TableName()
}
// TableEngine 获取数据使用的引擎.
func (m *TeamMember) TableEngine() string {
return "INNODB"
}
// 联合唯一键
func (m *TeamMember) TableUnique() [][]string {
return [][]string{{"team_id", "member_id"}}
}
2018-11-12 21:01:59 +08:00
func (m *TeamMember) QueryTable() orm.QuerySeter {
return orm.NewOrm().QueryTable(m.TableNameWithPrefix())
}
2018-11-05 18:50:01 +08:00
func NewTeamMember() *TeamMember {
return &TeamMember{}
}
func (m *TeamMember) SetLang(lang string) *TeamMember {
m.Lang = lang
return m
}
2018-11-05 18:50:01 +08:00
func (m *TeamMember) First(id int, cols ...string) (*TeamMember, error) {
if id <= 0 {
return nil, errors.New("参数错误")
}
o := orm.NewOrm()
err := o.QueryTable(m.TableNameWithPrefix()).Filter("team_member_id", id).One(m, cols...)
if err != nil && err != orm.ErrNoRows {
logs.Error("查询团队成员错误 ->", err)
2018-11-05 18:50:01 +08:00
}
return m.Include(), err
}
func (m *TeamMember) ChangeRoleId(teamId int, memberId int, roleId conf.BookRole) (member *TeamMember, err error) {
if teamId <= 0 || memberId <= 0 || roleId <= 0 || roleId > conf.BookObserver {
return nil, ErrInvalidParameter
}
o := orm.NewOrm()
2018-11-06 18:38:18 +08:00
err = o.QueryTable(m.TableNameWithPrefix()).Filter("team_id", teamId).Filter("member_id", memberId).OrderBy("-team_member_id").One(m)
2018-11-05 18:50:01 +08:00
if err != nil {
logs.Error("查询团队用户时失败 ->", err)
2018-11-05 18:50:01 +08:00
return m, err
}
m.RoleId = roleId
err = m.Save("role_id")
if err == nil {
m.Include()
}
return m, err
}
2018-11-06 18:38:18 +08:00
//查询团队中指定的用户.
2018-11-05 18:50:01 +08:00
func (m *TeamMember) FindFirst(teamId, memberId int) (*TeamMember, error) {
if teamId <= 0 || memberId <= 0 {
return nil, ErrInvalidParameter
}
o := orm.NewOrm()
2018-11-12 21:01:59 +08:00
err := o.QueryTable(m.TableNameWithPrefix()).Filter("team_id", teamId).Filter("member_id", memberId).One(m)
2018-11-05 18:50:01 +08:00
if err != nil {
logs.Error("查询团队用户失败 ->", err)
2018-11-12 21:01:59 +08:00
return nil, err
2018-11-05 18:50:01 +08:00
}
2018-11-12 21:01:59 +08:00
return m.Include(), nil
2018-11-05 18:50:01 +08:00
}
2018-11-06 18:38:18 +08:00
//更新或插入团队用户.
2018-11-05 18:50:01 +08:00
func (m *TeamMember) Save(cols ...string) (err error) {
if m.TeamId <= 0 {
return errors.New("团队不能为空")
}
if m.MemberId <= 0 {
return errors.New("用户不能为空")
}
o := orm.NewOrm()
if !o.QueryTable(NewTeam().TableNameWithPrefix()).Filter("team_id", m.TeamId).Exist() {
return errors.New("团队不存在")
}
if !o.QueryTable(NewMember()).Filter("member_id", m.MemberId).Filter("status", 0).Exist() {
return errors.New("用户不存在或已禁用")
}
2018-11-06 18:38:18 +08:00
2018-11-05 18:50:01 +08:00
if m.TeamMemberId <= 0 {
2018-11-12 21:01:59 +08:00
if o.QueryTable(m.TableNameWithPrefix()).Filter("team_id", m.TeamId).Filter("member_id", m.MemberId).Exist() {
2018-11-06 18:38:18 +08:00
return errors.New("团队中已存在该用户")
}
2018-11-05 18:50:01 +08:00
_, err = o.Insert(m)
} else {
_, err = o.Update(m, cols...)
}
if err != nil {
logs.Error("在保存团队时出错 ->", err)
2018-11-05 18:50:01 +08:00
}
return
}
2018-11-06 18:38:18 +08:00
//删除一个团队用户.
2018-11-05 18:50:01 +08:00
func (m *TeamMember) Delete(id int) (err error) {
if id <= 0 {
return ErrInvalidParameter
}
_, err = orm.NewOrm().QueryTable(m.TableNameWithPrefix()).Filter("team_member_id", id).Delete()
if err != nil {
logs.Error("删除团队用户时出错 ->", err)
2018-11-05 18:50:01 +08:00
}
return
}
2018-11-06 18:38:18 +08:00
//分页查询团队用户.
2018-11-05 18:50:01 +08:00
func (m *TeamMember) FindToPager(teamId, pageIndex, pageSize int) (list []*TeamMember, totalCount int, err error) {
if teamId <= 0 {
err = ErrInvalidParameter
return
}
offset := (pageIndex - 1) * pageSize
o := orm.NewOrm()
_, err = o.QueryTable(m.TableNameWithPrefix()).Filter("team_id", teamId).Offset(offset).Limit(pageSize).All(&list)
if err != nil {
if err != orm.ErrNoRows {
logs.Error("查询团队成员失败 ->", err)
2018-11-05 18:50:01 +08:00
}
return
}
c, err := o.QueryTable(m.TableNameWithPrefix()).Filter("team_id", teamId).Count()
if err != nil {
return
}
totalCount = int(c)
//将来优化
for _, item := range list {
item.Lang = m.Lang
2018-11-05 18:50:01 +08:00
item.Include()
}
return
}
2018-11-06 18:38:18 +08:00
//查询关联数据.
2018-11-05 18:50:01 +08:00
func (m *TeamMember) Include() *TeamMember {
if member, err := NewMember().Find(m.MemberId, "account", "real_name", "avatar"); err == nil {
m.Account = member.Account
m.RealName = member.RealName
m.Avatar = member.Avatar
}
if m.RoleId == 0 {
m.RoleName = i18n.Tr(m.Lang, "common.creator") //"创始人"
2018-11-05 18:50:01 +08:00
} else if m.RoleId == 1 {
m.RoleName = i18n.Tr(m.Lang, "common.administrator") //"管理员"
2018-11-05 18:50:01 +08:00
} else if m.RoleId == 2 {
m.RoleName = i18n.Tr(m.Lang, "common.editor") //"编辑者"
2018-11-05 18:50:01 +08:00
} else if m.RoleId == 3 {
m.RoleName = i18n.Tr(m.Lang, "common.observer") //"观察者"
2018-11-05 18:50:01 +08:00
}
return m
}
2018-11-06 18:38:18 +08:00
//查询未加入团队的用户。
2018-11-05 18:50:01 +08:00
func (m *TeamMember) FindNotJoinMemberByAccount(teamId int, account string, limit int) (*SelectMemberResult, error) {
if teamId <= 0 {
return nil, ErrInvalidParameter
}
o := orm.NewOrm()
sql := `select mdmb.member_id,mdmb.account,mdmb.real_name,team.team_member_id
from md_members as mdmb
left join md_team_member as team on team.team_id = ? and mdmb.member_id = team.member_id
where mdmb.account like ? or mdmb.real_name like ? AND team_member_id IS NULL
order by mdmb.member_id desc
2018-11-05 18:50:01 +08:00
limit ?;`
members := make([]*Member, 0)
_, err := o.Raw(sql, teamId, "%"+account+"%", "%"+account+"%", limit).QueryRows(&members)
2018-11-05 18:50:01 +08:00
if err != nil {
logs.Error("查询团队用户时出错 ->", err)
2018-11-05 18:50:01 +08:00
return nil, err
}
result := SelectMemberResult{}
items := make([]KeyValueItem, 0)
for _, member := range members {
item := KeyValueItem{}
item.Id = member.MemberId
item.Text = member.Account + "[" + member.RealName + "]"
2018-11-05 18:50:01 +08:00
items = append(items, item)
}
result.Result = items
return &result, err
}
2018-11-12 21:01:59 +08:00
func (m *TeamMember) FindByBookIdAndMemberId(bookId, memberId int) (*TeamMember, error) {
2018-11-12 21:01:59 +08:00
if bookId <= 0 || memberId <= 0 {
return nil, ErrInvalidParameter
2018-11-12 21:01:59 +08:00
}
//一个用户可能在多个团队中,且一个项目可能有多个团队参与。因此需要查询用户最大权限。
sql := `select *
from md_team_member as team
where team.team_id in (select rel.team_id from md_team_relationship as rel where rel.book_id = ?)
and team.member_id = ? order by team.role_id asc limit 1;`
o := orm.NewOrm()
err := o.Raw(sql, bookId, memberId).QueryRow(m)
2018-11-12 21:01:59 +08:00
if err != nil {
logs.Error("查询用户项目所在团队失败 ->bookId=", bookId, " memberId=", memberId, err)
return nil, err
2018-11-12 21:01:59 +08:00
}
return m, nil
2018-11-12 21:01:59 +08:00
}