添加查询Code-Create数据库，监控新增付费记录并对付费用户进行账号新增或延长账号有效期的操作
This commit is contained in:
@@ -4,6 +4,7 @@ import cn.hutool.core.collection.CollectionUtil;
|
||||
import com.ai.da.common.config.exception.BusinessException;
|
||||
import com.ai.da.common.context.UserContext;
|
||||
import com.ai.da.common.enums.AuthenticationOperationTypeEnum;
|
||||
import com.ai.da.common.enums.CreditsEventsEnum;
|
||||
import com.ai.da.common.enums.LoginTypeEnum;
|
||||
import com.ai.da.common.response.ResultEnum;
|
||||
import com.ai.da.common.security.jwt.JWTTokenHelper;
|
||||
@@ -22,6 +23,7 @@ import com.ai.da.model.vo.QuestionnaireVO;
|
||||
import com.ai.da.service.*;
|
||||
import com.alibaba.fastjson.JSON;
|
||||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
||||
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
||||
@@ -383,15 +385,18 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
|
||||
@Override
|
||||
public Boolean addUser(AccountAddDTO accountAddDTO) {
|
||||
Account account = new Account();
|
||||
account.setUserEmail(accountAddDTO.getEmail());
|
||||
account.setUserPassword("Third-000000");
|
||||
account.setUserName(accountAddDTO.getUserName());
|
||||
Account account;
|
||||
account = CopyUtil.copyObject(accountAddDTO,Account.class);
|
||||
// account.setUserEmail(accountAddDTO.getUserEmail());
|
||||
// account.setUserName(accountAddDTO.getUserName());
|
||||
// account.setIsTrial(accountAddDTO.getIsTrial());
|
||||
account.setValidStartTime(Long.valueOf(accountAddDTO.getValidStartTime()));
|
||||
account.setValidEndTime(Long.valueOf(accountAddDTO.getValidEndTime()));
|
||||
account.setUserPassword("Third-000000");
|
||||
account.setLanguage(Language.ENGLISH.name());
|
||||
account.setCreateDate(new Date());
|
||||
account.setIsTrial(accountAddDTO.getIsTrial());
|
||||
account.setSystemUser(1);
|
||||
account.setIsTrial(0);
|
||||
account.setIsBeginner(1);
|
||||
return accountMapper.insert(account) > 0;
|
||||
}
|
||||
|
||||
@@ -1010,10 +1015,10 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
}
|
||||
|
||||
private static final String QUERY_ORDER = "SELECT * FROM pmr_wc_order_stats " +
|
||||
"WHERE status = 'wc-processing' AND date_paid > '2024-06-10 00:00:00'" +
|
||||
"WHERE status = 'wc-processing' AND date_paid > '2024-06-20 00:00:00'" +
|
||||
" ORDER BY order_id DESC ";
|
||||
|
||||
private static final String QUERY_CUSTOMER_EMAIL = "SELECT username, email FROM pmr_wc_customer_lookup " +
|
||||
private static final String QUERY_CUSTOMER_EMAIL = "SELECT username, email, country FROM pmr_wc_customer_lookup " +
|
||||
"WHERE customer_id = ? ";
|
||||
private static final String UPDATE_ORDER_STATUS = "UPDATE pmr_wc_order_stats " +
|
||||
"SET status = 'wc-complete' , date_completed = ? " +
|
||||
@@ -1037,7 +1042,9 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
*
|
||||
* @return null
|
||||
*/
|
||||
public String extendValidityForCC() {
|
||||
@Override
|
||||
@Transactional(rollbackFor = Exception.class)
|
||||
public Boolean extendValidityForCC() {
|
||||
try (Connection connection = dataSource.getConnection();
|
||||
PreparedStatement preparedStatement = connection.prepareStatement(QUERY_ORDER)) {
|
||||
// preparedStatement.setString(1, "someCondition");
|
||||
@@ -1049,6 +1056,8 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
double totalSales = queryOrderResultSet.getDouble("total_sales");
|
||||
String email = "";
|
||||
String userName = "";
|
||||
// 为什么一般没有值
|
||||
String country = "";
|
||||
// 1、查pmr_wc_customer_lookup表；确认当前订单对应的用户邮箱
|
||||
PreparedStatement preparedQueryEmail = connection.prepareStatement(QUERY_CUSTOMER_EMAIL);
|
||||
preparedQueryEmail.setInt(1, customerId);
|
||||
@@ -1056,6 +1065,7 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
if (queryEmailResultSet.next()) {
|
||||
email = queryEmailResultSet.getString("email");
|
||||
userName = queryEmailResultSet.getString("username");
|
||||
country = queryEmailResultSet.getString("country");
|
||||
} else {
|
||||
log.error("未知错误。code-create的用户表中没有付费用户的信息");
|
||||
throw new BusinessException("user info missing");
|
||||
@@ -1083,7 +1093,27 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
|
||||
if (flag) {
|
||||
// 是新用户 => 新增一条数据
|
||||
Boolean b = addUser(new AccountAddDTO(email, StringUtil.isNullOrEmpty(userName) ? email.substring(0, email.indexOf("@") - 1) : userName, account.getValidStartTime().toString(), account.getValidEndTime().toString(), 0));
|
||||
String credits = "0";
|
||||
int systemUserType = 0;
|
||||
if (totalSales == 5000.0){
|
||||
log.info("年付用户，初始积分6000");
|
||||
credits = CreditsEventsEnum.INIT_MONTHLY.getValue();
|
||||
systemUserType = 1;
|
||||
}else if (totalSales == 500.0){
|
||||
log.info("月付用户，初始积分5000");
|
||||
credits = CreditsEventsEnum.INIT_MONTHLY.getValue();
|
||||
systemUserType = 2;
|
||||
}else if (totalSales == 0.0){
|
||||
log.info("测试用户，初始积分10");
|
||||
credits = "10";
|
||||
systemUserType = 3;
|
||||
}
|
||||
|
||||
Boolean b = addUser(new AccountAddDTO(email,
|
||||
StringUtil.isNullOrEmpty(userName) ? email.substring(0, email.indexOf("@")) : userName,
|
||||
country,
|
||||
account.getValidStartTime().toString(),
|
||||
account.getValidEndTime().toString(), 0,new BigDecimal(credits),systemUserType));
|
||||
if (b) log.info("付费新用户新增成功！");
|
||||
} else {
|
||||
userInfo.setValidEndTime(account.getValidEndTime());
|
||||
@@ -1112,9 +1142,10 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
} catch (Exception e) {
|
||||
// 记录异常并处理
|
||||
e.printStackTrace();
|
||||
return Boolean.FALSE;
|
||||
}
|
||||
|
||||
return null;
|
||||
return Boolean.TRUE;
|
||||
}
|
||||
|
||||
private Account extendValidity(Long validEndTime, double totalSales) {
|
||||
@@ -1199,4 +1230,17 @@ public class AccountServiceImpl extends ServiceImpl<AccountMapper, Account> impl
|
||||
return Boolean.TRUE;
|
||||
}
|
||||
|
||||
/**
|
||||
* 为年费用户每月更新积分
|
||||
*/
|
||||
public void refreshCreditsMonthly(){
|
||||
UpdateWrapper<Account> accountUpdateWrapper = new UpdateWrapper<>();
|
||||
// 刷新账号有效期截止之前的年付用户的积分
|
||||
long epochMilli = Instant.now().toEpochMilli();
|
||||
accountUpdateWrapper.lambda().set(Account::getCredits, CreditsEventsEnum.INIT_YEARLY.getValue())
|
||||
.eq(Account::getSystemUser,1)
|
||||
.gt(Account::getValidEndTime, epochMilli);
|
||||
baseMapper.update(null,accountUpdateWrapper);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user