| | |
| | | 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; |
| | |
| | | } |
| | | } |
| | | |
| | | 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; |
| | |
| | | @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; |
| | | } |
| | | } |