wlzboy
2 天以前 8cb5d3440208a3be3e772e65f1bd0ec63031ba62
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
package com.ruoyi.system.service.impl;
 
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.service.IQyWechatAccessTokenService;
import com.ruoyi.system.service.ISysConfigService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
/**
 * 企业微信AccessToken服务实现类
 * 
 * @author ruoyi
 * @date 2025-12-11
 */
@Service
public class QyWechatAccessTokenServiceImpl implements IQyWechatAccessTokenService {
 
    private static final Logger log = LoggerFactory.getLogger(QyWechatAccessTokenServiceImpl.class);
 
    @Autowired
    private ISysConfigService configService;
 
    /**
     * 企业微信获取access_token的URL
     */
    private static final String GET_ACCESS_TOKEN_URL = "https://qyapi.weixin.qq.com/cgi-bin/gettoken";
 
    /**
     * 获取企业微信应用的AccessToken
     *
     * @param corpId 企业ID
     * @param secret 应用密钥或小程序密钥
     * @return AccessToken
     */
    @Override
    public String getAppAccessToken(String corpId, String secret) {
        try {
            // 参数校验
            if (StringUtils.isEmpty(corpId) || StringUtils.isEmpty(secret)) {
                log.warn("企业微信配置参数不完整,corpId或secret为空");
                return null;
            }
 
            // 检查服务是否启用
            if (!isEnabled()) {
                log.info("企业微信服务已禁用,无法获取AccessToken");
                return null;
            }
 
            // 构建配置键名
            String tokenKey = "qy_wechat.access_token." + corpId;
            String expiresKey = "qy_wechat.access_token_expires." + corpId;
 
            // 从配置中获取Token和过期时间
            String accessToken = configService.selectConfigByKey(tokenKey);
            String expiresStr = configService.selectConfigByKey(expiresKey);
 
            // 检查Token是否存在且未过期
            if (StringUtils.isNotEmpty(accessToken) && StringUtils.isNotEmpty(expiresStr)) {
                try {
                    long expiresTime = Long.parseLong(expiresStr);
                    long currentTime = System.currentTimeMillis();
 
                    // 预留60秒安全边界,避免临界点过期
                    if (currentTime < expiresTime - 60000) {
                        log.debug("使用缓存的企业微信AccessToken,剩余有效时间: {}秒", 
                            (expiresTime - currentTime) / 1000);
                        return accessToken;
                    } else {
                        log.info("企业微信AccessToken已过期或即将过期,需要刷新");
                    }
                } catch (NumberFormatException e) {
                    log.warn("解析企业微信AccessToken过期时间失败: {}", expiresStr);
                }
            }
 
            // Token不存在或已过期,刷新Token
            return refreshAppAccessToken(corpId, secret);
        } catch (Exception e) {
            log.error("获取企业微信AccessToken失败", e);
            return null;
        }
    }
 
    /**
     * 刷新企业微信应用的AccessToken
     *
     * @param corpId 企业ID
     * @param secret 应用密钥或小程序密钥
     * @return 新的AccessToken
     */
    @Override
    public String refreshAppAccessToken(String corpId, String secret) {
        try {
            log.info("开始刷新企业微信AccessToken");
 
            // 构建请求URL
            String url = GET_ACCESS_TOKEN_URL + "?corpid=" + corpId + "&corpsecret=" + secret;
 
            // 发送HTTP请求获取Token
            String response = sendHttpGetRequest(url);
            
            if (StringUtils.isEmpty(response)) {
                log.error("获取企业微信AccessToken失败,响应为空");
                return null;
            }
 
            // 解析响应
            QyWechatTokenResponse tokenResponse = parseTokenResponse(response);
            
            if (tokenResponse == null || StringUtils.isEmpty(tokenResponse.getAccessToken())) {
                log.error("解析企业微信AccessToken响应失败: {}", response);
                return null;
            }
 
            // 计算过期时间(当前时间 + 有效期 - 60秒安全边界)
            long expiresTime = System.currentTimeMillis() + (tokenResponse.getExpiresIn() * 1000L) - 60000L;
 
            // 构建配置键名
            String tokenKey = "qy_wechat.access_token." + corpId;
            String expiresKey = "qy_wechat.access_token_expires." + corpId;
 
            // 保存到系统配置表
            configService.updateConfigValue(tokenKey, tokenResponse.getAccessToken());
            configService.updateConfigValue(expiresKey, String.valueOf(expiresTime));
 
            log.info("企业微信AccessToken刷新成功,有效期: {}秒", tokenResponse.getExpiresIn());
            return tokenResponse.getAccessToken();
        } catch (Exception e) {
            log.error("刷新企业微信AccessToken失败", e);
            return null;
        }
    }
 
    /**
     * 获取企业微信小程序的AccessToken
     *
     * @param corpId 企业ID
     * @param corpSecret 小程序密钥
     * @return AccessToken
     */
    @Override
    public String getQyMiniAccessToken(String corpId, String corpSecret) {
        try {
            // 参数校验
            if (StringUtils.isEmpty(corpId) || StringUtils.isEmpty(corpSecret)) {
                log.warn("企业微信小程序配置参数不完整,corpId或corpSecret为空");
                return null;
            }
 
            // 检查服务是否启用
            if (!isEnabled()) {
                log.info("企业微信服务已禁用,无法获取小程序AccessToken");
                return null;
            }
 
            // 构建配置键名(使用不同的键名以区分普通应用和小程序)
            String tokenKey = "qy_wechat.mini_access_token." + corpId;
            String expiresKey = "qy_wechat.mini_access_token_expires." + corpId;
 
            // 从配置中获取Token和过期时间
            String accessToken = configService.selectConfigByKey(tokenKey);
            String expiresStr = configService.selectConfigByKey(expiresKey);
 
            // 检查Token是否存在且未过期
            if (StringUtils.isNotEmpty(accessToken) && StringUtils.isNotEmpty(expiresStr)) {
                try {
                    long expiresTime = Long.parseLong(expiresStr);
                    long currentTime = System.currentTimeMillis();
 
                    // 预留60秒安全边界,避免临界点过期
                    if (currentTime < expiresTime - 60000) {
                        log.debug("使用缓存的企业微信小程序AccessToken,剩余有效时间: {}秒", 
                            (expiresTime - currentTime) / 1000);
                        return accessToken;
                    } else {
                        log.info("企业微信小程序AccessToken已过期或即将过期,需要刷新");
                    }
                } catch (NumberFormatException e) {
                    log.warn("解析企业微信小程序AccessToken过期时间失败: {}", expiresStr);
                }
            }
 
            // Token不存在或已过期,刷新Token
            return refreshQyMiniAccessToken(corpId, corpSecret);
        } catch (Exception e) {
            log.error("获取企业微信小程序AccessToken失败", e);
            return null;
        }
    }
 
    /**
     * 刷新企业微信小程序的AccessToken
     *
     * @param corpId 企业ID
     * @param corpSecret 小程序密钥
     * @return 新的AccessToken
     */
    public String refreshQyMiniAccessToken(String corpId, String corpSecret) {
        try {
            log.info("开始刷新企业微信小程序AccessToken");
 
            // 构建请求URL
            String url = GET_ACCESS_TOKEN_URL + "?corpid=" + corpId + "&corpsecret=" + corpSecret;
 
            // 发送HTTP请求获取Token
            String response = sendHttpGetRequest(url);
            
            if (StringUtils.isEmpty(response)) {
                log.error("获取企业微信小程序AccessToken失败,响应为空");
                return null;
            }
 
            // 解析响应
            QyWechatTokenResponse tokenResponse = parseTokenResponse(response);
            
            if (tokenResponse == null || StringUtils.isEmpty(tokenResponse.getAccessToken())) {
                log.error("解析企业微信小程序AccessToken响应失败: {}", response);
                return null;
            }
 
            // 计算过期时间(当前时间 + 有效期 - 60秒安全边界)
            long expiresTime = System.currentTimeMillis() + (tokenResponse.getExpiresIn() * 1000L) - 60000L;
 
            // 构建配置键名(使用不同的键名以区分普通应用和小程序)
            String tokenKey = "qy_wechat.mini_access_token." + corpId;
            String expiresKey = "qy_wechat.mini_access_token_expires." + corpId;
 
            // 保存到系统配置表
            configService.updateConfigValue(tokenKey, tokenResponse.getAccessToken());
            configService.updateConfigValue(expiresKey, String.valueOf(expiresTime));
 
            log.info("企业微信小程序AccessToken刷新成功,有效期: {}秒", tokenResponse.getExpiresIn());
            return tokenResponse.getAccessToken();
        } catch (Exception e) {
            log.error("刷新企业微信小程序AccessToken失败", e);
            return null;
        }
    }
 
    /**
     * 检查企业微信服务是否启用
     *
     * @return true-启用,false-禁用
     */
    @Override
    public boolean isEnabled() {
        try {
            String enabled = configService.selectConfigByKey("qy_wechat.enable");
            return !"false".equals(enabled); // 默认启用
        } catch (Exception e) {
            log.warn("获取企业微信服务启用状态失败,使用默认值(true)", e);
            return true;
        }
    }
 
    /**
     * 发送HTTP GET请求
     * 
     * @param url 请求URL
     * @return 响应内容
     */
    private String sendHttpGetRequest(String url) {
        try {
            java.net.HttpURLConnection conn = (java.net.HttpURLConnection) new java.net.URL(url).openConnection();
            conn.setRequestMethod("GET");
            conn.setConnectTimeout(5000);
            conn.setReadTimeout(5000);
 
            int responseCode = conn.getResponseCode();
            if (responseCode == 200) {
                java.io.BufferedReader reader = new java.io.BufferedReader(
                    new java.io.InputStreamReader(conn.getInputStream(), "UTF-8"));
                StringBuilder response = new StringBuilder();
                String line;
                while ((line = reader.readLine()) != null) {
                    response.append(line);
                }
                reader.close();
                return response.toString();
            } else {
                log.error("HTTP请求失败,响应码: {}", responseCode);
                return null;
            }
        } catch (Exception e) {
            log.error("发送HTTP请求失败", e);
            return null;
        }
    }
 
    /**
     * 解析企业微信Token响应
     * 
     * @param response 响应JSON
     * @return Token响应对象
     */
    private QyWechatTokenResponse parseTokenResponse(String response) {
        try {
            // 简单JSON解析(实际项目中建议使用Jackson或Gson)
            // 示例响应: {"errcode":0,"errmsg":"ok","access_token":"xxxxxx","expires_in":7200}
            
            // 移除花括号
            String content = response.substring(1, response.length() - 1);
            
            // 按逗号分割
            String[] pairs = content.split(",");
            
            QyWechatTokenResponse tokenResponse = new QyWechatTokenResponse();
            
            for (String pair : pairs) {
                String[] keyValue = pair.split(":");
                if (keyValue.length == 2) {
                    String key = keyValue[0].trim().replaceAll("\"", "");
                    String value = keyValue[1].trim().replaceAll("\"", "");
                    
                    switch (key) {
                        case "errcode":
                            tokenResponse.setErrcode(Integer.parseInt(value));
                            break;
                        case "errmsg":
                            tokenResponse.setErrmsg(value);
                            break;
                        case "access_token":
                            tokenResponse.setAccessToken(value);
                            break;
                        case "expires_in":
                            tokenResponse.setExpiresIn(Integer.parseInt(value));
                            break;
                    }
                }
            }
            
            // 检查是否有错误
            if (tokenResponse.getErrcode() != 0) {
                log.error("获取企业微信AccessToken失败,错误码: {}, 错误信息: {}", 
                    tokenResponse.getErrcode(), tokenResponse.getErrmsg());
                return null;
            }
            
            return tokenResponse;
        } catch (Exception e) {
            log.error("解析企业微信Token响应失败: {}", response, e);
            return null;
        }
    }
 
    /**
     * 企业微信Token响应内部类
     */
    private static class QyWechatTokenResponse {
        private int errcode;
        private String errmsg;
        private String accessToken;
        private int expiresIn;
 
        // Getters and Setters
        public int getErrcode() {
            return errcode;
        }
 
        public void setErrcode(int errcode) {
            this.errcode = errcode;
        }
 
        public String getErrmsg() {
            return errmsg;
        }
 
        public void setErrmsg(String errmsg) {
            this.errmsg = errmsg;
        }
 
        public String getAccessToken() {
            return accessToken;
        }
 
        public void setAccessToken(String accessToken) {
            this.accessToken = accessToken;
        }
 
        public int getExpiresIn() {
            return expiresIn;
        }
 
        public void setExpiresIn(int expiresIn) {
            this.expiresIn = expiresIn;
        }
    }
}