From cec65038e8b9b80dca71b017ef18d0f027b47891 Mon Sep 17 00:00:00 2001
From: tjx <t2856754968@163.com>
Date: 星期三, 26 十一月 2025 13:46:03 +0800
Subject: [PATCH] 新增企业微信的接口
---
src/main/java/com/gs/xky/service/WorkWXService.java | 167 ++++++++++++++++++++++++++++++++++++-------------------
1 files changed, 108 insertions(+), 59 deletions(-)
diff --git a/src/main/java/com/gs/xky/service/WorkWXService.java b/src/main/java/com/gs/xky/service/WorkWXService.java
index 44996b2..300521d 100644
--- a/src/main/java/com/gs/xky/service/WorkWXService.java
+++ b/src/main/java/com/gs/xky/service/WorkWXService.java
@@ -1,8 +1,11 @@
package com.gs.xky.service;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.gs.xky.config.DataAcquisitionConfiguration;
+import com.gs.xky.entity.QwStaff;
+import com.gs.xky.mapper.QwStaffMapper;
import lombok.Data;
import lombok.RequiredArgsConstructor;
import okhttp3.*;
@@ -16,6 +19,7 @@
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
@Service
@RequiredArgsConstructor
@@ -29,6 +33,8 @@
.build();
private final ObjectMapper objectMapper = new ObjectMapper();
+
+ private final QwStaffMapper qwStaffMapper;
public String getAccessToken() throws IOException {
String url = String.format("https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s",
@@ -61,50 +67,10 @@
}
}
- public String getUserIdByMobile(String mobile) throws IOException {
- String accessToken = getAccessToken();
- String url = String.format("https://qyapi.weixin.qq.com/cgi-bin/user/getuserid?access_token=%s", accessToken);
-
- Map<String, Object> requestBody = new HashMap<>();
- requestBody.put("mobile", mobile);
-
- 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("閫氳繃鎵嬫満鍙疯幏鍙杣serid澶辫触锛孒TTP鐘舵�佺爜: {}", response.code());
- throw new IOException("鑾峰彇userid澶辫触: " + response.message());
- }
-
- String responseBody = response.body().string();
- log.info("閫氳繃鎵嬫満鍙疯幏鍙杣serid鍝嶅簲: {}", responseBody);
-
- WorkWXUserIdResponse userIdResponse = objectMapper.readValue(responseBody, WorkWXUserIdResponse.class);
-
- if (userIdResponse.getErrcode() != 0) {
- log.error("閫氳繃鎵嬫満鍙疯幏鍙杣serid澶辫触锛岄敊璇爜: {}, 閿欒淇℃伅: {}",
- userIdResponse.getErrcode(), userIdResponse.getErrmsg());
- throw new IOException("鑾峰彇userid澶辫触: " + userIdResponse.getErrmsg());
- }
-
- log.info("鎴愬姛閫氳繃鎵嬫満鍙� {} 鑾峰彇userid: {}", mobile, userIdResponse.getUserid());
- return userIdResponse.getUserid();
- }
- }
-
- public List<WorkWXUser> getUserList() throws IOException {
- String accessToken = getAccessToken();
- String url = String.format(
- "https://qyapi.weixin.qq.com/cgi-bin/user/simplelist?access_token=%s&department_id=1&fetch_child=1",
- accessToken);
+ public String getContactAccessToken() throws IOException {
+ String url = String.format("https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s",
+ DataAcquisitionConfiguration.CORPID,
+ DataAcquisitionConfiguration.TXL_CORPSECRET);
Request request = new Request.Builder()
.url(url)
@@ -113,25 +79,105 @@
try (Response response = client.newCall(request).execute()) {
if (!response.isSuccessful()) {
- log.error("鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃澶辫触锛孒TTP鐘舵�佺爜: {}", response.code());
- throw new IOException("鑾峰彇鐢ㄦ埛鍒楄〃澶辫触: " + response.message());
+ log.error("鑾峰彇浼佷笟寰俊閫氳褰昦ccess_token澶辫触锛孒TTP鐘舵�佺爜: {}", response.code());
+ throw new IOException("鑾峰彇閫氳褰昦ccess_token澶辫触: " + response.message());
}
String responseBody = response.body().string();
- log.info("鑾峰彇鐢ㄦ埛鍒楄〃鍝嶅簲: {}", responseBody);
+ WorkWXTokenResponse tokenResponse = objectMapper.readValue(responseBody, WorkWXTokenResponse.class);
- WorkWXUserListResponse userListResponse = objectMapper.readValue(responseBody, WorkWXUserListResponse.class);
-
- if (userListResponse.getErrcode() != 0) {
- log.error("鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃澶辫触锛岄敊璇爜: {}, 閿欒淇℃伅: {}",
- userListResponse.getErrcode(), userListResponse.getErrmsg());
- throw new IOException("鑾峰彇鐢ㄦ埛鍒楄〃澶辫触: " + userListResponse.getErrmsg());
+ if (tokenResponse.getErrcode() != 0) {
+ log.error("鑾峰彇浼佷笟寰俊閫氳褰昦ccess_token澶辫触锛岄敊璇爜: {}, 閿欒淇℃伅: {}",
+ tokenResponse.getErrcode(), tokenResponse.getErrmsg());
+ throw new IOException("鑾峰彇閫氳褰昦ccess_token澶辫触: " + tokenResponse.getErrmsg());
}
- log.info("鎴愬姛鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃锛岀敤鎴锋暟閲�: {}",
- userListResponse.getUserlist() != null ? userListResponse.getUserlist().size() : 0);
- return userListResponse.getUserlist();
+ log.info("鎴愬姛鑾峰彇浼佷笟寰俊閫氳褰昦ccess_token锛屾湁鏁堟湡: {}绉�", tokenResponse.getExpiresIn());
+ return tokenResponse.getAccessToken();
}
+ }
+
+
+ public List<WorkWXUser> getUserList() throws IOException {
+ String accessToken = getContactAccessToken();
+ String url = String.format(
+ "https://qyapi.weixin.qq.com/cgi-bin/user/list_id?access_token=%s",
+ accessToken);
+
+ List<WorkWXUser> allUsers = new ArrayList<>();
+ String cursor = null;
+
+ do {
+ Map<String, Object> requestBody = new HashMap<>();
+ requestBody.put("limit", 10000);
+ if (cursor != null) {
+ requestBody.put("cursor", cursor);
+ }
+
+ 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("鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃澶辫触锛孒TTP鐘舵�佺爜: {}", response.code());
+ throw new IOException("鑾峰彇鐢ㄦ埛鍒楄〃澶辫触: " + response.message());
+ }
+
+ String responseBody = response.body().string();
+ log.info("鑾峰彇鐢ㄦ埛鍒楄〃鍝嶅簲: {}", responseBody);
+
+ WorkWXUserListIdResponse userListResponse = objectMapper.readValue(responseBody, WorkWXUserListIdResponse.class);
+
+ if (userListResponse.getErrcode() != 0) {
+ log.error("鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃澶辫触锛岄敊璇爜: {}, 閿欒淇℃伅: {}",
+ userListResponse.getErrcode(), userListResponse.getErrmsg());
+ throw new IOException("鑾峰彇鐢ㄦ埛鍒楄〃澶辫触: " + userListResponse.getErrmsg());
+ }
+
+ if (userListResponse.getDeptUser() != null && !userListResponse.getDeptUser().isEmpty()) {
+ allUsers.addAll(userListResponse.getDeptUser());
+ }
+
+ cursor = userListResponse.getNextCursor();
+
+ } catch (IOException e) {
+ log.error("瑙ｆ瀽鐢ㄦ埛鍒楄〃鍝嶅簲澶辫触", e);
+ throw e;
+ }
+
+ } while (cursor != null && !cursor.isEmpty());
+
+ log.info("鎴愬姛鑾峰彇浼佷笟寰俊鐢ㄦ埛鍒楄〃锛岀敤鎴锋暟閲�: {}", allUsers.size());
+ return allUsers;
+ }
+
+ public List<CheckinData> getCheckinDataByQwStaff(long startTime, long endTime) throws IOException {
+ List<QwStaff> qwStaffList = qwStaffMapper.selectList(new LambdaQueryWrapper<QwStaff>());
+
+ if (qwStaffList == null || qwStaffList.isEmpty()) {
+ log.warn("QW_STAFF琛ㄤ腑娌℃湁鏁版嵁");
+ return new ArrayList<>();
+ }
+
+ List<String> useridList = qwStaffList.stream()
+ .map(QwStaff::getAccount)
+ .filter(account -> account != null && !account.isEmpty())
+ .collect(Collectors.toList());
+
+ if (useridList.isEmpty()) {
+ log.warn("QW_STAFF琛ㄤ腑娌℃湁鏈夋晥鐨刟ccount鏁版嵁");
+ return new ArrayList<>();
+ }
+
+ log.info("浠嶲W_STAFF琛ㄨ幏鍙栧埌 {} 涓敤鎴穉ccount", useridList.size());
+ return getCheckinData(startTime, endTime, useridList);
}
public List<CheckinData> getCheckinData(long startTime, long endTime, List<String> useridList) throws IOException {
@@ -223,17 +269,20 @@
}
@Data
- private static class WorkWXUserListResponse {
+ private static class WorkWXUserListIdResponse {
private Integer errcode;
private String errmsg;
- private List<WorkWXUser> userlist;
+ @JsonProperty("next_cursor")
+ private String nextCursor;
+ @JsonProperty("dept_user")
+ private List<WorkWXUser> deptUser;
}
@Data
public static class WorkWXUser {
private String userid;
- private String name;
private List<Integer> department;
+ private String name;
@JsonProperty("open_userid")
private String openUserid;
}
--
Gitblit v1.9.3