aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

BUGFIX:用户订阅后更新用户身份和积分

Browse Source
This commit is contained in:
xupei
2024-12-27 14:29:07 +08:00
parent f45bd7acc4
commit ca4d75c63f
3 changed files with 42 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/ai/da/service/AccountService.java
View File

@@ -218,4 +218,8 @@ public interface AccountService extends IService<Account> {
Boolean unbindWeChat();
Boolean unbindGoogle();
void updateAccountValidity(Long accountId, Long currentPeriodEnd);
void updateUserRoleAndCredits(Long accountId, String type);
}

28
src/main/java/com/ai/da/service/impl/AccountServiceImpl.java
View File

@@ -2591,4 +2591,32 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
accountMapper.updateById(account);
return Boolean.TRUE;
}
public void updateAccountValidity(Long accountId, Long currentPeriodEnd){
// 不管当前用户的账号是否到期，都根据付款信息重置账号到期时间
Account account = accountMapper.selectById(accountId);
account.setValidEndTime(currentPeriodEnd * 1000);
accountMapper.updateById(account);
}
public void updateUserRoleAndCredits(Long accountId, String type){
Account account = accountMapper.selectById(accountId);
switch (type) {
case "month":
account.setSystemUser(2);
account.setCredits(BigDecimal.valueOf(Long.parseLong(CreditsEventsEnum.INIT_MONTHLY.getValue())));
break;
case "year":
account.setSystemUser(1);
account.setCredits(BigDecimal.valueOf(Long.parseLong(CreditsEventsEnum.INIT_YEARLY.getValue())));
break;
case "day":
account.setSystemUser(3);
account.setCredits(BigDecimal.valueOf(Long.parseLong(CreditsEventsEnum.INIT_WEEKLY.getValue())));
break;
}
accountMapper.updateById(account);
}
}

20
src/main/java/com/ai/da/service/impl/StripeServiceImpl.java
View File

@@ -58,6 +58,8 @@ public class StripeServiceImpl implements StripeService {
private CreditsService creditsService;
@Resource
private RefundInfoService refundInfoService;
@Resource
private AccountService accountService;
@Resource
private AccountMapper accountMapper;
@@ -368,6 +370,7 @@ public class StripeServiceImpl implements StripeService {
orderInfoService.updateById(orderInfo);
}
}
log.info("回调事件 {} 处理完成", event.getType());
return response;
}
@@ -466,7 +469,10 @@ public class StripeServiceImpl implements StripeService {
subscriptionInfoMapper.insert(subscriptionInfo);
// 更新账号到期时间
updateAccountValidity(subscriptionInfo.getAccountId(), subscriptionInfo.getCurrentPeriodEnd());
accountService.updateAccountValidity(subscriptionInfo.getAccountId(), subscriptionInfo.getCurrentPeriodEnd());
// 更新账号身份和积分
accountService.updateUserRoleAndCredits(subscriptionInfo.getAccountId(), interval);
}
return subscriptionInfo;
}
@@ -519,7 +525,9 @@ public class StripeServiceImpl implements StripeService {
subscriptionInfo.setCurrentPeriodEnd(subscription.getCurrentPeriodEnd());
subscriptionInfo.setNextPayDate(DateUtil.changeTimeStampFormat(subscription.getCurrentPeriodEnd(), "seconds", CommonConstant.TIME_FORMAT_MMM_dd_yyyy_EEEE));
// 更新账号到期时间
updateAccountValidity(subscriptionInfo.getAccountId(), subscriptionInfo.getCurrentPeriodEnd());
accountService.updateAccountValidity(subscriptionInfo.getAccountId(), subscriptionInfo.getCurrentPeriodEnd());
// 更新账号身份和积分
accountService.updateUserRoleAndCredits(subscriptionInfo.getAccountId(), subscriptionInfo.getType());
log.info("更新 {} 账号到期时间为：{}", subscriptionInfo.getAccountId(), DateUtil.changeTimeStampFormat(subscriptionInfo.getCurrentPeriodEnd(), "seconds", CommonConstant.TIME_FORMAT_MMM_dd_yyyy_EEEE));
flag = true;
}
@@ -530,14 +538,6 @@ public class StripeServiceImpl implements StripeService {
return subscriptionInfo;
}
private void updateAccountValidity(Long accountId, Long currentPeriodEnd){
// 不管当前用户的账号是否到期，都根据付款信息重置账号到期时间
com.ai.da.mapper.primary.entity.Account account = accountMapper.selectById(accountId);
account.setValidEndTime(currentPeriodEnd * 1000);
accountMapper.updateById(account);
}
// 取消连续订阅 将订阅从pause状态转为cancel状态（使用定时器，定期检索DB中，过期且不续订的订阅）
public void cancelSubscription(String subscriptionId, String cancelReason) {
Stripe.apiKey = privateKey;