126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
package qyweixin
|
|
|
|
import (
|
|
"fmt"
|
|
"git.u8t.cn/open/gosdk/util"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/smbrave/goutil"
|
|
"github.com/spf13/cast"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
urlQyWeixinHrGetAllField = "https://qyapi.weixin.qq.com/cgi-bin/hr/get_fields"
|
|
urlQyWeixinHrGetStaffInfo = "https://qyapi.weixin.qq.com/cgi-bin/hr/get_staff_info"
|
|
)
|
|
|
|
type AppHr struct {
|
|
App
|
|
config *AppConfig
|
|
}
|
|
|
|
type StaffInfo struct {
|
|
UserName string
|
|
RealName string
|
|
Extra map[string]string
|
|
Phone string
|
|
StaffType string
|
|
Idno string
|
|
Salary float64
|
|
PerfSalary float64
|
|
Stock float64
|
|
EntryDate string
|
|
BirthDate string
|
|
OfficialDate string
|
|
BankName string
|
|
BankCard string
|
|
AlipayUid string
|
|
}
|
|
|
|
func NewAppHr(cfg *AppConfig) *AppHr {
|
|
return &AppHr{
|
|
App: *NewApp(cfg),
|
|
}
|
|
}
|
|
|
|
func (h *AppHr) GetStaffInfo(userId string) (*StaffInfo, error) {
|
|
reqUrl := fmt.Sprintf("%s?access_token=%s", urlQyWeixinHrGetStaffInfo, h.GetToken())
|
|
reqBody := make(map[string]interface{})
|
|
reqBody["userid"] = userId
|
|
reqBody["get_all"] = true
|
|
rspBody, err := util.HttpPostJson(reqUrl, nil, []byte(goutil.EncodeJSON(reqBody)))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
staff := new(StaffInfo)
|
|
result, err := h.GetResult(rspBody)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fieldMap := make(map[string]map[string]interface{})
|
|
for _, fieldInfo := range cast.ToSlice(result["field_info"]) {
|
|
fi := cast.ToStringMap(fieldInfo)
|
|
fieldMap[cast.ToString(fi["fieldid"])] = fi
|
|
}
|
|
|
|
userInfo, err := h.GetUserInfo(userId)
|
|
if err != nil {
|
|
log.Errorf("GetUserInfo error:%s", err.Error())
|
|
return nil, err
|
|
}
|
|
staff.Extra = make(map[string]string)
|
|
for k, v := range fieldMap {
|
|
staff.Extra[k] = h.getFieldValue(v)
|
|
}
|
|
|
|
staff.UserName = userId
|
|
staff.RealName = userInfo.RealName
|
|
staff.Salary = cast.ToFloat64(h.getFieldValue(fieldMap["20001"]))
|
|
staff.Stock = cast.ToFloat64(h.getFieldValue(fieldMap["20002"]))
|
|
staff.Phone = cast.ToString(h.getFieldValue(fieldMap["17003"]))
|
|
staff.StaffType = cast.ToString(h.getFieldValue(fieldMap["12003"]))
|
|
staff.Idno = cast.ToString(h.getFieldValue(fieldMap["11015"]))
|
|
staff.BankName = cast.ToString(h.getFieldValue(fieldMap["13001"]))
|
|
staff.BankCard = cast.ToString(h.getFieldValue(fieldMap["13002"]))
|
|
staff.EntryDate = time.Unix(cast.ToInt64(h.getFieldValue(fieldMap["12018"])), 0).Format("2006-01-02")
|
|
staff.BirthDate = time.Unix(cast.ToInt64(h.getFieldValue(fieldMap["11005"])), 0).Format("2006-01-02")
|
|
staff.OfficialDate = time.Unix(cast.ToInt64(h.getFieldValue(fieldMap["12023"])), 0).Format("2006-01-02")
|
|
staff.AlipayUid = cast.ToString(h.getFieldValue(fieldMap["20004"]))
|
|
staff.PerfSalary = cast.ToFloat64(h.getFieldValue(fieldMap["20004"]))
|
|
|
|
//fmt.Println(goutil.EncodeJSON(staff))
|
|
return staff, nil
|
|
}
|
|
|
|
func (h *AppHr) getFieldValue(fieldInfo map[string]interface{}) string {
|
|
valueType := cast.ToInt(fieldInfo["value_type"])
|
|
if valueType == 1 {
|
|
return cast.ToString(fieldInfo["value_string"])
|
|
} else if valueType == 2 {
|
|
return cast.ToString(fieldInfo["value_uint64"])
|
|
} else if valueType == 3 {
|
|
return cast.ToString(fieldInfo["value_uint32"])
|
|
} else if valueType == 4 {
|
|
return cast.ToString(fieldInfo["value_int64"])
|
|
} else if valueType == 5 {
|
|
moble := cast.ToStringMap(fieldInfo["value_mobile"])
|
|
return cast.ToString(moble["value_mobile"])
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (h *AppHr) GetAllField() ([]byte, error) {
|
|
reqUrl := fmt.Sprintf("%s?access_token=%s", urlQyWeixinHrGetAllField, h.GetToken())
|
|
rspBody, err := util.HttpGet(reqUrl, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := h.GetResult(rspBody)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fmt.Println(goutil.EncodeJSONIndent(result))
|
|
return rspBody, err
|
|
}
|