mindoc/models/MemberGroupModel.go

package models

import (
	"time"
	"github.com/lifei6671/mindoc/conf"
	"github.com/astaxie/beego/orm"
	"github.com/astaxie/beego"
)

type MemberGroup struct {
	GroupId int 		`orm:"column(group_id);pk;auto;unique;" json:"group_id"`
	GroupName string	`orm:"column(group_name);size(255);" json:"group_name"`
	GroupNumber int		`orm:"column(group_number);default(0)" json:"group_number"`
	CreateTime    time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"`
	CreateAt      int       `orm:"type(int);column(create_at)" json:"create_at"`
	CreateName 	string 		`orm:"-" json:"create_name"`
	CreateRealName string 	 `orm:"-" json:"create_real_name"`
	ModifyTime time.Time     `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
	ModifyAt   int           `orm:"column(modify_at);type(int)" json:"-"`
	ModifyName string 		 `orm:"-" json:"modify_name"`
	ModifyRealName string 	 `orm:"-" json:"modify_real_name"`
}



// TableName 获取对应数据库表名.
func (m *MemberGroup) TableName() string {
	return "member_group"
}

// TableEngine 获取数据使用的引擎.
func (m *MemberGroup) TableEngine() string {
	return "INNODB"
}

func (m *MemberGroup) TableNameWithPrefix() string {
	return conf.GetDatabasePrefix() + m.TableName()
}

func NewMemberGroup() *MemberGroup {
	return &MemberGroup{}
}

//根据id查询用户组
func (m *MemberGroup) FindFirst(id int) (*MemberGroup,error){
	o := orm.NewOrm()

	if err :=o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",id).One(m); err != nil {
		beego.Error("查询用户组时出错 =>",err)
		return m,err
	}
	createMember,err := NewMember().Find(m.CreateAt);
	if err != nil {
		beego.Error("查询用户组创建人失败 =>",err)
	}else{

		m.CreateName = createMember.Account
		m.CreateRealName = createMember.RealName
	}

	if m.ModifyAt > 0 {
		modifyMember, err := NewMember().Find(m.ModifyAt)
		if err != nil {
			beego.Error("查询用户组修改人失败 =>",err)
		}else{

			m.ModifyName = modifyMember.Account
			m.ModifyRealName = modifyMember.RealName
		}
	}
	return m,nil
}

//删除指定用户组
func (m *MemberGroup) Delete(id int) error {
	o := orm.NewOrm()

	o.Begin()
	_,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",id).Delete()

	if err != nil {
		o.Rollback()
		beego.Error("删除用户组失败 =>",err)
	}
	_,err = o.QueryTable(NewMemberGroupMembers().TableNameWithPrefix()).Filter("group_id",id).Delete()
	if err != nil {
		o.Rollback()
		beego.Error("删除用户组失败 =>",err)
	}
	return o.Commit()
}

//分页查询用户组
func (m *MemberGroup) FindByPager(pageIndex, pageSize int) ([]*MemberGroup,int,error){
	o := orm.NewOrm()

	if pageIndex <= 0 {
		pageIndex = 1
	}

	offset := (pageIndex - 1) * pageSize
	var memberGroups []*MemberGroup
	totalCount := 0
	_,err := o.QueryTable(m.TableNameWithPrefix()).OrderBy("-group_id").Offset(offset).Limit(pageSize).All(&memberGroups)

	if err != nil {
		beego.Error("分页查询用户组失败 =>",err)
	}else{
		i,err := o.QueryTable(m.TableNameWithPrefix()).Count()
		if err != nil {
			beego.Error("分页查询用户组失败 =>",err)
		}else {
			totalCount = int(i)
		}
	}
	memberIds := make([]int,0)

	for _,memberGroup := range memberGroups {
		if memberGroup.CreateAt > 0 {
			memberIds = append(memberIds,memberGroup.CreateAt)
		}
		if memberGroup.ModifyAt > 0 {
			memberIds = append(memberIds,memberGroup.ModifyAt)
		}
	}

	var members []*Member

	_,err = o.QueryTable(NewMember().TableNameWithPrefix()).Filter("member_id__in",memberIds).All(&members,"member_id","account","real_name")

	if err != nil {
		beego.Error("查询用户组信息时出错 =>",err)
	}else {
		for _,memberGroup := range memberGroups {
			for _,member := range members {
				if memberGroup.ModifyAt == member.MemberId {
					memberGroup.ModifyRealName = member.RealName
					memberGroup.ModifyName = member.Account
				}
				if memberGroup.CreateAt == member.MemberId {
					memberGroup.CreateRealName = member.RealName
					memberGroup.CreateName = member.Account
				}
			}
		}
	}

	return memberGroups,totalCount,err
}

//添加或更新用户组信息
func (m *MemberGroup) InsertOrUpdate(cols...string) error {
	o := orm.NewOrm()
	var err error
	if m.GroupId > 0 {
		_,err = o.Update(m, cols...)
	}else{
		_,err = o.Insert(m)
	}
	return err
}

//重置用户组数量
func (m *MemberGroup) ResetMemberGroupNumber(groupId int) error {
	o := orm.NewOrm()

	i,err := o.QueryTable(NewMemberGroupMembers().TableNameWithPrefix()).Filter("group_id",groupId).Count()
	if err != nil {
		beego.Error("重置用户组用户数量失败 =>",err)
	}else{
		err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).One(m)
		if err != nil {
			beego.Error("重置用户组用户数量失败 =>",err)
			return err
		}else{
			m.GroupNumber  = int(i)
			_,err = o.Update(m)
			return err
		}
	}
	return  nil
}

func (m *MemberGroup) Exist(groupId int) bool {
	o := orm.NewOrm()
	return o.QueryTable(m.TableNameWithPrefix()).Filter("group_id",groupId).Exist()
}

func (m *MemberGroup) FindMemberGroupList(keyword string) ([]*MemberGroup,error) {
	o := orm.NewOrm()
	var memberGroups []*MemberGroup
	_,err := o.QueryTable(m.TableNameWithPrefix()).Filter("group_name__icontains",keyword).All(&memberGroups)

	return memberGroups,err
}