Merge branch 'master' of https://codeup.aliyun.com/64d9c5feeceb191898f636d7/honor-dd-light-ds-java
wangchaoxu
1 year ago
commit
48974cd1c2
@ -0,0 +1,27 @@
|
|||||||
|
package com.ms.api.spi.timer;
|
||||||
|
|
||||||
|
import com.jinritemai.cloud.base.api.BaseRequest;
|
||||||
|
import com.jinritemai.cloud.base.api.BaseResponse;
|
||||||
|
import com.jinritemai.cloud.base.api.ExtensionService;
|
||||||
|
import com.jinritemai.cloud.base.api.ExtensionServiceHandler;
|
||||||
|
import com.ms.api.common.R;
|
||||||
|
import com.ms.api.common.Ret;
|
||||||
|
import com.ms.api.common.TimerBaseService;
|
||||||
|
import com.ms.api.dto.ItemDTO;
|
||||||
|
import com.ms.api.task.CheckAuditStatusTimeoutTaskService;
|
||||||
|
import com.ms.api.task.RefreshTokenTimerTaskService;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
|
||||||
|
@ExtensionService("RefreshTokenTimer")
|
||||||
|
@Slf4j
|
||||||
|
public class RefreshTokenTimerService extends TimerBaseService implements ExtensionServiceHandler<ItemDTO, Ret> {
|
||||||
|
@Autowired
|
||||||
|
RefreshTokenTimerTaskService refreshTokenTimerTaskService;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BaseResponse<Ret> handle(BaseRequest<ItemDTO> req) {
|
||||||
|
refreshTokenTimerTaskService.runTask();
|
||||||
|
return R.ok(Ret.success());
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,141 @@
|
|||||||
|
package com.ms.api.task;
|
||||||
|
|
||||||
|
import cn.hutool.core.date.DateUtil;
|
||||||
|
import cn.hutool.core.util.ObjectUtil;
|
||||||
|
import com.doudian.open.api.token_refresh.TokenRefreshRequest;
|
||||||
|
import com.doudian.open.api.token_refresh.TokenRefreshResponse;
|
||||||
|
import com.doudian.open.api.token_refresh.param.TokenRefreshParam;
|
||||||
|
import com.ms.api.common.E;
|
||||||
|
import com.ms.api.common.R;
|
||||||
|
import com.ms.api.common.Ret;
|
||||||
|
import com.ms.api.common.TaskBaseService;
|
||||||
|
import com.ms.api.consts.StatusConst;
|
||||||
|
import com.ms.api.service.MaterialAuditStatusBufferService;
|
||||||
|
import com.ms.api.service.ShopService;
|
||||||
|
import com.ms.api.tool.CommonTool;
|
||||||
|
import com.ms.api.tool.SecurityTool;
|
||||||
|
import com.ms.api.tool.TimeTool;
|
||||||
|
import com.ms.dal.entity.MaterialAuditStatusBuffer;
|
||||||
|
import com.ms.dal.entity.MaterialBizAuditStatusLog;
|
||||||
|
import com.ms.dal.entity.Shop;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.apache.http.client.utils.DateUtils;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
|
||||||
|
import javax.annotation.Resource;
|
||||||
|
import java.sql.Time;
|
||||||
|
import java.util.*;
|
||||||
|
import java.util.concurrent.Executor;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 刷新抖店token
|
||||||
|
*/
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
@Component
|
||||||
|
@Slf4j
|
||||||
|
public class RefreshTokenTimerTaskService extends TaskBaseService {
|
||||||
|
@Autowired
|
||||||
|
ShopService shopService;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 同时开启任务处理数量
|
||||||
|
*/
|
||||||
|
public int getCorePoolSiz() {
|
||||||
|
return 3;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 任务管理器名称
|
||||||
|
*/
|
||||||
|
public String getTaskExecutorName() {
|
||||||
|
return "RefreshTokenTimerTaskPool";
|
||||||
|
}
|
||||||
|
|
||||||
|
@Bean(name = "RefreshTokenTimerTaskPool")
|
||||||
|
@Override
|
||||||
|
public Executor getAsyncExecutor() {
|
||||||
|
return super.getAsyncExecutor();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Resource(name = "RefreshTokenTimerTaskPool")
|
||||||
|
protected Executor taskPool;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Executor getTaskPool() {
|
||||||
|
return taskPool;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void runTask() {
|
||||||
|
super.runTask();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取任务
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public Object getTask() {
|
||||||
|
String gmtExpireStart = TimeTool.getDateTimeBySeconds(3600);
|
||||||
|
String gmtExpireEnd = TimeTool.getDateTimeBySeconds(-3600);
|
||||||
|
List<Shop> shops = shopService.getExpiresTokenShops(gmtExpireStart, gmtExpireEnd);
|
||||||
|
return shops;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 处理任务
|
||||||
|
*
|
||||||
|
* @param params
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public Object processTask(Object params) {
|
||||||
|
List<Shop> shops = (List<Shop>) params;
|
||||||
|
for (int i=0; i < shops.size(); i++) {
|
||||||
|
Shop shop = shops.get(i);
|
||||||
|
Long shopId = shop.getShopId();
|
||||||
|
// 不理解
|
||||||
|
// $failCnt = $this->redis->hGet($redisKey, $shopId);
|
||||||
|
// if ($failCnt > 5) {
|
||||||
|
// continue;
|
||||||
|
// }
|
||||||
|
String oldAccessToken = SecurityTool.decodeByAES(shop.getAccessToken());
|
||||||
|
String oldRefreshToken = SecurityTool.decodeByAES(shop.getRefreshToken());
|
||||||
|
TokenRefreshRequest request = new TokenRefreshRequest();
|
||||||
|
TokenRefreshParam param = request.getParam();
|
||||||
|
param.setRefreshToken(oldRefreshToken);
|
||||||
|
param.setGrantType("refresh_token");
|
||||||
|
TokenRefreshResponse response = request.execute();
|
||||||
|
String newAccessToken = response.getData().getAccessToken();
|
||||||
|
String newRefreshToken = response.getData().getRefreshToken();
|
||||||
|
Long newExpires = response.getData().getExpiresIn();
|
||||||
|
if (ObjectUtil.isEmpty(newAccessToken)) {
|
||||||
|
E.throwMSException("获取accessToken失败");
|
||||||
|
}
|
||||||
|
if (ObjectUtil.isEmpty(newAccessToken)) {
|
||||||
|
Map result = new HashMap();
|
||||||
|
result.put("msg", "accessToken无需更新");
|
||||||
|
return R.ok(Ret.success());
|
||||||
|
}
|
||||||
|
shop.setExpiresIn(newExpires.toString());
|
||||||
|
shop.setRefreshToken(SecurityTool.encodeByAES(newRefreshToken));
|
||||||
|
shop.setAccessToken(SecurityTool.encodeByAES(newAccessToken));
|
||||||
|
shop.setGmtExpire(new Date(TimeTool.getDateTimeBySeconds(new Long(newExpires).intValue())));
|
||||||
|
shop.setGmtModified(new Date(TimeTool.getDateTimeBySeconds(0)));
|
||||||
|
shopService.updateByPrimaryKey(shop);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 清理任务
|
||||||
|
*
|
||||||
|
* @param params
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public void clearTask(Object params) {
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,26 @@
|
|||||||
|
package com.ms.api.tool;
|
||||||
|
|
||||||
|
import cn.hutool.core.date.DateUtil;
|
||||||
|
import com.ms.api.common.Ret;
|
||||||
|
import com.ms.api.common.StrObjMap;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.stereotype.Component;
|
||||||
|
|
||||||
|
import javax.imageio.ImageIO;
|
||||||
|
import java.awt.image.BufferedImage;
|
||||||
|
import java.io.ByteArrayOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.net.URL;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
@Component
|
||||||
|
public class TimeTool {
|
||||||
|
public static String getDateTimeBySeconds(int seconds) {
|
||||||
|
String currentTime = DateUtil.now();
|
||||||
|
// 将当前时间转换为Date类型
|
||||||
|
java.util.Date date = DateUtil.parseDateTime(currentTime);
|
||||||
|
// 将当前时间减去x秒
|
||||||
|
java.util.Date newDate = DateUtil.offsetSecond(date, seconds);
|
||||||
|
return DateUtil.formatDateTime(newDate);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue