user_position.go 2.6 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:无效"`
}

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
}

// GetUserPositionById retrieves UserPosition by Id. Returns error if
// Id doesn't exist
func GetUserPositionById(id int64) (v *UserPosition, err error) {
	o := orm.NewOrm()
	v = &UserPosition{Id: id}
	if err = o.Read(v); err == nil {
		return v, nil
	}
	return nil, err
}

// UpdateUserPosition updates UserPosition by Id and returns error if
// the record to be updated doesn't exist
func UpdateUserPositionById(m *UserPosition) (err error) {
	o := orm.NewOrm()
	v := UserPosition{Id: m.Id}
	// ascertain id exists in the database
	if err = o.Read(&v); err == nil {
		var num int64
		if num, err = o.Update(m); err == nil {
			fmt.Println("Number of records updated in database:", num)
		}
	}
	return
}

// DeleteUserPosition deletes UserPosition by Id and returns error if
// the record to be deleted doesn't exist
func DeleteUserPosition(id int64) (err error) {
	o := orm.NewOrm()
	v := UserPosition{Id: id}
	// ascertain id exists in the database
	if err = o.Read(&v); err == nil {
		var num int64
		if num, err = o.Delete(&UserPosition{Id: id}); err == nil {
			fmt.Println("Number of records deleted in database:", num)
		}
	}
	return
}