|
@@ -71,6 +71,20 @@ func (hc *HomeClient) ReadMessage() {
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: pong}
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: pong}
|
|
|
|
|
|
hc.send <- sendData
|
|
hc.send <- sendData
|
|
|
|
+ case wsLeaveReq:
|
|
|
|
+ // 解析数据
|
|
|
|
+ var leaveReq WSLeaveReq
|
|
|
|
+ err = json.Unmarshal([]byte(receiveData.Data), &leaveReq)
|
|
|
|
+ if err != nil {
|
|
|
|
+ log.Error(code.ErrCodeMap[code.ErrTokenInvalidCode].Error())
|
|
|
|
+ continue
|
|
|
|
+ }
|
|
|
|
+ // 发送请假详情
|
|
|
|
+ err = sendLeaveInfo(leaveReq.StaffID)
|
|
|
|
+ if err != nil {
|
|
|
|
+ log.Error(err.Error())
|
|
|
|
+ continue
|
|
|
|
+ }
|
|
case clientToken:
|
|
case clientToken:
|
|
// 解析客户端发来的token
|
|
// 解析客户端发来的token
|
|
claims, err := global.JWT.ParseToken(receiveData.Data)
|
|
claims, err := global.JWT.ParseToken(receiveData.Data)
|
|
@@ -79,19 +93,19 @@ func (hc *HomeClient) ReadMessage() {
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: code.ErrCodeMap[code.ErrTokenExpiredCode].Error()}
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: code.ErrCodeMap[code.ErrTokenExpiredCode].Error()}
|
|
hc.send <- sendData
|
|
hc.send <- sendData
|
|
log.Error(code.ErrCodeMap[code.ErrTokenExpiredCode].Error())
|
|
log.Error(code.ErrCodeMap[code.ErrTokenExpiredCode].Error())
|
|
- return
|
|
|
|
|
|
+ continue
|
|
}
|
|
}
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: code.ErrCodeMap[code.ErrTokenInvalidCode].Error()}
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: code.ErrCodeMap[code.ErrTokenInvalidCode].Error()}
|
|
hc.send <- sendData
|
|
hc.send <- sendData
|
|
log.Error(code.ErrCodeMap[code.ErrTokenInvalidCode].Error())
|
|
log.Error(code.ErrCodeMap[code.ErrTokenInvalidCode].Error())
|
|
- return
|
|
|
|
|
|
+ continue
|
|
}
|
|
}
|
|
deptIds, err := systemService.ServicesGroupApp.GetSonDeptIds(claims.DeptID)
|
|
deptIds, err := systemService.ServicesGroupApp.GetSonDeptIds(claims.DeptID)
|
|
if err != nil {
|
|
if err != nil {
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: err.Error()}
|
|
sendData := &HomeMsg{DeptID: 0, MsgType: clientToken, Data: err.Error()}
|
|
hc.send <- sendData
|
|
hc.send <- sendData
|
|
log.Error(err.Error())
|
|
log.Error(err.Error())
|
|
- return
|
|
|
|
|
|
+ continue
|
|
}
|
|
}
|
|
deptMap := make(map[int]struct{})
|
|
deptMap := make(map[int]struct{})
|
|
for _, dep := range deptIds {
|
|
for _, dep := range deptIds {
|