| @@ -1,74 +1,74 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import java.io.File; | |||
| /** | |||
| * 微信客户端配置存储 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpConfigStorage { | |||
| String getAccessToken(); | |||
| boolean isAccessTokenExpired(); | |||
| /** | |||
| * 强制将access token过期掉 | |||
| */ | |||
| void expireAccessToken(); | |||
| void updateAccessToken(WxAccessToken accessToken); | |||
| void updateAccessToken(String accessToken, int expiresIn); | |||
| String getJsapiTicket(); | |||
| boolean isJsapiTicketExpired(); | |||
| /** | |||
| * 强制将jsapi ticket过期掉 | |||
| */ | |||
| void expireJsapiTicket(); | |||
| /** | |||
| * 应该是线程安全的 | |||
| * | |||
| * @param jsapiTicket | |||
| */ | |||
| void updateJsapiTicket(String jsapiTicket, int expiresInSeconds); | |||
| String getCorpId(); | |||
| String getCorpSecret(); | |||
| Integer getAgentId(); | |||
| String getToken(); | |||
| String getAesKey(); | |||
| long getExpiresTime(); | |||
| String getOauth2redirectUri(); | |||
| String getHttpProxyHost(); | |||
| int getHttpProxyPort(); | |||
| String getHttpProxyUsername(); | |||
| String getHttpProxyPassword(); | |||
| File getTmpDirFile(); | |||
| /** | |||
| * http client builder | |||
| * | |||
| * @return ApacheHttpClientBuilder | |||
| */ | |||
| ApacheHttpClientBuilder getApacheHttpClientBuilder(); | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import java.io.File; | |||
| /** | |||
| * 微信客户端配置存储 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpConfigStorage { | |||
| String getAccessToken(); | |||
| boolean isAccessTokenExpired(); | |||
| /** | |||
| * 强制将access token过期掉 | |||
| */ | |||
| void expireAccessToken(); | |||
| void updateAccessToken(WxAccessToken accessToken); | |||
| void updateAccessToken(String accessToken, int expiresIn); | |||
| String getJsapiTicket(); | |||
| boolean isJsapiTicketExpired(); | |||
| /** | |||
| * 强制将jsapi ticket过期掉 | |||
| */ | |||
| void expireJsapiTicket(); | |||
| /** | |||
| * 应该是线程安全的 | |||
| * | |||
| * @param jsapiTicket | |||
| */ | |||
| void updateJsapiTicket(String jsapiTicket, int expiresInSeconds); | |||
| String getCorpId(); | |||
| String getCorpSecret(); | |||
| Integer getAgentId(); | |||
| String getToken(); | |||
| String getAesKey(); | |||
| long getExpiresTime(); | |||
| String getOauth2redirectUri(); | |||
| String getHttpProxyHost(); | |||
| int getHttpProxyPort(); | |||
| String getHttpProxyUsername(); | |||
| String getHttpProxyPassword(); | |||
| File getTmpDirFile(); | |||
| /** | |||
| * http client builder | |||
| * | |||
| * @return ApacheHttpClientBuilder | |||
| */ | |||
| ApacheHttpClientBuilder getApacheHttpClientBuilder(); | |||
| } | |||
| @@ -1,224 +1,224 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.ToStringUtils; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import java.io.File; | |||
| /** | |||
| * 基于内存的微信配置provider,在实际生产环境中应该将这些配置持久化 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public class WxCpInMemoryConfigStorage implements WxCpConfigStorage { | |||
| protected volatile String corpId; | |||
| protected volatile String corpSecret; | |||
| protected volatile String token; | |||
| protected volatile String accessToken; | |||
| protected volatile String aesKey; | |||
| protected volatile Integer agentId; | |||
| protected volatile long expiresTime; | |||
| protected volatile String oauth2redirectUri; | |||
| protected volatile String httpProxyHost; | |||
| protected volatile int httpProxyPort; | |||
| protected volatile String httpProxyUsername; | |||
| protected volatile String httpProxyPassword; | |||
| protected volatile String jsapiTicket; | |||
| protected volatile long jsapiTicketExpiresTime; | |||
| protected volatile File tmpDirFile; | |||
| private volatile ApacheHttpClientBuilder apacheHttpClientBuilder; | |||
| @Override | |||
| public String getAccessToken() { | |||
| return this.accessToken; | |||
| } | |||
| public void setAccessToken(String accessToken) { | |||
| this.accessToken = accessToken; | |||
| } | |||
| @Override | |||
| public boolean isAccessTokenExpired() { | |||
| return System.currentTimeMillis() > this.expiresTime; | |||
| } | |||
| @Override | |||
| public void expireAccessToken() { | |||
| this.expiresTime = 0; | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(WxAccessToken accessToken) { | |||
| updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn()); | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) { | |||
| this.accessToken = accessToken; | |||
| this.expiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000l; | |||
| } | |||
| @Override | |||
| public String getJsapiTicket() { | |||
| return this.jsapiTicket; | |||
| } | |||
| public void setJsapiTicket(String jsapiTicket) { | |||
| this.jsapiTicket = jsapiTicket; | |||
| } | |||
| public long getJsapiTicketExpiresTime() { | |||
| return this.jsapiTicketExpiresTime; | |||
| } | |||
| public void setJsapiTicketExpiresTime(long jsapiTicketExpiresTime) { | |||
| this.jsapiTicketExpiresTime = jsapiTicketExpiresTime; | |||
| } | |||
| @Override | |||
| public boolean isJsapiTicketExpired() { | |||
| return System.currentTimeMillis() > this.jsapiTicketExpiresTime; | |||
| } | |||
| @Override | |||
| public synchronized void updateJsapiTicket(String jsapiTicket, int expiresInSeconds) { | |||
| this.jsapiTicket = jsapiTicket; | |||
| // 预留200秒的时间 | |||
| this.jsapiTicketExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000l; | |||
| } | |||
| @Override | |||
| public void expireJsapiTicket() { | |||
| this.jsapiTicketExpiresTime = 0; | |||
| } | |||
| @Override | |||
| public String getCorpId() { | |||
| return this.corpId; | |||
| } | |||
| public void setCorpId(String corpId) { | |||
| this.corpId = corpId; | |||
| } | |||
| @Override | |||
| public String getCorpSecret() { | |||
| return this.corpSecret; | |||
| } | |||
| public void setCorpSecret(String corpSecret) { | |||
| this.corpSecret = corpSecret; | |||
| } | |||
| @Override | |||
| public String getToken() { | |||
| return this.token; | |||
| } | |||
| public void setToken(String token) { | |||
| this.token = token; | |||
| } | |||
| @Override | |||
| public long getExpiresTime() { | |||
| return this.expiresTime; | |||
| } | |||
| public void setExpiresTime(long expiresTime) { | |||
| this.expiresTime = expiresTime; | |||
| } | |||
| @Override | |||
| public String getAesKey() { | |||
| return this.aesKey; | |||
| } | |||
| public void setAesKey(String aesKey) { | |||
| this.aesKey = aesKey; | |||
| } | |||
| @Override | |||
| public Integer getAgentId() { | |||
| return this.agentId; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| @Override | |||
| public String getOauth2redirectUri() { | |||
| return this.oauth2redirectUri; | |||
| } | |||
| public void setOauth2redirectUri(String oauth2redirectUri) { | |||
| this.oauth2redirectUri = oauth2redirectUri; | |||
| } | |||
| @Override | |||
| public String getHttpProxyHost() { | |||
| return this.httpProxyHost; | |||
| } | |||
| public void setHttpProxyHost(String httpProxyHost) { | |||
| this.httpProxyHost = httpProxyHost; | |||
| } | |||
| @Override | |||
| public int getHttpProxyPort() { | |||
| return this.httpProxyPort; | |||
| } | |||
| public void setHttpProxyPort(int httpProxyPort) { | |||
| this.httpProxyPort = httpProxyPort; | |||
| } | |||
| @Override | |||
| public String getHttpProxyUsername() { | |||
| return this.httpProxyUsername; | |||
| } | |||
| public void setHttpProxyUsername(String httpProxyUsername) { | |||
| this.httpProxyUsername = httpProxyUsername; | |||
| } | |||
| @Override | |||
| public String getHttpProxyPassword() { | |||
| return this.httpProxyPassword; | |||
| } | |||
| public void setHttpProxyPassword(String httpProxyPassword) { | |||
| this.httpProxyPassword = httpProxyPassword; | |||
| } | |||
| @Override | |||
| public String toString() { | |||
| return ToStringUtils.toSimpleString(this); | |||
| } | |||
| @Override | |||
| public File getTmpDirFile() { | |||
| return this.tmpDirFile; | |||
| } | |||
| public void setTmpDirFile(File tmpDirFile) { | |||
| this.tmpDirFile = tmpDirFile; | |||
| } | |||
| @Override | |||
| public ApacheHttpClientBuilder getApacheHttpClientBuilder() { | |||
| return this.apacheHttpClientBuilder; | |||
| } | |||
| public void setApacheHttpClientBuilder(ApacheHttpClientBuilder apacheHttpClientBuilder) { | |||
| this.apacheHttpClientBuilder = apacheHttpClientBuilder; | |||
| } | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.ToStringUtils; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import java.io.File; | |||
| /** | |||
| * 基于内存的微信配置provider,在实际生产环境中应该将这些配置持久化 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public class WxCpInMemoryConfigStorage implements WxCpConfigStorage { | |||
| protected volatile String corpId; | |||
| protected volatile String corpSecret; | |||
| protected volatile String token; | |||
| protected volatile String accessToken; | |||
| protected volatile String aesKey; | |||
| protected volatile Integer agentId; | |||
| protected volatile long expiresTime; | |||
| protected volatile String oauth2redirectUri; | |||
| protected volatile String httpProxyHost; | |||
| protected volatile int httpProxyPort; | |||
| protected volatile String httpProxyUsername; | |||
| protected volatile String httpProxyPassword; | |||
| protected volatile String jsapiTicket; | |||
| protected volatile long jsapiTicketExpiresTime; | |||
| protected volatile File tmpDirFile; | |||
| private volatile ApacheHttpClientBuilder apacheHttpClientBuilder; | |||
| @Override | |||
| public String getAccessToken() { | |||
| return this.accessToken; | |||
| } | |||
| public void setAccessToken(String accessToken) { | |||
| this.accessToken = accessToken; | |||
| } | |||
| @Override | |||
| public boolean isAccessTokenExpired() { | |||
| return System.currentTimeMillis() > this.expiresTime; | |||
| } | |||
| @Override | |||
| public void expireAccessToken() { | |||
| this.expiresTime = 0; | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(WxAccessToken accessToken) { | |||
| updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn()); | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) { | |||
| this.accessToken = accessToken; | |||
| this.expiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000l; | |||
| } | |||
| @Override | |||
| public String getJsapiTicket() { | |||
| return this.jsapiTicket; | |||
| } | |||
| public void setJsapiTicket(String jsapiTicket) { | |||
| this.jsapiTicket = jsapiTicket; | |||
| } | |||
| public long getJsapiTicketExpiresTime() { | |||
| return this.jsapiTicketExpiresTime; | |||
| } | |||
| public void setJsapiTicketExpiresTime(long jsapiTicketExpiresTime) { | |||
| this.jsapiTicketExpiresTime = jsapiTicketExpiresTime; | |||
| } | |||
| @Override | |||
| public boolean isJsapiTicketExpired() { | |||
| return System.currentTimeMillis() > this.jsapiTicketExpiresTime; | |||
| } | |||
| @Override | |||
| public synchronized void updateJsapiTicket(String jsapiTicket, int expiresInSeconds) { | |||
| this.jsapiTicket = jsapiTicket; | |||
| // 预留200秒的时间 | |||
| this.jsapiTicketExpiresTime = System.currentTimeMillis() + (expiresInSeconds - 200) * 1000l; | |||
| } | |||
| @Override | |||
| public void expireJsapiTicket() { | |||
| this.jsapiTicketExpiresTime = 0; | |||
| } | |||
| @Override | |||
| public String getCorpId() { | |||
| return this.corpId; | |||
| } | |||
| public void setCorpId(String corpId) { | |||
| this.corpId = corpId; | |||
| } | |||
| @Override | |||
| public String getCorpSecret() { | |||
| return this.corpSecret; | |||
| } | |||
| public void setCorpSecret(String corpSecret) { | |||
| this.corpSecret = corpSecret; | |||
| } | |||
| @Override | |||
| public String getToken() { | |||
| return this.token; | |||
| } | |||
| public void setToken(String token) { | |||
| this.token = token; | |||
| } | |||
| @Override | |||
| public long getExpiresTime() { | |||
| return this.expiresTime; | |||
| } | |||
| public void setExpiresTime(long expiresTime) { | |||
| this.expiresTime = expiresTime; | |||
| } | |||
| @Override | |||
| public String getAesKey() { | |||
| return this.aesKey; | |||
| } | |||
| public void setAesKey(String aesKey) { | |||
| this.aesKey = aesKey; | |||
| } | |||
| @Override | |||
| public Integer getAgentId() { | |||
| return this.agentId; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| @Override | |||
| public String getOauth2redirectUri() { | |||
| return this.oauth2redirectUri; | |||
| } | |||
| public void setOauth2redirectUri(String oauth2redirectUri) { | |||
| this.oauth2redirectUri = oauth2redirectUri; | |||
| } | |||
| @Override | |||
| public String getHttpProxyHost() { | |||
| return this.httpProxyHost; | |||
| } | |||
| public void setHttpProxyHost(String httpProxyHost) { | |||
| this.httpProxyHost = httpProxyHost; | |||
| } | |||
| @Override | |||
| public int getHttpProxyPort() { | |||
| return this.httpProxyPort; | |||
| } | |||
| public void setHttpProxyPort(int httpProxyPort) { | |||
| this.httpProxyPort = httpProxyPort; | |||
| } | |||
| @Override | |||
| public String getHttpProxyUsername() { | |||
| return this.httpProxyUsername; | |||
| } | |||
| public void setHttpProxyUsername(String httpProxyUsername) { | |||
| this.httpProxyUsername = httpProxyUsername; | |||
| } | |||
| @Override | |||
| public String getHttpProxyPassword() { | |||
| return this.httpProxyPassword; | |||
| } | |||
| public void setHttpProxyPassword(String httpProxyPassword) { | |||
| this.httpProxyPassword = httpProxyPassword; | |||
| } | |||
| @Override | |||
| public String toString() { | |||
| return ToStringUtils.toSimpleString(this); | |||
| } | |||
| @Override | |||
| public File getTmpDirFile() { | |||
| return this.tmpDirFile; | |||
| } | |||
| public void setTmpDirFile(File tmpDirFile) { | |||
| this.tmpDirFile = tmpDirFile; | |||
| } | |||
| @Override | |||
| public ApacheHttpClientBuilder getApacheHttpClientBuilder() { | |||
| return this.apacheHttpClientBuilder; | |||
| } | |||
| public void setApacheHttpClientBuilder(ApacheHttpClientBuilder apacheHttpClientBuilder) { | |||
| this.apacheHttpClientBuilder = apacheHttpClientBuilder; | |||
| } | |||
| } | |||
| @@ -1,269 +1,269 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import redis.clients.jedis.Jedis; | |||
| import redis.clients.jedis.JedisPool; | |||
| import redis.clients.jedis.JedisPoolConfig; | |||
| import java.io.File; | |||
| /** | |||
| * Jedis client implementor for wechat config storage | |||
| * | |||
| * @author gaigeshen | |||
| */ | |||
| public class WxCpJedisConfigStorage implements WxCpConfigStorage { | |||
| /* Redis keys here */ | |||
| private static final String ACCESS_TOKEN_KEY = "WX_CP_ACCESS_TOKEN"; | |||
| private static final String ACCESS_TOKEN_EXPIRES_TIME_KEY = "WX_CP_ACCESS_TOKEN_EXPIRES_TIME"; | |||
| private static final String JS_API_TICKET_KEY = "WX_CP_JS_API_TICKET"; | |||
| private static final String JS_API_TICKET_EXPIRES_TIME_KEY = "WX_CP_JS_API_TICKET_EXPIRES_TIME"; | |||
| /* Redis clients pool */ | |||
| private final JedisPool jedisPool; | |||
| private volatile String corpId; | |||
| private volatile String corpSecret; | |||
| private volatile String token; | |||
| private volatile String aesKey; | |||
| private volatile Integer agentId; | |||
| private volatile String oauth2redirectUri; | |||
| private volatile String httpProxyHost; | |||
| private volatile int httpProxyPort; | |||
| private volatile String httpProxyUsername; | |||
| private volatile String httpProxyPassword; | |||
| private volatile File tmpDirFile; | |||
| private volatile ApacheHttpClientBuilder apacheHttpClientBuilder; | |||
| public WxCpJedisConfigStorage(String host, int port) { | |||
| this.jedisPool = new JedisPool(host, port); | |||
| } | |||
| public WxCpJedisConfigStorage(JedisPoolConfig poolConfig, String host, int port) { | |||
| this.jedisPool = new JedisPool(poolConfig, host, port); | |||
| } | |||
| public WxCpJedisConfigStorage(JedisPoolConfig poolConfig, String host, int port, int timeout, final String password) { | |||
| this.jedisPool = new JedisPool(poolConfig, host, port, timeout, password); | |||
| } | |||
| /** | |||
| * This method will be destroy jedis pool | |||
| */ | |||
| public void destroy() { | |||
| this.jedisPool.destroy(); | |||
| } | |||
| @Override | |||
| public String getAccessToken() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| return jedis.get(ACCESS_TOKEN_KEY); | |||
| } | |||
| } | |||
| @Override | |||
| public boolean isAccessTokenExpired() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(ACCESS_TOKEN_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return System.currentTimeMillis() > expiresTime; | |||
| } | |||
| return true; | |||
| } | |||
| } | |||
| @Override | |||
| public void expireAccessToken() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(ACCESS_TOKEN_EXPIRES_TIME_KEY, "0"); | |||
| } | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(WxAccessToken accessToken) { | |||
| this.updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn()); | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(ACCESS_TOKEN_KEY, accessToken); | |||
| jedis.set(ACCESS_TOKEN_EXPIRES_TIME_KEY, | |||
| (System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L) + ""); | |||
| } | |||
| } | |||
| @Override | |||
| public String getJsapiTicket() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| return jedis.get(JS_API_TICKET_KEY); | |||
| } | |||
| } | |||
| @Override | |||
| public boolean isJsapiTicketExpired() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(JS_API_TICKET_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return System.currentTimeMillis() > expiresTime; | |||
| } | |||
| return true; | |||
| } | |||
| } | |||
| @Override | |||
| public void expireJsapiTicket() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(JS_API_TICKET_EXPIRES_TIME_KEY, "0"); | |||
| } | |||
| } | |||
| @Override | |||
| public synchronized void updateJsapiTicket(String jsapiTicket, int expiresInSeconds) { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(JS_API_TICKET_KEY, jsapiTicket); | |||
| jedis.set(JS_API_TICKET_EXPIRES_TIME_KEY, | |||
| (System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L + "")); | |||
| } | |||
| } | |||
| @Override | |||
| public String getCorpId() { | |||
| return this.corpId; | |||
| } | |||
| public void setCorpId(String corpId) { | |||
| this.corpId = corpId; | |||
| } | |||
| @Override | |||
| public String getCorpSecret() { | |||
| return this.corpSecret; | |||
| } | |||
| public void setCorpSecret(String corpSecret) { | |||
| this.corpSecret = corpSecret; | |||
| } | |||
| @Override | |||
| public Integer getAgentId() { | |||
| return this.agentId; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| @Override | |||
| public String getToken() { | |||
| return this.token; | |||
| } | |||
| public void setToken(String token) { | |||
| this.token = token; | |||
| } | |||
| @Override | |||
| public String getAesKey() { | |||
| return this.aesKey; | |||
| } | |||
| public void setAesKey(String aesKey) { | |||
| this.aesKey = aesKey; | |||
| } | |||
| @Override | |||
| public long getExpiresTime() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(ACCESS_TOKEN_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return expiresTime; | |||
| } | |||
| return 0L; | |||
| } | |||
| } | |||
| @Override | |||
| public String getOauth2redirectUri() { | |||
| return this.oauth2redirectUri; | |||
| } | |||
| public void setOauth2redirectUri(String oauth2redirectUri) { | |||
| this.oauth2redirectUri = oauth2redirectUri; | |||
| } | |||
| @Override | |||
| public String getHttpProxyHost() { | |||
| return this.httpProxyHost; | |||
| } | |||
| public void setHttpProxyHost(String httpProxyHost) { | |||
| this.httpProxyHost = httpProxyHost; | |||
| } | |||
| @Override | |||
| public int getHttpProxyPort() { | |||
| return this.httpProxyPort; | |||
| } | |||
| public void setHttpProxyPort(int httpProxyPort) { | |||
| this.httpProxyPort = httpProxyPort; | |||
| } | |||
| @Override | |||
| public String getHttpProxyUsername() { | |||
| return this.httpProxyUsername; | |||
| } | |||
| // ============================ Setters below | |||
| public void setHttpProxyUsername(String httpProxyUsername) { | |||
| this.httpProxyUsername = httpProxyUsername; | |||
| } | |||
| @Override | |||
| public String getHttpProxyPassword() { | |||
| return this.httpProxyPassword; | |||
| } | |||
| public void setHttpProxyPassword(String httpProxyPassword) { | |||
| this.httpProxyPassword = httpProxyPassword; | |||
| } | |||
| @Override | |||
| public File getTmpDirFile() { | |||
| return this.tmpDirFile; | |||
| } | |||
| public void setTmpDirFile(File tmpDirFile) { | |||
| this.tmpDirFile = tmpDirFile; | |||
| } | |||
| @Override | |||
| public ApacheHttpClientBuilder getApacheHttpClientBuilder() { | |||
| return this.apacheHttpClientBuilder; | |||
| } | |||
| public void setApacheHttpClientBuilder(ApacheHttpClientBuilder apacheHttpClientBuilder) { | |||
| this.apacheHttpClientBuilder = apacheHttpClientBuilder; | |||
| } | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.bean.WxAccessToken; | |||
| import me.chanjar.weixin.common.util.http.apache.ApacheHttpClientBuilder; | |||
| import redis.clients.jedis.Jedis; | |||
| import redis.clients.jedis.JedisPool; | |||
| import redis.clients.jedis.JedisPoolConfig; | |||
| import java.io.File; | |||
| /** | |||
| * Jedis client implementor for wechat config storage | |||
| * | |||
| * @author gaigeshen | |||
| */ | |||
| public class WxCpJedisConfigStorage implements WxCpConfigStorage { | |||
| /* Redis keys here */ | |||
| private static final String ACCESS_TOKEN_KEY = "WX_CP_ACCESS_TOKEN"; | |||
| private static final String ACCESS_TOKEN_EXPIRES_TIME_KEY = "WX_CP_ACCESS_TOKEN_EXPIRES_TIME"; | |||
| private static final String JS_API_TICKET_KEY = "WX_CP_JS_API_TICKET"; | |||
| private static final String JS_API_TICKET_EXPIRES_TIME_KEY = "WX_CP_JS_API_TICKET_EXPIRES_TIME"; | |||
| /* Redis clients pool */ | |||
| private final JedisPool jedisPool; | |||
| private volatile String corpId; | |||
| private volatile String corpSecret; | |||
| private volatile String token; | |||
| private volatile String aesKey; | |||
| private volatile Integer agentId; | |||
| private volatile String oauth2redirectUri; | |||
| private volatile String httpProxyHost; | |||
| private volatile int httpProxyPort; | |||
| private volatile String httpProxyUsername; | |||
| private volatile String httpProxyPassword; | |||
| private volatile File tmpDirFile; | |||
| private volatile ApacheHttpClientBuilder apacheHttpClientBuilder; | |||
| public WxCpJedisConfigStorage(String host, int port) { | |||
| this.jedisPool = new JedisPool(host, port); | |||
| } | |||
| public WxCpJedisConfigStorage(JedisPoolConfig poolConfig, String host, int port) { | |||
| this.jedisPool = new JedisPool(poolConfig, host, port); | |||
| } | |||
| public WxCpJedisConfigStorage(JedisPoolConfig poolConfig, String host, int port, int timeout, final String password) { | |||
| this.jedisPool = new JedisPool(poolConfig, host, port, timeout, password); | |||
| } | |||
| /** | |||
| * This method will be destroy jedis pool | |||
| */ | |||
| public void destroy() { | |||
| this.jedisPool.destroy(); | |||
| } | |||
| @Override | |||
| public String getAccessToken() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| return jedis.get(ACCESS_TOKEN_KEY); | |||
| } | |||
| } | |||
| @Override | |||
| public boolean isAccessTokenExpired() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(ACCESS_TOKEN_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return System.currentTimeMillis() > expiresTime; | |||
| } | |||
| return true; | |||
| } | |||
| } | |||
| @Override | |||
| public void expireAccessToken() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(ACCESS_TOKEN_EXPIRES_TIME_KEY, "0"); | |||
| } | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(WxAccessToken accessToken) { | |||
| this.updateAccessToken(accessToken.getAccessToken(), accessToken.getExpiresIn()); | |||
| } | |||
| @Override | |||
| public synchronized void updateAccessToken(String accessToken, int expiresInSeconds) { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(ACCESS_TOKEN_KEY, accessToken); | |||
| jedis.set(ACCESS_TOKEN_EXPIRES_TIME_KEY, | |||
| (System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L) + ""); | |||
| } | |||
| } | |||
| @Override | |||
| public String getJsapiTicket() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| return jedis.get(JS_API_TICKET_KEY); | |||
| } | |||
| } | |||
| @Override | |||
| public boolean isJsapiTicketExpired() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(JS_API_TICKET_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return System.currentTimeMillis() > expiresTime; | |||
| } | |||
| return true; | |||
| } | |||
| } | |||
| @Override | |||
| public void expireJsapiTicket() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(JS_API_TICKET_EXPIRES_TIME_KEY, "0"); | |||
| } | |||
| } | |||
| @Override | |||
| public synchronized void updateJsapiTicket(String jsapiTicket, int expiresInSeconds) { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| jedis.set(JS_API_TICKET_KEY, jsapiTicket); | |||
| jedis.set(JS_API_TICKET_EXPIRES_TIME_KEY, | |||
| (System.currentTimeMillis() + (expiresInSeconds - 200) * 1000L + "")); | |||
| } | |||
| } | |||
| @Override | |||
| public String getCorpId() { | |||
| return this.corpId; | |||
| } | |||
| public void setCorpId(String corpId) { | |||
| this.corpId = corpId; | |||
| } | |||
| @Override | |||
| public String getCorpSecret() { | |||
| return this.corpSecret; | |||
| } | |||
| public void setCorpSecret(String corpSecret) { | |||
| this.corpSecret = corpSecret; | |||
| } | |||
| @Override | |||
| public Integer getAgentId() { | |||
| return this.agentId; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| @Override | |||
| public String getToken() { | |||
| return this.token; | |||
| } | |||
| public void setToken(String token) { | |||
| this.token = token; | |||
| } | |||
| @Override | |||
| public String getAesKey() { | |||
| return this.aesKey; | |||
| } | |||
| public void setAesKey(String aesKey) { | |||
| this.aesKey = aesKey; | |||
| } | |||
| @Override | |||
| public long getExpiresTime() { | |||
| try (Jedis jedis = this.jedisPool.getResource()) { | |||
| String expiresTimeStr = jedis.get(ACCESS_TOKEN_EXPIRES_TIME_KEY); | |||
| if (expiresTimeStr != null) { | |||
| Long expiresTime = Long.parseLong(expiresTimeStr); | |||
| return expiresTime; | |||
| } | |||
| return 0L; | |||
| } | |||
| } | |||
| @Override | |||
| public String getOauth2redirectUri() { | |||
| return this.oauth2redirectUri; | |||
| } | |||
| public void setOauth2redirectUri(String oauth2redirectUri) { | |||
| this.oauth2redirectUri = oauth2redirectUri; | |||
| } | |||
| @Override | |||
| public String getHttpProxyHost() { | |||
| return this.httpProxyHost; | |||
| } | |||
| public void setHttpProxyHost(String httpProxyHost) { | |||
| this.httpProxyHost = httpProxyHost; | |||
| } | |||
| @Override | |||
| public int getHttpProxyPort() { | |||
| return this.httpProxyPort; | |||
| } | |||
| public void setHttpProxyPort(int httpProxyPort) { | |||
| this.httpProxyPort = httpProxyPort; | |||
| } | |||
| @Override | |||
| public String getHttpProxyUsername() { | |||
| return this.httpProxyUsername; | |||
| } | |||
| // ============================ Setters below | |||
| public void setHttpProxyUsername(String httpProxyUsername) { | |||
| this.httpProxyUsername = httpProxyUsername; | |||
| } | |||
| @Override | |||
| public String getHttpProxyPassword() { | |||
| return this.httpProxyPassword; | |||
| } | |||
| public void setHttpProxyPassword(String httpProxyPassword) { | |||
| this.httpProxyPassword = httpProxyPassword; | |||
| } | |||
| @Override | |||
| public File getTmpDirFile() { | |||
| return this.tmpDirFile; | |||
| } | |||
| public void setTmpDirFile(File tmpDirFile) { | |||
| this.tmpDirFile = tmpDirFile; | |||
| } | |||
| @Override | |||
| public ApacheHttpClientBuilder getApacheHttpClientBuilder() { | |||
| return this.apacheHttpClientBuilder; | |||
| } | |||
| public void setApacheHttpClientBuilder(ApacheHttpClientBuilder apacheHttpClientBuilder) { | |||
| this.apacheHttpClientBuilder = apacheHttpClientBuilder; | |||
| } | |||
| } | |||
| @@ -1,29 +1,29 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import java.util.Map; | |||
| /** | |||
| * 处理微信推送消息的处理器接口 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpMessageHandler { | |||
| /** | |||
| * @param wxMessage | |||
| * @param context 上下文,如果handler或interceptor之间有信息要传递,可以用这个 | |||
| * @param wxCpService | |||
| * @param sessionManager | |||
| * @return xml格式的消息,如果在异步规则里处理的话,可以返回null | |||
| */ | |||
| WxCpXmlOutMessage handle(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager) throws WxErrorException; | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import java.util.Map; | |||
| /** | |||
| * 处理微信推送消息的处理器接口 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpMessageHandler { | |||
| /** | |||
| * @param wxMessage | |||
| * @param context 上下文,如果handler或interceptor之间有信息要传递,可以用这个 | |||
| * @param wxCpService | |||
| * @param sessionManager | |||
| * @return xml格式的消息,如果在异步规则里处理的话,可以返回null | |||
| */ | |||
| WxCpXmlOutMessage handle(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager) throws WxErrorException; | |||
| } | |||
| @@ -1,30 +1,30 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import java.util.Map; | |||
| /** | |||
| * 微信消息拦截器,可以用来做验证 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpMessageInterceptor { | |||
| /** | |||
| * 拦截微信消息 | |||
| * | |||
| * @param wxMessage | |||
| * @param context 上下文,如果handler或interceptor之间有信息要传递,可以用这个 | |||
| * @param wxCpService | |||
| * @param sessionManager | |||
| * @return true代表OK,false代表不OK | |||
| */ | |||
| boolean intercept(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager) throws WxErrorException; | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import java.util.Map; | |||
| /** | |||
| * 微信消息拦截器,可以用来做验证 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public interface WxCpMessageInterceptor { | |||
| /** | |||
| * 拦截微信消息 | |||
| * | |||
| * @param wxMessage | |||
| * @param context 上下文,如果handler或interceptor之间有信息要传递,可以用这个 | |||
| * @param wxCpService | |||
| * @param sessionManager | |||
| * @return true代表OK,false代表不OK | |||
| */ | |||
| boolean intercept(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager) throws WxErrorException; | |||
| } | |||
| @@ -1,15 +1,15 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| /** | |||
| * 消息匹配器,用在消息路由的时候 | |||
| */ | |||
| public interface WxCpMessageMatcher { | |||
| /** | |||
| * 消息是否匹配某种模式 | |||
| */ | |||
| boolean match(WxCpXmlMessage message); | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| /** | |||
| * 消息匹配器,用在消息路由的时候 | |||
| */ | |||
| public interface WxCpMessageMatcher { | |||
| /** | |||
| * 消息是否匹配某种模式 | |||
| */ | |||
| boolean match(WxCpXmlMessage message); | |||
| } | |||
| @@ -1,249 +1,249 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.api.WxErrorExceptionHandler; | |||
| import me.chanjar.weixin.common.api.WxMessageDuplicateChecker; | |||
| import me.chanjar.weixin.common.api.WxMessageInMemoryDuplicateChecker; | |||
| import me.chanjar.weixin.common.session.InternalSession; | |||
| import me.chanjar.weixin.common.session.InternalSessionManager; | |||
| import me.chanjar.weixin.common.session.StandardSessionManager; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.common.util.LogExceptionHandler; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import org.slf4j.Logger; | |||
| import org.slf4j.LoggerFactory; | |||
| import java.util.ArrayList; | |||
| import java.util.HashMap; | |||
| import java.util.List; | |||
| import java.util.Map; | |||
| import java.util.concurrent.ExecutionException; | |||
| import java.util.concurrent.ExecutorService; | |||
| import java.util.concurrent.Executors; | |||
| import java.util.concurrent.Future; | |||
| /** | |||
| * <pre> | |||
| * 微信消息路由器,通过代码化的配置,把来自微信的消息交给handler处理 | |||
| * | |||
| * 说明: | |||
| * 1. 配置路由规则时要按照从细到粗的原则,否则可能消息可能会被提前处理 | |||
| * 2. 默认情况下消息只会被处理一次,除非使用 {@link WxCpMessageRouterRule#next()} | |||
| * 3. 规则的结束必须用{@link WxCpMessageRouterRule#end()}或者{@link WxCpMessageRouterRule#next()},否则不会生效 | |||
| * | |||
| * 使用方法: | |||
| * WxCpMessageRouter router = new WxCpMessageRouter(); | |||
| * router | |||
| * .rule() | |||
| * .msgType("MSG_TYPE").event("EVENT").eventKey("EVENT_KEY").content("CONTENT") | |||
| * .interceptor(interceptor, ...).handler(handler, ...) | |||
| * .end() | |||
| * .rule() | |||
| * // 另外一个匹配规则 | |||
| * .end() | |||
| * ; | |||
| * | |||
| * // 将WxXmlMessage交给消息路由器 | |||
| * router.route(message); | |||
| * | |||
| * </pre> | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public class WxCpMessageRouter { | |||
| private static final int DEFAULT_THREAD_POOL_SIZE = 100; | |||
| protected final Logger log = LoggerFactory.getLogger(WxCpMessageRouter.class); | |||
| private final List<WxCpMessageRouterRule> rules = new ArrayList<>(); | |||
| private final WxCpService wxCpService; | |||
| private ExecutorService executorService; | |||
| private WxMessageDuplicateChecker messageDuplicateChecker; | |||
| private WxSessionManager sessionManager; | |||
| private WxErrorExceptionHandler exceptionHandler; | |||
| public WxCpMessageRouter(WxCpService wxCpService) { | |||
| this.wxCpService = wxCpService; | |||
| this.executorService = Executors.newFixedThreadPool(DEFAULT_THREAD_POOL_SIZE); | |||
| this.messageDuplicateChecker = new WxMessageInMemoryDuplicateChecker(); | |||
| this.sessionManager = new StandardSessionManager(); | |||
| this.exceptionHandler = new LogExceptionHandler(); | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的 {@link ExecutorService} | |||
| * 如果不调用该方法,默认使用 Executors.newFixedThreadPool(100) | |||
| * </pre> | |||
| * | |||
| * @param executorService | |||
| */ | |||
| public void setExecutorService(ExecutorService executorService) { | |||
| this.executorService = executorService; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的 {@link me.chanjar.weixin.common.api.WxMessageDuplicateChecker} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.api.WxMessageInMemoryDuplicateChecker} | |||
| * </pre> | |||
| * | |||
| * @param messageDuplicateChecker | |||
| */ | |||
| public void setMessageDuplicateChecker(WxMessageDuplicateChecker messageDuplicateChecker) { | |||
| this.messageDuplicateChecker = messageDuplicateChecker; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的{@link me.chanjar.weixin.common.session.WxSessionManager} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.session.StandardSessionManager} | |||
| * </pre> | |||
| * | |||
| * @param sessionManager | |||
| */ | |||
| public void setSessionManager(WxSessionManager sessionManager) { | |||
| this.sessionManager = sessionManager; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的{@link me.chanjar.weixin.common.api.WxErrorExceptionHandler} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.util.LogExceptionHandler} | |||
| * </pre> | |||
| * | |||
| * @param exceptionHandler | |||
| */ | |||
| public void setExceptionHandler(WxErrorExceptionHandler exceptionHandler) { | |||
| this.exceptionHandler = exceptionHandler; | |||
| } | |||
| List<WxCpMessageRouterRule> getRules() { | |||
| return this.rules; | |||
| } | |||
| /** | |||
| * 开始一个新的Route规则 | |||
| */ | |||
| public WxCpMessageRouterRule rule() { | |||
| return new WxCpMessageRouterRule(this); | |||
| } | |||
| /** | |||
| * 处理微信消息 | |||
| * | |||
| * @param wxMessage | |||
| * @param context | |||
| */ | |||
| public WxCpXmlOutMessage route(final WxCpXmlMessage wxMessage, final Map<String, Object> context) { | |||
| if (isDuplicateMessage(wxMessage)) { | |||
| // 如果是重复消息,那么就不做处理 | |||
| return null; | |||
| } | |||
| final List<WxCpMessageRouterRule> matchRules = new ArrayList<>(); | |||
| // 收集匹配的规则 | |||
| for (final WxCpMessageRouterRule rule : this.rules) { | |||
| if (rule.test(wxMessage)) { | |||
| matchRules.add(rule); | |||
| if (!rule.isReEnter()) { | |||
| break; | |||
| } | |||
| } | |||
| } | |||
| if (matchRules.size() == 0) { | |||
| return null; | |||
| } | |||
| WxCpXmlOutMessage res = null; | |||
| final List<Future> futures = new ArrayList<>(); | |||
| for (final WxCpMessageRouterRule rule : matchRules) { | |||
| // 返回最后一个非异步的rule的执行结果 | |||
| if (rule.isAsync()) { | |||
| futures.add( | |||
| this.executorService.submit(new Runnable() { | |||
| @Override | |||
| public void run() { | |||
| rule.service(wxMessage, context, WxCpMessageRouter.this.wxCpService, WxCpMessageRouter.this.sessionManager, WxCpMessageRouter.this.exceptionHandler); | |||
| } | |||
| }) | |||
| ); | |||
| } else { | |||
| res = rule.service(wxMessage, context, this.wxCpService, this.sessionManager, this.exceptionHandler); | |||
| // 在同步操作结束,session访问结束 | |||
| this.log.debug("End session access: async=false, sessionId={}", wxMessage.getFromUserName()); | |||
| sessionEndAccess(wxMessage); | |||
| } | |||
| } | |||
| if (futures.size() > 0) { | |||
| this.executorService.submit(new Runnable() { | |||
| @Override | |||
| public void run() { | |||
| for (Future future : futures) { | |||
| try { | |||
| future.get(); | |||
| WxCpMessageRouter.this.log.debug("End session access: async=true, sessionId={}", wxMessage.getFromUserName()); | |||
| // 异步操作结束,session访问结束 | |||
| sessionEndAccess(wxMessage); | |||
| } catch (InterruptedException e) { | |||
| WxCpMessageRouter.this.log.error("Error happened when wait task finish", e); | |||
| } catch (ExecutionException e) { | |||
| WxCpMessageRouter.this.log.error("Error happened when wait task finish", e); | |||
| } | |||
| } | |||
| } | |||
| }); | |||
| } | |||
| return res; | |||
| } | |||
| /** | |||
| * 处理微信消息 | |||
| * | |||
| * @param wxMessage | |||
| */ | |||
| public WxCpXmlOutMessage route(final WxCpXmlMessage wxMessage) { | |||
| return this.route(wxMessage, new HashMap<String, Object>()); | |||
| } | |||
| protected boolean isDuplicateMessage(WxCpXmlMessage wxMessage) { | |||
| String messageId = ""; | |||
| if (wxMessage.getMsgId() == null) { | |||
| messageId = String.valueOf(wxMessage.getCreateTime()) | |||
| + "-" + String.valueOf(wxMessage.getAgentId() == null ? "" : wxMessage.getAgentId()) | |||
| + "-" + wxMessage.getFromUserName() | |||
| + "-" + String.valueOf(wxMessage.getEventKey() == null ? "" : wxMessage.getEventKey()) | |||
| + "-" + String.valueOf(wxMessage.getEvent() == null ? "" : wxMessage.getEvent()) | |||
| ; | |||
| } else { | |||
| messageId = String.valueOf(wxMessage.getMsgId()); | |||
| } | |||
| return this.messageDuplicateChecker.isDuplicate(messageId); | |||
| } | |||
| /** | |||
| * 对session的访问结束 | |||
| * | |||
| * @param wxMessage | |||
| */ | |||
| protected void sessionEndAccess(WxCpXmlMessage wxMessage) { | |||
| InternalSession session = ((InternalSessionManager) this.sessionManager).findSession(wxMessage.getFromUserName()); | |||
| if (session != null) { | |||
| session.endAccess(); | |||
| } | |||
| } | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.api.WxErrorExceptionHandler; | |||
| import me.chanjar.weixin.common.api.WxMessageDuplicateChecker; | |||
| import me.chanjar.weixin.common.api.WxMessageInMemoryDuplicateChecker; | |||
| import me.chanjar.weixin.common.session.InternalSession; | |||
| import me.chanjar.weixin.common.session.InternalSessionManager; | |||
| import me.chanjar.weixin.common.session.StandardSessionManager; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.common.util.LogExceptionHandler; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import org.slf4j.Logger; | |||
| import org.slf4j.LoggerFactory; | |||
| import java.util.ArrayList; | |||
| import java.util.HashMap; | |||
| import java.util.List; | |||
| import java.util.Map; | |||
| import java.util.concurrent.ExecutionException; | |||
| import java.util.concurrent.ExecutorService; | |||
| import java.util.concurrent.Executors; | |||
| import java.util.concurrent.Future; | |||
| /** | |||
| * <pre> | |||
| * 微信消息路由器,通过代码化的配置,把来自微信的消息交给handler处理 | |||
| * | |||
| * 说明: | |||
| * 1. 配置路由规则时要按照从细到粗的原则,否则可能消息可能会被提前处理 | |||
| * 2. 默认情况下消息只会被处理一次,除非使用 {@link WxCpMessageRouterRule#next()} | |||
| * 3. 规则的结束必须用{@link WxCpMessageRouterRule#end()}或者{@link WxCpMessageRouterRule#next()},否则不会生效 | |||
| * | |||
| * 使用方法: | |||
| * WxCpMessageRouter router = new WxCpMessageRouter(); | |||
| * router | |||
| * .rule() | |||
| * .msgType("MSG_TYPE").event("EVENT").eventKey("EVENT_KEY").content("CONTENT") | |||
| * .interceptor(interceptor, ...).handler(handler, ...) | |||
| * .end() | |||
| * .rule() | |||
| * // 另外一个匹配规则 | |||
| * .end() | |||
| * ; | |||
| * | |||
| * // 将WxXmlMessage交给消息路由器 | |||
| * router.route(message); | |||
| * | |||
| * </pre> | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| public class WxCpMessageRouter { | |||
| private static final int DEFAULT_THREAD_POOL_SIZE = 100; | |||
| protected final Logger log = LoggerFactory.getLogger(WxCpMessageRouter.class); | |||
| private final List<WxCpMessageRouterRule> rules = new ArrayList<>(); | |||
| private final WxCpService wxCpService; | |||
| private ExecutorService executorService; | |||
| private WxMessageDuplicateChecker messageDuplicateChecker; | |||
| private WxSessionManager sessionManager; | |||
| private WxErrorExceptionHandler exceptionHandler; | |||
| public WxCpMessageRouter(WxCpService wxCpService) { | |||
| this.wxCpService = wxCpService; | |||
| this.executorService = Executors.newFixedThreadPool(DEFAULT_THREAD_POOL_SIZE); | |||
| this.messageDuplicateChecker = new WxMessageInMemoryDuplicateChecker(); | |||
| this.sessionManager = new StandardSessionManager(); | |||
| this.exceptionHandler = new LogExceptionHandler(); | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的 {@link ExecutorService} | |||
| * 如果不调用该方法,默认使用 Executors.newFixedThreadPool(100) | |||
| * </pre> | |||
| * | |||
| * @param executorService | |||
| */ | |||
| public void setExecutorService(ExecutorService executorService) { | |||
| this.executorService = executorService; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的 {@link me.chanjar.weixin.common.api.WxMessageDuplicateChecker} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.api.WxMessageInMemoryDuplicateChecker} | |||
| * </pre> | |||
| * | |||
| * @param messageDuplicateChecker | |||
| */ | |||
| public void setMessageDuplicateChecker(WxMessageDuplicateChecker messageDuplicateChecker) { | |||
| this.messageDuplicateChecker = messageDuplicateChecker; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的{@link me.chanjar.weixin.common.session.WxSessionManager} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.session.StandardSessionManager} | |||
| * </pre> | |||
| * | |||
| * @param sessionManager | |||
| */ | |||
| public void setSessionManager(WxSessionManager sessionManager) { | |||
| this.sessionManager = sessionManager; | |||
| } | |||
| /** | |||
| * <pre> | |||
| * 设置自定义的{@link me.chanjar.weixin.common.api.WxErrorExceptionHandler} | |||
| * 如果不调用该方法,默认使用 {@link me.chanjar.weixin.common.util.LogExceptionHandler} | |||
| * </pre> | |||
| * | |||
| * @param exceptionHandler | |||
| */ | |||
| public void setExceptionHandler(WxErrorExceptionHandler exceptionHandler) { | |||
| this.exceptionHandler = exceptionHandler; | |||
| } | |||
| List<WxCpMessageRouterRule> getRules() { | |||
| return this.rules; | |||
| } | |||
| /** | |||
| * 开始一个新的Route规则 | |||
| */ | |||
| public WxCpMessageRouterRule rule() { | |||
| return new WxCpMessageRouterRule(this); | |||
| } | |||
| /** | |||
| * 处理微信消息 | |||
| * | |||
| * @param wxMessage | |||
| * @param context | |||
| */ | |||
| public WxCpXmlOutMessage route(final WxCpXmlMessage wxMessage, final Map<String, Object> context) { | |||
| if (isDuplicateMessage(wxMessage)) { | |||
| // 如果是重复消息,那么就不做处理 | |||
| return null; | |||
| } | |||
| final List<WxCpMessageRouterRule> matchRules = new ArrayList<>(); | |||
| // 收集匹配的规则 | |||
| for (final WxCpMessageRouterRule rule : this.rules) { | |||
| if (rule.test(wxMessage)) { | |||
| matchRules.add(rule); | |||
| if (!rule.isReEnter()) { | |||
| break; | |||
| } | |||
| } | |||
| } | |||
| if (matchRules.size() == 0) { | |||
| return null; | |||
| } | |||
| WxCpXmlOutMessage res = null; | |||
| final List<Future> futures = new ArrayList<>(); | |||
| for (final WxCpMessageRouterRule rule : matchRules) { | |||
| // 返回最后一个非异步的rule的执行结果 | |||
| if (rule.isAsync()) { | |||
| futures.add( | |||
| this.executorService.submit(new Runnable() { | |||
| @Override | |||
| public void run() { | |||
| rule.service(wxMessage, context, WxCpMessageRouter.this.wxCpService, WxCpMessageRouter.this.sessionManager, WxCpMessageRouter.this.exceptionHandler); | |||
| } | |||
| }) | |||
| ); | |||
| } else { | |||
| res = rule.service(wxMessage, context, this.wxCpService, this.sessionManager, this.exceptionHandler); | |||
| // 在同步操作结束,session访问结束 | |||
| this.log.debug("End session access: async=false, sessionId={}", wxMessage.getFromUserName()); | |||
| sessionEndAccess(wxMessage); | |||
| } | |||
| } | |||
| if (futures.size() > 0) { | |||
| this.executorService.submit(new Runnable() { | |||
| @Override | |||
| public void run() { | |||
| for (Future future : futures) { | |||
| try { | |||
| future.get(); | |||
| WxCpMessageRouter.this.log.debug("End session access: async=true, sessionId={}", wxMessage.getFromUserName()); | |||
| // 异步操作结束,session访问结束 | |||
| sessionEndAccess(wxMessage); | |||
| } catch (InterruptedException e) { | |||
| WxCpMessageRouter.this.log.error("Error happened when wait task finish", e); | |||
| } catch (ExecutionException e) { | |||
| WxCpMessageRouter.this.log.error("Error happened when wait task finish", e); | |||
| } | |||
| } | |||
| } | |||
| }); | |||
| } | |||
| return res; | |||
| } | |||
| /** | |||
| * 处理微信消息 | |||
| * | |||
| * @param wxMessage | |||
| */ | |||
| public WxCpXmlOutMessage route(final WxCpXmlMessage wxMessage) { | |||
| return this.route(wxMessage, new HashMap<String, Object>()); | |||
| } | |||
| protected boolean isDuplicateMessage(WxCpXmlMessage wxMessage) { | |||
| String messageId = ""; | |||
| if (wxMessage.getMsgId() == null) { | |||
| messageId = String.valueOf(wxMessage.getCreateTime()) | |||
| + "-" + String.valueOf(wxMessage.getAgentId() == null ? "" : wxMessage.getAgentId()) | |||
| + "-" + wxMessage.getFromUserName() | |||
| + "-" + String.valueOf(wxMessage.getEventKey() == null ? "" : wxMessage.getEventKey()) | |||
| + "-" + String.valueOf(wxMessage.getEvent() == null ? "" : wxMessage.getEvent()) | |||
| ; | |||
| } else { | |||
| messageId = String.valueOf(wxMessage.getMsgId()); | |||
| } | |||
| return this.messageDuplicateChecker.isDuplicate(messageId); | |||
| } | |||
| /** | |||
| * 对session的访问结束 | |||
| * | |||
| * @param wxMessage | |||
| */ | |||
| protected void sessionEndAccess(WxCpXmlMessage wxMessage) { | |||
| InternalSession session = ((InternalSessionManager) this.sessionManager).findSession(wxMessage.getFromUserName()); | |||
| if (session != null) { | |||
| session.endAccess(); | |||
| } | |||
| } | |||
| } | |||
| @@ -1,321 +1,321 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.api.WxErrorExceptionHandler; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import java.util.ArrayList; | |||
| import java.util.HashMap; | |||
| import java.util.List; | |||
| import java.util.Map; | |||
| import java.util.regex.Pattern; | |||
| public class WxCpMessageRouterRule { | |||
| private final WxCpMessageRouter routerBuilder; | |||
| private boolean async = true; | |||
| private String fromUser; | |||
| private String msgType; | |||
| private String event; | |||
| private String eventKey; | |||
| private String content; | |||
| private String rContent; | |||
| private WxCpMessageMatcher matcher; | |||
| private boolean reEnter = false; | |||
| private Integer agentId; | |||
| private List<WxCpMessageHandler> handlers = new ArrayList<>(); | |||
| private List<WxCpMessageInterceptor> interceptors = new ArrayList<>(); | |||
| protected WxCpMessageRouterRule(WxCpMessageRouter routerBuilder) { | |||
| this.routerBuilder = routerBuilder; | |||
| } | |||
| /** | |||
| * 设置是否异步执行,默认是true | |||
| * | |||
| * @param async | |||
| */ | |||
| public WxCpMessageRouterRule async(boolean async) { | |||
| this.async = async; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果agentId匹配 | |||
| * | |||
| * @param agentId | |||
| */ | |||
| public WxCpMessageRouterRule agentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果msgType等于某值 | |||
| * | |||
| * @param msgType | |||
| */ | |||
| public WxCpMessageRouterRule msgType(String msgType) { | |||
| this.msgType = msgType; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果event等于某值 | |||
| * | |||
| * @param event | |||
| */ | |||
| public WxCpMessageRouterRule event(String event) { | |||
| this.event = event; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果eventKey等于某值 | |||
| * | |||
| * @param eventKey | |||
| */ | |||
| public WxCpMessageRouterRule eventKey(String eventKey) { | |||
| this.eventKey = eventKey; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果content等于某值 | |||
| * | |||
| * @param content | |||
| */ | |||
| public WxCpMessageRouterRule content(String content) { | |||
| this.content = content; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果content匹配该正则表达式 | |||
| * | |||
| * @param regex | |||
| */ | |||
| public WxCpMessageRouterRule rContent(String regex) { | |||
| this.rContent = regex; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果fromUser等于某值 | |||
| * | |||
| * @param fromUser | |||
| */ | |||
| public WxCpMessageRouterRule fromUser(String fromUser) { | |||
| this.fromUser = fromUser; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果消息匹配某个matcher,用在用户需要自定义更复杂的匹配规则的时候 | |||
| * | |||
| * @param matcher | |||
| */ | |||
| public WxCpMessageRouterRule matcher(WxCpMessageMatcher matcher) { | |||
| this.matcher = matcher; | |||
| return this; | |||
| } | |||
| /** | |||
| * 设置微信消息拦截器 | |||
| * | |||
| * @param interceptor | |||
| */ | |||
| public WxCpMessageRouterRule interceptor(WxCpMessageInterceptor interceptor) { | |||
| return interceptor(interceptor, (WxCpMessageInterceptor[]) null); | |||
| } | |||
| /** | |||
| * 设置微信消息拦截器 | |||
| * | |||
| * @param interceptor | |||
| * @param otherInterceptors | |||
| */ | |||
| public WxCpMessageRouterRule interceptor(WxCpMessageInterceptor interceptor, WxCpMessageInterceptor... otherInterceptors) { | |||
| this.interceptors.add(interceptor); | |||
| if (otherInterceptors != null && otherInterceptors.length > 0) { | |||
| for (WxCpMessageInterceptor i : otherInterceptors) { | |||
| this.interceptors.add(i); | |||
| } | |||
| } | |||
| return this; | |||
| } | |||
| /** | |||
| * 设置微信消息处理器 | |||
| * | |||
| * @param handler | |||
| */ | |||
| public WxCpMessageRouterRule handler(WxCpMessageHandler handler) { | |||
| return handler(handler, (WxCpMessageHandler[]) null); | |||
| } | |||
| /** | |||
| * 设置微信消息处理器 | |||
| * | |||
| * @param handler | |||
| * @param otherHandlers | |||
| */ | |||
| public WxCpMessageRouterRule handler(WxCpMessageHandler handler, WxCpMessageHandler... otherHandlers) { | |||
| this.handlers.add(handler); | |||
| if (otherHandlers != null && otherHandlers.length > 0) { | |||
| for (WxCpMessageHandler i : otherHandlers) { | |||
| this.handlers.add(i); | |||
| } | |||
| } | |||
| return this; | |||
| } | |||
| /** | |||
| * 规则结束,代表如果一个消息匹配该规则,那么它将不再会进入其他规则 | |||
| */ | |||
| public WxCpMessageRouter end() { | |||
| this.routerBuilder.getRules().add(this); | |||
| return this.routerBuilder; | |||
| } | |||
| /** | |||
| * 规则结束,但是消息还会进入其他规则 | |||
| */ | |||
| public WxCpMessageRouter next() { | |||
| this.reEnter = true; | |||
| return end(); | |||
| } | |||
| protected boolean test(WxCpXmlMessage wxMessage) { | |||
| return | |||
| (this.fromUser == null || this.fromUser.equals(wxMessage.getFromUserName())) | |||
| && | |||
| (this.agentId == null || this.agentId.equals(wxMessage.getAgentId())) | |||
| && | |||
| (this.msgType == null || this.msgType.equals(wxMessage.getMsgType())) | |||
| && | |||
| (this.event == null || this.event.equals(wxMessage.getEvent())) | |||
| && | |||
| (this.eventKey == null || this.eventKey.equals(wxMessage.getEventKey())) | |||
| && | |||
| (this.content == null || this.content | |||
| .equals(wxMessage.getContent() == null ? null : wxMessage.getContent().trim())) | |||
| && | |||
| (this.rContent == null || Pattern | |||
| .matches(this.rContent, wxMessage.getContent() == null ? "" : wxMessage.getContent().trim())) | |||
| && | |||
| (this.matcher == null || this.matcher.match(wxMessage)) | |||
| ; | |||
| } | |||
| /** | |||
| * 处理微信推送过来的消息 | |||
| * | |||
| * @param wxMessage | |||
| * @return true 代表继续执行别的router,false 代表停止执行别的router | |||
| */ | |||
| protected WxCpXmlOutMessage service(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager, | |||
| WxErrorExceptionHandler exceptionHandler) { | |||
| if (context == null) { | |||
| context = new HashMap<>(); | |||
| } | |||
| try { | |||
| // 如果拦截器不通过 | |||
| for (WxCpMessageInterceptor interceptor : this.interceptors) { | |||
| if (!interceptor.intercept(wxMessage, context, wxCpService, sessionManager)) { | |||
| return null; | |||
| } | |||
| } | |||
| // 交给handler处理 | |||
| WxCpXmlOutMessage res = null; | |||
| for (WxCpMessageHandler handler : this.handlers) { | |||
| // 返回最后handler的结果 | |||
| res = handler.handle(wxMessage, context, wxCpService, sessionManager); | |||
| } | |||
| return res; | |||
| } catch (WxErrorException e) { | |||
| exceptionHandler.handle(e); | |||
| } | |||
| return null; | |||
| } | |||
| public void setFromUser(String fromUser) { | |||
| this.fromUser = fromUser; | |||
| } | |||
| public void setMsgType(String msgType) { | |||
| this.msgType = msgType; | |||
| } | |||
| public void setEvent(String event) { | |||
| this.event = event; | |||
| } | |||
| public void setEventKey(String eventKey) { | |||
| this.eventKey = eventKey; | |||
| } | |||
| public void setContent(String content) { | |||
| this.content = content; | |||
| } | |||
| public void setrContent(String rContent) { | |||
| this.rContent = rContent; | |||
| } | |||
| public void setMatcher(WxCpMessageMatcher matcher) { | |||
| this.matcher = matcher; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| public void setHandlers(List<WxCpMessageHandler> handlers) { | |||
| this.handlers = handlers; | |||
| } | |||
| public void setInterceptors(List<WxCpMessageInterceptor> interceptors) { | |||
| this.interceptors = interceptors; | |||
| } | |||
| public boolean isAsync() { | |||
| return this.async; | |||
| } | |||
| public void setAsync(boolean async) { | |||
| this.async = async; | |||
| } | |||
| public boolean isReEnter() { | |||
| return this.reEnter; | |||
| } | |||
| public void setReEnter(boolean reEnter) { | |||
| this.reEnter = reEnter; | |||
| } | |||
| } | |||
| package me.chanjar.weixin.cp.api; | |||
| import me.chanjar.weixin.common.api.WxErrorExceptionHandler; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.common.session.WxSessionManager; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlMessage; | |||
| import me.chanjar.weixin.cp.bean.WxCpXmlOutMessage; | |||
| import java.util.ArrayList; | |||
| import java.util.HashMap; | |||
| import java.util.List; | |||
| import java.util.Map; | |||
| import java.util.regex.Pattern; | |||
| public class WxCpMessageRouterRule { | |||
| private final WxCpMessageRouter routerBuilder; | |||
| private boolean async = true; | |||
| private String fromUser; | |||
| private String msgType; | |||
| private String event; | |||
| private String eventKey; | |||
| private String content; | |||
| private String rContent; | |||
| private WxCpMessageMatcher matcher; | |||
| private boolean reEnter = false; | |||
| private Integer agentId; | |||
| private List<WxCpMessageHandler> handlers = new ArrayList<>(); | |||
| private List<WxCpMessageInterceptor> interceptors = new ArrayList<>(); | |||
| protected WxCpMessageRouterRule(WxCpMessageRouter routerBuilder) { | |||
| this.routerBuilder = routerBuilder; | |||
| } | |||
| /** | |||
| * 设置是否异步执行,默认是true | |||
| * | |||
| * @param async | |||
| */ | |||
| public WxCpMessageRouterRule async(boolean async) { | |||
| this.async = async; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果agentId匹配 | |||
| * | |||
| * @param agentId | |||
| */ | |||
| public WxCpMessageRouterRule agentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果msgType等于某值 | |||
| * | |||
| * @param msgType | |||
| */ | |||
| public WxCpMessageRouterRule msgType(String msgType) { | |||
| this.msgType = msgType; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果event等于某值 | |||
| * | |||
| * @param event | |||
| */ | |||
| public WxCpMessageRouterRule event(String event) { | |||
| this.event = event; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果eventKey等于某值 | |||
| * | |||
| * @param eventKey | |||
| */ | |||
| public WxCpMessageRouterRule eventKey(String eventKey) { | |||
| this.eventKey = eventKey; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果content等于某值 | |||
| * | |||
| * @param content | |||
| */ | |||
| public WxCpMessageRouterRule content(String content) { | |||
| this.content = content; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果content匹配该正则表达式 | |||
| * | |||
| * @param regex | |||
| */ | |||
| public WxCpMessageRouterRule rContent(String regex) { | |||
| this.rContent = regex; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果fromUser等于某值 | |||
| * | |||
| * @param fromUser | |||
| */ | |||
| public WxCpMessageRouterRule fromUser(String fromUser) { | |||
| this.fromUser = fromUser; | |||
| return this; | |||
| } | |||
| /** | |||
| * 如果消息匹配某个matcher,用在用户需要自定义更复杂的匹配规则的时候 | |||
| * | |||
| * @param matcher | |||
| */ | |||
| public WxCpMessageRouterRule matcher(WxCpMessageMatcher matcher) { | |||
| this.matcher = matcher; | |||
| return this; | |||
| } | |||
| /** | |||
| * 设置微信消息拦截器 | |||
| * | |||
| * @param interceptor | |||
| */ | |||
| public WxCpMessageRouterRule interceptor(WxCpMessageInterceptor interceptor) { | |||
| return interceptor(interceptor, (WxCpMessageInterceptor[]) null); | |||
| } | |||
| /** | |||
| * 设置微信消息拦截器 | |||
| * | |||
| * @param interceptor | |||
| * @param otherInterceptors | |||
| */ | |||
| public WxCpMessageRouterRule interceptor(WxCpMessageInterceptor interceptor, WxCpMessageInterceptor... otherInterceptors) { | |||
| this.interceptors.add(interceptor); | |||
| if (otherInterceptors != null && otherInterceptors.length > 0) { | |||
| for (WxCpMessageInterceptor i : otherInterceptors) { | |||
| this.interceptors.add(i); | |||
| } | |||
| } | |||
| return this; | |||
| } | |||
| /** | |||
| * 设置微信消息处理器 | |||
| * | |||
| * @param handler | |||
| */ | |||
| public WxCpMessageRouterRule handler(WxCpMessageHandler handler) { | |||
| return handler(handler, (WxCpMessageHandler[]) null); | |||
| } | |||
| /** | |||
| * 设置微信消息处理器 | |||
| * | |||
| * @param handler | |||
| * @param otherHandlers | |||
| */ | |||
| public WxCpMessageRouterRule handler(WxCpMessageHandler handler, WxCpMessageHandler... otherHandlers) { | |||
| this.handlers.add(handler); | |||
| if (otherHandlers != null && otherHandlers.length > 0) { | |||
| for (WxCpMessageHandler i : otherHandlers) { | |||
| this.handlers.add(i); | |||
| } | |||
| } | |||
| return this; | |||
| } | |||
| /** | |||
| * 规则结束,代表如果一个消息匹配该规则,那么它将不再会进入其他规则 | |||
| */ | |||
| public WxCpMessageRouter end() { | |||
| this.routerBuilder.getRules().add(this); | |||
| return this.routerBuilder; | |||
| } | |||
| /** | |||
| * 规则结束,但是消息还会进入其他规则 | |||
| */ | |||
| public WxCpMessageRouter next() { | |||
| this.reEnter = true; | |||
| return end(); | |||
| } | |||
| protected boolean test(WxCpXmlMessage wxMessage) { | |||
| return | |||
| (this.fromUser == null || this.fromUser.equals(wxMessage.getFromUserName())) | |||
| && | |||
| (this.agentId == null || this.agentId.equals(wxMessage.getAgentId())) | |||
| && | |||
| (this.msgType == null || this.msgType.equals(wxMessage.getMsgType())) | |||
| && | |||
| (this.event == null || this.event.equals(wxMessage.getEvent())) | |||
| && | |||
| (this.eventKey == null || this.eventKey.equals(wxMessage.getEventKey())) | |||
| && | |||
| (this.content == null || this.content | |||
| .equals(wxMessage.getContent() == null ? null : wxMessage.getContent().trim())) | |||
| && | |||
| (this.rContent == null || Pattern | |||
| .matches(this.rContent, wxMessage.getContent() == null ? "" : wxMessage.getContent().trim())) | |||
| && | |||
| (this.matcher == null || this.matcher.match(wxMessage)) | |||
| ; | |||
| } | |||
| /** | |||
| * 处理微信推送过来的消息 | |||
| * | |||
| * @param wxMessage | |||
| * @return true 代表继续执行别的router,false 代表停止执行别的router | |||
| */ | |||
| protected WxCpXmlOutMessage service(WxCpXmlMessage wxMessage, | |||
| Map<String, Object> context, | |||
| WxCpService wxCpService, | |||
| WxSessionManager sessionManager, | |||
| WxErrorExceptionHandler exceptionHandler) { | |||
| if (context == null) { | |||
| context = new HashMap<>(); | |||
| } | |||
| try { | |||
| // 如果拦截器不通过 | |||
| for (WxCpMessageInterceptor interceptor : this.interceptors) { | |||
| if (!interceptor.intercept(wxMessage, context, wxCpService, sessionManager)) { | |||
| return null; | |||
| } | |||
| } | |||
| // 交给handler处理 | |||
| WxCpXmlOutMessage res = null; | |||
| for (WxCpMessageHandler handler : this.handlers) { | |||
| // 返回最后handler的结果 | |||
| res = handler.handle(wxMessage, context, wxCpService, sessionManager); | |||
| } | |||
| return res; | |||
| } catch (WxErrorException e) { | |||
| exceptionHandler.handle(e); | |||
| } | |||
| return null; | |||
| } | |||
| public void setFromUser(String fromUser) { | |||
| this.fromUser = fromUser; | |||
| } | |||
| public void setMsgType(String msgType) { | |||
| this.msgType = msgType; | |||
| } | |||
| public void setEvent(String event) { | |||
| this.event = event; | |||
| } | |||
| public void setEventKey(String eventKey) { | |||
| this.eventKey = eventKey; | |||
| } | |||
| public void setContent(String content) { | |||
| this.content = content; | |||
| } | |||
| public void setrContent(String rContent) { | |||
| this.rContent = rContent; | |||
| } | |||
| public void setMatcher(WxCpMessageMatcher matcher) { | |||
| this.matcher = matcher; | |||
| } | |||
| public void setAgentId(Integer agentId) { | |||
| this.agentId = agentId; | |||
| } | |||
| public void setHandlers(List<WxCpMessageHandler> handlers) { | |||
| this.handlers = handlers; | |||
| } | |||
| public void setInterceptors(List<WxCpMessageInterceptor> interceptors) { | |||
| this.interceptors = interceptors; | |||
| } | |||
| public boolean isAsync() { | |||
| return this.async; | |||
| } | |||
| public void setAsync(boolean async) { | |||
| this.async = async; | |||
| } | |||
| public boolean isReEnter() { | |||
| return this.reEnter; | |||
| } | |||
| public void setReEnter(boolean reEnter) { | |||
| this.reEnter = reEnter; | |||
| } | |||
| } | |||
| @@ -1,67 +0,0 @@ | |||
| package me.chanjar.weixin.cp.api; | |||
| import com.google.inject.Inject; | |||
| import me.chanjar.weixin.common.exception.WxErrorException; | |||
| import me.chanjar.weixin.cp.api.impl.WxCpServiceImpl; | |||
| import me.chanjar.weixin.cp.bean.WxCpDepart; | |||
| import org.testng.Assert; | |||
| import org.testng.annotations.Guice; | |||
| import org.testng.annotations.Test; | |||
| import java.util.List; | |||
| import static org.testng.Assert.assertNotNull; | |||
| import static org.testng.Assert.assertTrue; | |||
| /** | |||
| * 测试部门接口 | |||
| * | |||
| * @author Daniel Qian | |||
| */ | |||
| @Test(groups = "departAPI") | |||
| @Guice(modules = ApiTestModule.class) | |||
| public class WxCpDepartAPITest { | |||
| @Inject | |||
| protected WxCpServiceImpl wxCpService; | |||
| protected WxCpDepart depart; | |||
| @Test | |||
| public void testDepartCreate() throws WxErrorException { | |||
| WxCpDepart cpDepart = new WxCpDepart(); | |||
| cpDepart.setName("子部门" + System.currentTimeMillis()); | |||
| cpDepart.setParentId(1); | |||
| cpDepart.setOrder(1); | |||
| Integer departId = this.wxCpService.departCreate(cpDepart); | |||
| System.out.println(departId); | |||
| } | |||
| @Test//(dependsOnMethods = "testDepartCreate") | |||
| public void testDepartGet() throws WxErrorException { | |||
| System.out.println("=================获取部门"); | |||
| List<WxCpDepart> departList = this.wxCpService.departGet(); | |||
| assertNotNull(departList); | |||
| assertTrue(departList.size() > 0); | |||
| for (WxCpDepart g : departList) { | |||
| this.depart = g; | |||
| System.out.println(this.depart.getId() + ":" + this.depart.getName()); | |||
| assertNotNull(g.getName()); | |||
| } | |||
| } | |||
| @Test(dependsOnMethods = {"testDepartGet", "testDepartCreate"}) | |||
| public void testDepartUpdate() throws WxErrorException { | |||
| System.out.println("=================更新部门"); | |||
| this.depart.setName("子部门改名" + System.currentTimeMillis()); | |||
| this.wxCpService.departUpdate(this.depart); | |||
| } | |||
| @Test(dependsOnMethods = "testDepartUpdate") | |||
| public void testDepartDelete() throws WxErrorException { | |||
| System.out.println("=================删除部门"); | |||
| System.out.println(this.depart.getId() + ":" + this.depart.getName()); | |||
| this.wxCpService.departDelete(this.depart.getId()); | |||
| } | |||
| } | |||