user_position.go
2.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
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
}