hhb@hotmail.com
3 months ago
parent
eb617c938f
commit
cf3f5d300b
@ -0,0 +1,56 @@
|
||||
package cn.teammodel.callAble;
|
||||
|
||||
import cn.teammodel.model.entity.school.LessonRecord;
|
||||
import cn.teammodel.repository.LessonRecordRepository;
|
||||
import com.azure.spring.data.cosmos.core.query.CosmosPageRequest;
|
||||
import org.springframework.data.domain.Slice;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.Callable;
|
||||
|
||||
public class LessonRecordQueryTask implements Callable<List<LessonRecord>> {
|
||||
private final LessonRecordRepository repository;
|
||||
private final String lessonRecordKey;
|
||||
private final Long startTime;
|
||||
private final Long endTime;
|
||||
private final String subjectId;
|
||||
private final String tmdId;
|
||||
private final String grade;
|
||||
private final String periodId;
|
||||
private static final int SLICE_SIZE = 1000;
|
||||
|
||||
public LessonRecordQueryTask(LessonRecordRepository repository, String lessonRecordKey,
|
||||
long startTime, long endTime,
|
||||
String subjectId, String tmdId, String grade,
|
||||
String periodId) {
|
||||
this.repository = repository;
|
||||
this.lessonRecordKey = lessonRecordKey;
|
||||
this.startTime = startTime;
|
||||
this.endTime = endTime;
|
||||
this.subjectId = subjectId;
|
||||
this.tmdId = tmdId;
|
||||
this.grade = grade;
|
||||
this.periodId = periodId;
|
||||
}
|
||||
|
||||
@Override
|
||||
public List<LessonRecord> call() throws Exception {
|
||||
List<LessonRecord> result = new ArrayList<>();
|
||||
CosmosPageRequest pageRequest = new CosmosPageRequest(0, SLICE_SIZE, null);
|
||||
|
||||
while (true) {
|
||||
Slice<LessonRecord> slice = repository.getLessonsByConditions(
|
||||
lessonRecordKey, startTime, endTime, subjectId, tmdId, grade, periodId, pageRequest
|
||||
);
|
||||
result.addAll(slice.getContent());
|
||||
|
||||
if (!slice.hasNext()) {
|
||||
break;
|
||||
}
|
||||
pageRequest = (CosmosPageRequest) slice.nextPageable();
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,70 @@
|
||||
package cn.teammodel.test;
|
||||
|
||||
import cn.teammodel.callAble.LessonRecordQueryTask;
|
||||
import cn.teammodel.model.entity.school.LessonRecord;
|
||||
import cn.teammodel.repository.LessonRecordRepository;
|
||||
import com.azure.spring.data.cosmos.core.query.CosmosPageRequest;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.temporal.ChronoUnit;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.*;
|
||||
|
||||
public class LessonRecordQueryService {
|
||||
//private static final int SLICE_SIZE = 1000; // 分页大小
|
||||
private static final int THREAD_POOL_SIZE = 4; // 线程池大小
|
||||
|
||||
private final LessonRecordRepository repository;
|
||||
private final ExecutorService executor;
|
||||
|
||||
public LessonRecordQueryService(LessonRecordRepository repository) {
|
||||
this.repository = repository;
|
||||
this.executor = Executors.newFixedThreadPool(THREAD_POOL_SIZE);
|
||||
}
|
||||
|
||||
public List<LessonRecord> queryLessonsInParallel(String lessonRecordKey,
|
||||
Long startTime,
|
||||
Long endTime,
|
||||
String subjectId,
|
||||
String tmdId,
|
||||
String grade,
|
||||
String periodId) throws InterruptedException, ExecutionException {
|
||||
List<Future<List<LessonRecord>>> futures = new ArrayList<>();
|
||||
|
||||
// 按时间范围拆分任务
|
||||
long timeRange = endTime - startTime;
|
||||
int numTasks = THREAD_POOL_SIZE; // 拆分为与线程池大小相同的任务数
|
||||
|
||||
for (int i = 0; i < numTasks; i++) {
|
||||
long chunkStart = startTime + (i * timeRange / numTasks);
|
||||
long chunkEnd = (i == numTasks - 1) ? endTime : startTime + ((i + 1) * timeRange / numTasks);
|
||||
|
||||
//CosmosPageRequest pageRequest = new CosmosPageRequest(0, SLICE_SIZE, null);
|
||||
LessonRecordQueryTask task = new LessonRecordQueryTask(
|
||||
repository, lessonRecordKey, chunkStart, chunkEnd,
|
||||
subjectId, tmdId, grade, periodId
|
||||
);
|
||||
futures.add(executor.submit(task));
|
||||
}
|
||||
|
||||
// 合并结果
|
||||
List<LessonRecord> allRecords = new ArrayList<>();
|
||||
for (Future<List<LessonRecord>> future : futures) {
|
||||
allRecords.addAll(future.get());
|
||||
}
|
||||
|
||||
return allRecords;
|
||||
}
|
||||
|
||||
public void shutdown() {
|
||||
executor.shutdown();
|
||||
try {
|
||||
if (!executor.awaitTermination(60, TimeUnit.SECONDS)) {
|
||||
executor.shutdownNow(); // 强制关闭未完成的任务
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
executor.shutdownNow();
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue