16 changed files with 265 additions and 140 deletions
@ -0,0 +1,110 @@ |
|||||
|
package com.gxwebsoft; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.JsonNode; |
||||
|
import com.fasterxml.jackson.databind.ObjectMapper; |
||||
|
import okhttp3.*; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.data.redis.core.StringRedisTemplate; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.time.Instant; |
||||
|
import java.util.concurrent.atomic.AtomicReference; |
||||
|
|
||||
|
@Service |
||||
|
public class WxDev { |
||||
|
|
||||
|
@Value("${wechat.appid}") |
||||
|
private String appId; |
||||
|
@Value("${wechat.secret}") |
||||
|
private String secret; |
||||
|
|
||||
|
private final StringRedisTemplate redisTemplate; |
||||
|
|
||||
|
private final OkHttpClient http = new OkHttpClient(); |
||||
|
private final ObjectMapper om = new ObjectMapper(); |
||||
|
|
||||
|
/** 简单本地缓存 access_token（生产建议放 Redis） */ |
||||
|
private final AtomicReference<String> cachedToken = new AtomicReference<>(); |
||||
|
private volatile long tokenExpireEpoch = 0L; // 过期的 epoch 秒
|
||||
|
|
||||
|
public WxDev(StringRedisTemplate redisTemplate) { |
||||
|
this.redisTemplate = redisTemplate; |
||||
|
} |
||||
|
|
||||
|
/** 获取/刷新 access_token */ |
||||
|
public String getAccessToken() throws IOException { |
||||
|
long now = Instant.now().getEpochSecond(); |
||||
|
System.out.println("cachedToken.get = " + cachedToken.get()); |
||||
|
if (cachedToken.get() != null && now < tokenExpireEpoch - 60) { |
||||
|
return cachedToken.get(); |
||||
|
} |
||||
|
HttpUrl url = HttpUrl.parse("https://api.weixin.qq.com/cgi-bin/token") |
||||
|
.newBuilder() |
||||
|
.addQueryParameter("grant_type", "client_credential") |
||||
|
.addQueryParameter("appid", "wx51962d6ac21f2ed2") |
||||
|
.addQueryParameter("secret", "d821f98de8a6c1ba7bc7e0ee84bcbc8e") |
||||
|
.build(); |
||||
|
Request req = new Request.Builder().url(url).get().build(); |
||||
|
try (Response resp = http.newCall(req).execute()) { |
||||
|
String body = resp.body().string(); |
||||
|
JsonNode json = om.readTree(body); |
||||
|
if (json.has("access_token")) { |
||||
|
String token = json.get("access_token").asText(); |
||||
|
int expiresIn = json.get("expires_in").asInt(7200); |
||||
|
System.out.println("token1 = " + token); |
||||
|
cachedToken.set(token); |
||||
|
tokenExpireEpoch = now + expiresIn; |
||||
|
return token; |
||||
|
} else { |
||||
|
throw new IOException("Get access_token failed: " + body); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** 调用 getwxacodeunlimit，返回图片二进制 */ |
||||
|
public byte[] getUnlimitedCode(String scene, String page, Integer width, |
||||
|
Boolean isHyaline, String envVersion) throws IOException { |
||||
|
String accessToken = getAccessToken(); |
||||
|
System.out.println("accessToken = " + accessToken); |
||||
|
HttpUrl url = HttpUrl.parse("https://api.weixin.qq.com/wxa/getwxacodeunlimit") |
||||
|
.newBuilder() |
||||
|
.addQueryParameter("access_token", accessToken) |
||||
|
.build(); |
||||
|
|
||||
|
// 构造请求 JSON
|
||||
|
// 注意：scene 仅支持可见字符，长度上限 32，尽量 URL-safe（字母数字下划线等）
|
||||
|
// page 必须是已发布小程序内的路径（不带开头斜杠也可）
|
||||
|
var root = om.createObjectNode(); |
||||
|
root.put("scene", scene); |
||||
|
if (page != null) root.put("page", page); |
||||
|
if (width != null) root.put("width", width); // 默认 430，建议 280~1280
|
||||
|
if (isHyaline != null) root.put("is_hyaline", isHyaline); |
||||
|
if (envVersion != null) root.put("env_version", envVersion); // release/trial/develop
|
||||
|
|
||||
|
RequestBody reqBody = RequestBody.create( |
||||
|
root.toString(), MediaType.parse("application/json; charset=utf-8")); |
||||
|
Request req = new Request.Builder().url(url).post(reqBody).build(); |
||||
|
|
||||
|
try (Response resp = http.newCall(req).execute()) { |
||||
|
if (!resp.isSuccessful()) { |
||||
|
throw new IOException("HTTP " + resp.code() + " calling getwxacodeunlimit"); |
||||
|
} |
||||
|
MediaType ct = resp.body().contentType(); |
||||
|
byte[] bytes = resp.body().bytes(); |
||||
|
// 微信出错时返回 JSON，需要识别一下
|
||||
|
if (ct != null && ct.subtype() != null && ct.subtype().contains("json")) { |
||||
|
String err = new String(bytes); |
||||
|
throw new IOException("WeChat error: " + err); |
||||
|
} |
||||
|
return bytes; // 成功就是图片二进制（PNG）
|
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void getQrCode() throws IOException { |
||||
|
final byte[] test = getUnlimitedCode("register", "pages/index/index",180,false,"develop"); |
||||
|
System.out.println("test = " + test); |
||||
|
} |
||||
|
} |
||||
@ -1,52 +0,0 @@ |
|||||
package com.gxwebsoft.hjm; |
|
||||
|
|
||||
import com.gxwebsoft.hjm.service.MqttService; |
|
||||
import org.junit.jupiter.api.Test; |
|
||||
import org.springframework.boot.test.context.SpringBootTest; |
|
||||
import org.springframework.test.context.ActiveProfiles; |
|
||||
|
|
||||
import javax.annotation.Resource; |
|
||||
|
|
||||
/** |
|
||||
* MQTT服务测试类 |
|
||||
* |
|
||||
* @author 科技小王子 |
|
||||
* @since 2025-07-02 |
|
||||
*/ |
|
||||
@SpringBootTest |
|
||||
@ActiveProfiles("dev") |
|
||||
public class MqttServiceTest { |
|
||||
|
|
||||
@Resource |
|
||||
private MqttService mqttService; |
|
||||
|
|
||||
@Test |
|
||||
public void testMqttConnection() { |
|
||||
System.out.println("MQTT连接状态: " + mqttService.isConnected()); |
|
||||
System.out.println("MQTT客户端信息: " + mqttService.getClientInfo()); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testMqttReconnect() { |
|
||||
try { |
|
||||
mqttService.reconnect(); |
|
||||
System.out.println("MQTT重连测试完成"); |
|
||||
} catch (Exception e) { |
|
||||
System.err.println("MQTT重连测试失败: " + e.getMessage()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testMqttPublish() { |
|
||||
try { |
|
||||
if (mqttService.isConnected()) { |
|
||||
mqttService.publish("/test/topic", "测试消息"); |
|
||||
System.out.println("MQTT消息发布测试完成"); |
|
||||
} else { |
|
||||
System.out.println("MQTT未连接，跳过发布测试"); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
System.err.println("MQTT消息发布测试失败: " + e.getMessage()); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
Loading…
Reference in new issue