前言

最近小组需要统计部门各种请假次数，写了一个方法，第一次实战中用到递归函数，简单记录一下。

效果展示

这些数据不需要返回json，这里这样是为了方便测试。可以通过这些数据完成其它的操作。

功能实现

钉钉服务端调试工具API Explorer

感兴趣的可以在接口里面详细看一下参数，这里就不一 一解释了

定义model

请求参数

package modeltype RequestDingLeave struct {UseridList string `json:"userid_list"` // 待查询用户的ID列表，每次最多100个。StartTime  int    `json:"start_time"`  // 开始时间 ，Unix时间戳，支持最多180天的查询。EndTime    int    `json:"end_time"`    // 结束时间，Unix时间戳，支持最多180天的查询。Offset     int    `json:"offset"`      // 支持分页查询，与size参数同时设置时才生效，此参数代表偏移量，偏移量从0开始。Size       int    `json:"size"`        // 支持分页查询，与offset参数同时设置时才生效，此参数代表分页大小，最大20。
}

响应参数

package model// DingLeaveStatus 请假状态
type DingLeaveStatus struct {StartTime       int64  `json:"start_time"`DurationPercent int    `json:"duration_percent"`EndTime         int64  `json:"end_time"`LeaveCode       string `json:"leave_code"` // 请假类型 个人事假：d4edf257-e581-45f9-b9b9-35755b598952  非个人事假：baf811bc-3daa-4988-9604-d68ec1edaf50  病假：a7ffa2e6-872a-498d-aca7-4554c56fbb52DurationUnit    string `json:"duration_unit"`UserID          string `json:"userid"`
}// DingLeaveResult 请假列表
type DingLeaveResult struct {LeaveStatus *[]DingLeaveStatus `json:"leave_status"`HasMore     bool               `json:"has_more"` // 是否有更多数据
}// DingResponse 钉钉响应
type DingResponse struct {ErrCode   int             `json:"errcode"`Result    DingLeaveResult `json:"result"`Success   bool            `json:"success"`ErrMsg    string          `json:"errmsg"`RequestID string          `json:"request_id"`
}type DingUser struct {Id   string `json:"id"`Name string `json:"name"`Type map[string]int
}

功能逻辑

我这里只是想简单的实现一下功能，并没有按照严格标准的mvc分层，还请见谅

package controllerimport ("bytes""encoding/json""fmt""github.com/gin-gonic/gin""leave/model""net/http""time"
)var tmp = map[string]string{"d4edf257-e581-45f9-b9b9-35755b598952": "个人事假", "a7ffa2e6-872a-498d-aca7-4554c56fbb52": "病假", "baf811bc-3daa-4988-9604-d68ec1edaf50": "非个人事假"} // code与请假类型对应// GetDepartmentLeaveStatus 统计部门请假状态
func GetDepartmentLeaveStatus(c *gin.Context) {userList := "323832232226246897,01320463195324064909,02280849645326263552,01546916272226288917"	// 模拟数据leave := model.RequestDingLeave{UseridList: userList,StartTime:  1698624000000,	EndTime:    int(time.Now().UnixMilli()), // 拿到当前的时间戳Offset:     0,Size:       2,}res := []model.DingLeaveStatus{}res, err := GetLeaveStatus(leave, res)if err != nil {return}statistics := map[string]model.DingUser{}for _, v := range res {_, exist := statistics[v.UserID]if !exist {statistics[v.UserID] = model.DingUser{Type: map[string]int{}}}statistics[v.UserID].Type[tmp[v.LeaveCode]]++}c.JSON(http.StatusOK, statistics)fmt.Println(int(time.Now().Weekday()))return
}// GetLeaveStatus 获取请假状态
func GetLeaveStatus(leave model.RequestDingLeave, res []model.DingLeaveStatus) ([]model.DingLeaveStatus, error) {// 将json数据编码为字节数组var send func(leave model.RequestDingLeave) error// 切片作为函数参数传递时为值传递，如果每次传入新的切片需要增加返回值接收，这里直接使用的传入的res作为变量并返回send = func(leave model.RequestDingLeave) error {jsonLeave, err := json.Marshal(leave)if err != nil {fmt.Println("json.Marshal(leave) failed:", err)return err}url := fmt.Sprintf("https://oapi.dingtalk.com/topapi/attendance/getleavestatus?%s", "access_token=企业token") // access_token 填企业tokenbuffer := bytes.NewBuffer(jsonLeave)response, err := http.Post(url, "application/json", buffer)if err != nil {fmt.Println("http.Post(\"https://oapi.dingtalk.com/topapi/attendance/getleavestatus\", \"application/json\", buffer) failed:", err)return err}var dingResp model.DingResponseerr = json.NewDecoder(response.Body).Decode(&dingResp)	// 将响应的json绑定到dingResp中if err != nil {return err}res = append(res, *dingResp.Result.LeaveStatus...)if dingResp.Result.HasMore {	// 如果有更多的记录，再次请求并添加到切片后面leave.Offset += leave.Sizesend(leave)					// 递归}return nil}err := send(leave)return res, err
}

其他

package routerimport ("github.com/gin-gonic/gin""leave/controller"
)func SetupRouter() *gin.Engine {r := gin.Default()r.GET("/test", controller.GetDepartmentLeaveStatus)return r
}

package mainimport "leave/router"func main() {router := router.SetupRouter()router.Run()
}