user_position.go 2.2 KB
package models

import (
	"fmt"
	"time"

	"github.com/astaxie/beego/orm"
)

type UserPosition struct {
	Id            int64     `orm:"column(id);pk" description:"唯一键值"`
	UserId        int64     `orm:"column(user_id)" description:"表user.id 用户编号"`
	PositionId    int64     `orm:"column(position_id)" description:"表position.id 职位编号"`
	CreateAt      time.Time `orm:"column(create_at);type(timestamp);null" description:"创建时间"`
	CompanyId     int64     `orm:"column(company_id)" description:"表company.id 公司编号"`
	EnableStatus  int8      `orm:"column(enable_status);null" description:"是否有效 1:有效 2:无效"`
	UserCompanyId int64     `orm:"column(user_company_id)"`
}

func (t *UserPosition) TableName() string {
	return "user_position"
}

//EnableStatus 是否有效
const (
	USER_POSITION_ENABLE_YES int8 = 1 //有效
	USER_POSITION_ENABLE_NO  int8 = 2 //无效
)

func (t *UserPosition) IsEnable() bool {
	switch t.EnableStatus {
	case USER_POSITION_ENABLE_YES:
		return true
	case USER_POSITION_ENABLE_NO:
		return false
	}
	return false
}

func (t *UserPosition) ValidCompanyPosition() error {
	depart, err := GetPositionById(t.PositionId)
	if err != nil {
		return err
	}
	if depart.CompanyId != t.CompanyId {
		e := fmt.Errorf(" position.CompanyId != param.CompanyId ")
		return e
	}
	return nil
}

func init() {
	orm.RegisterModel(new(UserPosition))
}

// AddUserPosition insert a new UserPosition into database and returns
// last inserted Id on success.
func AddUserPosition(m *UserPosition) (id int64, err error) {
	o := orm.NewOrm()
	id, err = o.Insert(m)
	return
}

func GetUserPosition(userid, companyid int64) ([]*UserPosition, error) {
	o := orm.NewOrm()
	var (
		err    error
		result []*UserPosition
	)
	_, err = o.QueryTable(&UserPosition{}).
		Filter("user_id", userid).
		Filter("company_id", companyid).
		Filter("enable_status", 1).
		All(&result)
	return result, err
}

func CountUserPositionByPosition(positionid int64) (int64, error) {
	var (
		cnt int64
		err error
	)
	o := orm.NewOrm()
	cnt, err = o.QueryTable(&UserPosition{}).
		Filter("position_id", positionid).
		Filter("enable_status", 1).
		Count()
	return cnt, err
}