MemberGroupMemberModel.go 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257
  1. package models
  2. import (
  3. "github.com/lifei6671/mindoc/conf"
  4. "time"
  5. "github.com/astaxie/beego/orm"
  6. "github.com/astaxie/beego"
  7. "errors"
  8. )
  9. type MemberGroupMembers struct {
  10. GroupMemberId int `orm:"column(group_member_id);pk;auto;unique;" json:"group_member_id"`
  11. MemberId int `orm:"column(member_id);index" json:"member_id"`
  12. GroupId int `orm:"column(group_id);index" json:"group_id"`
  13. CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
  14. CreateAt int `orm:"type(int);column(create_at)" json:"create_at"`
  15. }
  16. type MemberGroupMemberResult struct {
  17. GroupMemberId int `json:"group_member_id"`
  18. MemberId int `json:"member_id"`
  19. Account string `json:"account"`
  20. RealName string `json:"real_name"`
  21. RoleName string `json:"role_name"`
  22. Avatar string `json:"avatar"`
  23. GroupId int `json:"group_id"`
  24. CreateTime time.Time `json:"create_time"`
  25. CreateAt int `json:"create_at"`
  26. }
  27. //搜索未加入当前用户组的用户信息
  28. type MemberGroupMemberNoJoinSearchResult struct {
  29. MemberId int `json:"member_id"`
  30. Account string `json:"account"`
  31. }
  32. // TableName 获取对应数据库表名.
  33. func (m *MemberGroupMembers) TableName() string {
  34. return "member_group_members"
  35. }
  36. // TableEngine 获取数据使用的引擎.
  37. func (m *MemberGroupMembers) TableEngine() string {
  38. return "INNODB"
  39. }
  40. func (m *MemberGroupMembers) TableNameWithPrefix() string {
  41. return conf.GetDatabasePrefix() + m.TableName()
  42. }
  43. // 多字段唯一键
  44. func (m *MemberGroupMembers) TableUnique() [][]string {
  45. return [][]string{
  46. {"member_id", "group_id"},
  47. }
  48. }
  49. func NewMemberGroupMembers() *MemberGroupMembers {
  50. return &MemberGroupMembers{}
  51. }
  52. // 查询用户组成员
  53. func (m *MemberGroupMembers) FindByGroupId(groupId int) ([]*MemberGroupMemberResult,error) {
  54. o := orm.NewOrm()
  55. var groupMembers []*MemberGroupMemberResult
  56. _,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).All(&groupMembers);
  57. if err != nil {
  58. beego.Error("获取用户组成员出错 =>",err)
  59. return nil,err
  60. }
  61. ids := make([]int,0)
  62. for _,member := range groupMembers {
  63. ids = append(ids,member.MemberId)
  64. }
  65. var members []*Member
  66. _,err = o.QueryTable(NewMember().TableNameWithPrefix()).Filter("member_id__in",ids).All(&members)
  67. if err != nil {
  68. beego.Error("获取用户组成员出错 =>",err)
  69. return nil,err
  70. }
  71. for _,member := range members {
  72. for _,groupMember := range groupMembers {
  73. if groupMember.MemberId == member.MemberId {
  74. groupMember.Account = member.Account
  75. groupMember.Avatar = member.Avatar
  76. groupMember.RealName = member.RealName
  77. }
  78. }
  79. }
  80. return groupMembers,nil
  81. }
  82. //判断一个用户是否加入了指定用户组
  83. func (m *MemberGroupMembers) IsJoin(groupId ,memberId int) (bool) {
  84. o := orm.NewOrm()
  85. return o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).Filter("member_id",memberId).Exist()
  86. }
  87. //添加或更新用户组成员
  88. func (m *MemberGroupMembers) InsertOrUpdate(cols ...string) error {
  89. o := orm.NewOrm()
  90. if m.GroupMemberId > 0 {
  91. _,err := o.Update(m,cols...)
  92. if err != nil {
  93. beego.Error("更新用户组成员失败 =>",err)
  94. }
  95. return err
  96. }else{
  97. if m.GroupId <= 0 {
  98. return errors.New("用户组不能为空")
  99. }
  100. _,err := o.Insert(m)
  101. if err != nil {
  102. beego.Error("添加用户组成员失败 =>",err)
  103. }else{
  104. go NewMemberGroup().ResetMemberGroupNumber(m.GroupId)
  105. }
  106. return err
  107. }
  108. }
  109. //删除用户组成员
  110. func (m *MemberGroupMembers) Delete(id int) error {
  111. o := orm.NewOrm()
  112. if err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_member_id",id).One(m);err != nil {
  113. beego.Error("删除用户组成员失败 =>",err)
  114. return err
  115. }
  116. i,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_member_id",id).Delete()
  117. if err != nil {
  118. beego.Error("删除用户组成员失败 =>",err)
  119. return err
  120. }
  121. if i <= 0 {
  122. beego.Info("删除用户组成员返回行数 =>",i)
  123. }
  124. go NewMemberGroup().ResetMemberGroupNumber(m.GroupId)
  125. return nil
  126. }
  127. //分页获取用户组成员
  128. func (m *MemberGroupMembers) FindToPager(pageIndex, pageSize, groupId int) ([]*MemberGroupMemberResult,int,error) {
  129. o := orm.NewOrm()
  130. if pageIndex <= 0 {
  131. pageIndex = 1
  132. }
  133. offset := (pageIndex - 1) * pageSize
  134. var memberGroupMembers []*MemberGroupMembers
  135. totalCount := 0
  136. _,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).Offset(offset).Limit(pageSize).All(&memberGroupMembers)
  137. memberGroupMemberList := make([]*MemberGroupMemberResult,0)
  138. if err != nil {
  139. beego.Error("分页查询用户组成员失败 =>",err)
  140. }else{
  141. i,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).Count()
  142. if err != nil {
  143. beego.Error("分页查询用户组成员失败 =>",err)
  144. }else {
  145. totalCount = int(i)
  146. }
  147. for _,member := range memberGroupMembers {
  148. memberGroupMemberList = append(memberGroupMemberList,member.ToMemberRelationshipResult())
  149. }
  150. }
  151. return memberGroupMemberList,totalCount,err
  152. }
  153. //将用户组信息转换为完整的用户信息
  154. func (m *MemberGroupMembers) ToMemberRelationshipResult() *MemberGroupMemberResult {
  155. memberGroupMemberResult := &MemberGroupMemberResult{}
  156. memberGroupMemberResult.GroupId = m.GroupId
  157. memberGroupMemberResult.GroupMemberId = m.GroupMemberId
  158. memberGroupMemberResult.MemberId = m.MemberId
  159. memberGroupMemberResult.CreateAt = m.CreateAt
  160. memberGroupMemberResult.CreateTime = m.CreateTime
  161. if m.MemberId > 0 {
  162. member,err := NewMember().Find(memberGroupMemberResult.MemberId)
  163. if err != nil {
  164. beego.Error("查询用户组成员信息时出错 =>",err)
  165. }else{
  166. memberGroupMemberResult.RealName = member.RealName
  167. memberGroupMemberResult.Avatar = member.Avatar
  168. memberGroupMemberResult.Account = member.Account
  169. memberGroupMemberResult.RoleName = member.RoleName
  170. }
  171. }
  172. return memberGroupMemberResult
  173. }
  174. //获取未加入用户组的用户成员
  175. func (m *MemberGroupMembers) FindMemberGroupMemberNoJoinSearchResult(groupId int,q string) ([]*MemberGroupMemberNoJoinSearchResult,error) {
  176. o := orm.NewOrm()
  177. sql := "select member.member_id,member.account from md_members as member left join md_member_group_members as member_group on member_group.member_id = member.member_id and member_group.group_id=? where member_group.member_id isnull and account like ?limit 20;"
  178. var memberGroupMembers []*MemberGroupMemberNoJoinSearchResult
  179. _,err := o.Raw(sql,groupId,"%" + q + "%").QueryRows(&memberGroupMembers)
  180. if err != nil {
  181. beego.Error("获取未加入用户组的用户失败 =>",err)
  182. }
  183. return memberGroupMembers,err
  184. }