新增企业微信的接口

tjx

2025-11-25 2e7cf0157390b2b89a4657a6a78c7147afc09ea8

新增企业微信的接口

 src/main/java/com/gs/xky/service/WorkWXService.java

@@ -11,6 +11,7 @@
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -133,6 +134,77 @@
        }
    }

    public List<CheckinData> getCheckinData(long startTime, long endTime, List<String> useridList) throws IOException {
        String accessToken = getAccessToken();
        String url = String.format("https://qyapi.weixin.qq.com/cgi-bin/checkin/getcheckindata?access_token=%s", accessToken);

        List<CheckinData> allCheckinData = new ArrayList<>();

        int batchSize = 100;
        int totalUsers = useridList.size();
        int batchCount = (totalUsers + batchSize - 1) / batchSize;

        log.info("开始获取打卡数据，总用户数: {}, 分批数: {}, 时间范围: {} - {}", totalUsers, batchCount, startTime, endTime);

        for (int i = 0; i < batchCount; i++) {
            int fromIndex = i * batchSize;
            int toIndex = Math.min((i + 1) * batchSize, totalUsers);

            List<String> batchUserList = useridList.subList(fromIndex, toIndex);
            log.info("正在获取第 {}/{} 批打卡数据，用户数: {}", i + 1, batchCount, batchUserList.size());

            Map<String, Object> requestBody = new HashMap<>();
            requestBody.put("opencheckindatatype", 3);
            requestBody.put("starttime", startTime);
            requestBody.put("endtime", endTime);
            requestBody.put("useridlist", batchUserList);

            MediaType mediaType = MediaType.parse("application/json; charset=UTF-8");
            String jsonBody = objectMapper.writeValueAsString(requestBody);
            RequestBody body = RequestBody.create(mediaType, jsonBody);

            Request request = new Request.Builder()
                    .url(url)
                    .post(body)
                    .addHeader("Content-Type", "application/json; charset=UTF-8")
                    .build();

            try (Response response = client.newCall(request).execute()) {
                if (!response.isSuccessful()) {
                    log.error("获取打卡数据失败，HTTP状态码: {}", response.code());
                    throw new IOException("获取打卡数据失败: " + response.message());
                }

                String responseBody = response.body().string();

                WorkWXCheckinResponse checkinResponse = objectMapper.readValue(responseBody, WorkWXCheckinResponse.class);

                if (checkinResponse.getErrcode() != 0) {
                    log.error("获取打卡数据失败，错误码: {}, 错误信息: {}",
                            checkinResponse.getErrcode(), checkinResponse.getErrmsg());
                    throw new IOException("获取打卡数据失败: " + checkinResponse.getErrmsg());
                }

                if (checkinResponse.getCheckindata() != null) {
                    allCheckinData.addAll(checkinResponse.getCheckindata());
                    log.info("第 {}/{} 批获取到打卡记录数: {}", i + 1, batchCount, checkinResponse.getCheckindata().size());
                }
            }

            if (i < batchCount - 1) {
                try {
                    Thread.sleep(500);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                    log.warn("批次间等待被中断");
                }
            }
        }

        log.info("打卡数据获取完成，总记录数: {}", allCheckinData.size());
        return allCheckinData;
    }

    @Data
    private static class WorkWXTokenResponse {
        private Integer errcode;
@@ -165,4 +237,33 @@
        @JsonProperty("open_userid")
        private String openUserid;
    }

    @Data
    private static class WorkWXCheckinResponse {
        private Integer errcode;
        private String errmsg;
        private List<CheckinData> checkindata;
    }

    @Data
    public static class CheckinData {
        private String userid;
        private String groupname;
        @JsonProperty("checkin_type")
        private String checkinType;
        @JsonProperty("exception_type")
        private String exceptionType;
        @JsonProperty("checkin_time")
        private Long checkinTime;
        @JsonProperty("location_title")
        private String locationTitle;
        @JsonProperty("location_detail")
        private String locationDetail;
        private String wifiname;
        private String notes;
        private String wifimac;
        private String mediaids;
        private Double lat;
        private Double lng;
    }
}

 src/test/java/com/gs/xky/XkyApplicationTests.java

@@ -7,6 +7,7 @@
import org.springframework.boot.test.context.SpringBootTest;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

@SpringBootTest
@@ -129,4 +130,53 @@

        System.out.println("=== 测试结束 ===");
    }

    /**
     * 测试获取企业微信打卡数据
     * 功能：获取指定时间范围内的员工打卡记录
     */
    @Test
    void testGetCheckinData() {
        System.out.println("=== 开始测试获取企业微信打卡数据 ===");

        try {
            List<WorkWXService.WorkWXUser> userList = workWXService.getUserList();

            if (userList == null || userList.isEmpty()) {
                System.out.println("✗ 未获取到用户列表，无法继续测试");
                return;
            }

            List<String> useridList = new ArrayList<>();
            userList.forEach(user -> useridList.add(user.getUserid()));

            long endTime = System.currentTimeMillis() / 1000;
            long startTime = endTime - 86400;

            System.out.println("  - 用户总数: " + useridList.size());
            System.out.println("  - 开始时间: " + new java.util.Date(startTime * 1000));
            System.out.println("  - 结束时间: " + new java.util.Date(endTime * 1000));

            List<WorkWXService.CheckinData> checkinDataList = workWXService.getCheckinData(startTime, endTime, useridList);

            if (checkinDataList != null && !checkinDataList.isEmpty()) {
                System.out.println("✓ 成功获取打卡数据");
                System.out.println("  - 打卡记录总数: " + checkinDataList.size());
                System.out.println("  - 前3条数据:");
                checkinDataList.stream().limit(3).forEach(data -> {
                    System.out.println("    * userid: " + data.getUserid() +
                            ", 打卡时间: " + new java.util.Date(data.getCheckinTime() * 1000) +
                            ", 打卡类型: " + data.getCheckinType() +
                            ", 地点: " + data.getLocationDetail());
                });
            } else {
                System.out.println("✗ 获取的打卡数据为空");
            }
        } catch (IOException e) {
            System.out.println("✗ 获取打卡数据失败: " + e.getMessage());
            e.printStackTrace();
        }

        System.out.println("=== 测试结束 ===");
    }
}