test update

This commit is contained in:
tianfeng 2025-11-24 19:38:40 +08:00
parent ed2633ac0f
commit 323505ca48

View File

@ -27,9 +27,16 @@ import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.test.context.junit4.SpringRunner;
import java.text.SimpleDateFormat;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Collection;
import java.util.Date;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
@ -57,6 +64,8 @@ public class BatchSendBadgeTest {
private BadgeBackpackService badgeBackpackService;
@Autowired
private BadgeConfigService badgeConfigService;
@Autowired
private MongoTemplate mongoTemplate;
// 线程池大小
private static final int THREAD_POOL_SIZE = 10;
@ -69,10 +78,10 @@ public class BatchSendBadgeTest {
try {
// 1. 批量发放代理主播勋章 (HOST + AGENCY)
// sendHostAndAgencyBadges(executor);
sendHostAndAgencyBadges(executor);
// 2. 批量发放BD LEADER勋章
sendBdLeaderBadges(executor);
// sendBdLeaderBadges(executor);
// 3. 批量发放BD勋章
// sendBdBadges(executor);
@ -94,7 +103,10 @@ public class BatchSendBadgeTest {
* 发放代理主播勋章.
*/
private void sendHostAndAgencyBadges(ExecutorService executor) {
List<TeamMember> list = teamMemberService.list();
DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
LocalDateTime startDateTime = LocalDateTime.parse("2025-11-23 00:00:00", formatter);
List<TeamMember> list = mongoTemplate.find(Query.query(Criteria.where("createTime").gte(startDateTime)), TeamMember.class);
log.info("开始发放代理主播勋章,总数: {}", list.size());
BadgeConfig agencyConfig = badgeConfigService.getByBadgeKey(BadgeKeyEnum.AGENCY.name());