初始化游戏小组管理系统后端项目
- 基于 NestJS + TypeScript + MySQL + Redis 架构 - 完整的模块化设计（认证、用户、小组、游戏、预约等） - JWT 认证和 RBAC 权限控制系统 - Docker 容器化部署支持 - 添加 CLAUDE.md 项目开发指南 - 配置 .gitignore 忽略文件 Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
UGREEN USER
446
src/modules/schedules/schedules.service.ts
Normal file
446
src/modules/schedules/schedules.service.ts
Normal file
@@ -0,0 +1,446 @@
|
||||
import {
|
||||
Injectable,
|
||||
NotFoundException,
|
||||
ForbiddenException,
|
||||
BadRequestException,
|
||||
} from '@nestjs/common';
|
||||
import { InjectRepository } from '@nestjs/typeorm';
|
||||
import { Repository, Between } from 'typeorm';
|
||||
import { Schedule } from '../../entities/schedule.entity';
|
||||
import { Group } from '../../entities/group.entity';
|
||||
import { GroupMember } from '../../entities/group-member.entity';
|
||||
import {
|
||||
CreateScheduleDto,
|
||||
UpdateScheduleDto,
|
||||
QuerySchedulesDto,
|
||||
FindCommonSlotsDto,
|
||||
} from './dto/schedule.dto';
|
||||
import { ErrorCode, ErrorMessage } from '../../common/interfaces/response.interface';
|
||||
import { PaginationUtil } from '../../common/utils/pagination.util';
|
||||
|
||||
export interface TimeSlot {
|
||||
startTime: Date;
|
||||
endTime: Date;
|
||||
note?: string;
|
||||
}
|
||||
|
||||
export interface CommonSlot {
|
||||
startTime: Date;
|
||||
endTime: Date;
|
||||
participants: string[];
|
||||
participantCount: number;
|
||||
}
|
||||
|
||||
@Injectable()
|
||||
export class SchedulesService {
|
||||
constructor(
|
||||
@InjectRepository(Schedule)
|
||||
private scheduleRepository: Repository<Schedule>,
|
||||
@InjectRepository(Group)
|
||||
private groupRepository: Repository<Group>,
|
||||
@InjectRepository(GroupMember)
|
||||
private groupMemberRepository: Repository<GroupMember>,
|
||||
) {}
|
||||
|
||||
/**
|
||||
* 创建排班
|
||||
*/
|
||||
async create(userId: string, createDto: CreateScheduleDto) {
|
||||
const { groupId, availableSlots, ...rest } = createDto;
|
||||
|
||||
// 验证小组是否存在
|
||||
const group = await this.groupRepository.findOne({
|
||||
where: { id: groupId, isActive: true },
|
||||
});
|
||||
if (!group) {
|
||||
throw new NotFoundException({
|
||||
code: ErrorCode.GROUP_NOT_FOUND,
|
||||
message: ErrorMessage[ErrorCode.GROUP_NOT_FOUND],
|
||||
});
|
||||
}
|
||||
|
||||
// 验证用户权限
|
||||
const membership = await this.groupMemberRepository.findOne({
|
||||
where: { groupId, userId, isActive: true },
|
||||
});
|
||||
if (!membership) {
|
||||
throw new ForbiddenException({
|
||||
code: ErrorCode.NOT_IN_GROUP,
|
||||
message: ErrorMessage[ErrorCode.NOT_IN_GROUP],
|
||||
});
|
||||
}
|
||||
|
||||
// 验证时间段
|
||||
this.validateTimeSlots(availableSlots);
|
||||
|
||||
// 创建排班
|
||||
const schedule = this.scheduleRepository.create({
|
||||
...rest,
|
||||
groupId,
|
||||
userId,
|
||||
availableSlots: availableSlots as any,
|
||||
});
|
||||
|
||||
await this.scheduleRepository.save(schedule);
|
||||
|
||||
return this.findOne(schedule.id);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取排班列表
|
||||
*/
|
||||
async findAll(userId: string, queryDto: QuerySchedulesDto) {
|
||||
const {
|
||||
groupId,
|
||||
userId: targetUserId,
|
||||
startTime,
|
||||
endTime,
|
||||
page = 1,
|
||||
limit = 10,
|
||||
} = queryDto;
|
||||
const { offset } = PaginationUtil.formatPaginationParams(page, limit);
|
||||
|
||||
const queryBuilder = this.scheduleRepository
|
||||
.createQueryBuilder('schedule')
|
||||
.leftJoinAndSelect('schedule.group', 'group')
|
||||
.leftJoinAndSelect('schedule.user', 'user');
|
||||
|
||||
// 筛选条件
|
||||
if (groupId) {
|
||||
// 验证用户是否在小组中
|
||||
await this.checkGroupMembership(userId, groupId);
|
||||
queryBuilder.andWhere('schedule.groupId = :groupId', { groupId });
|
||||
} else {
|
||||
// 如果没有指定小组，只返回用户所在小组的排班
|
||||
const memberGroups = await this.groupMemberRepository.find({
|
||||
where: { userId, isActive: true },
|
||||
select: ['groupId'],
|
||||
});
|
||||
const groupIds = memberGroups.map((m) => m.groupId);
|
||||
if (groupIds.length === 0) {
|
||||
return {
|
||||
items: [],
|
||||
total: 0,
|
||||
page,
|
||||
limit,
|
||||
totalPages: 0,
|
||||
};
|
||||
}
|
||||
queryBuilder.andWhere('schedule.groupId IN (:...groupIds)', { groupIds });
|
||||
}
|
||||
|
||||
if (targetUserId) {
|
||||
queryBuilder.andWhere('schedule.userId = :userId', { userId: targetUserId });
|
||||
}
|
||||
|
||||
if (startTime && endTime) {
|
||||
queryBuilder.andWhere('schedule.createdAt BETWEEN :startTime AND :endTime', {
|
||||
startTime: new Date(startTime),
|
||||
endTime: new Date(endTime),
|
||||
});
|
||||
}
|
||||
|
||||
// 分页
|
||||
const [items, total] = await queryBuilder
|
||||
.orderBy('schedule.createdAt', 'DESC')
|
||||
.skip(offset)
|
||||
.take(limit)
|
||||
.getManyAndCount();
|
||||
|
||||
// 解析 availableSlots
|
||||
const formattedItems = items.map((item) => ({
|
||||
...item,
|
||||
availableSlots: this.normalizeAvailableSlots(item.availableSlots),
|
||||
}));
|
||||
|
||||
return {
|
||||
items: formattedItems,
|
||||
total,
|
||||
page,
|
||||
limit,
|
||||
totalPages: PaginationUtil.getTotalPages(total, limit),
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取排班详情
|
||||
*/
|
||||
async findOne(id: string) {
|
||||
const schedule = await this.scheduleRepository.findOne({
|
||||
where: { id },
|
||||
relations: ['group', 'user'],
|
||||
});
|
||||
|
||||
if (!schedule) {
|
||||
throw new NotFoundException({
|
||||
code: ErrorCode.NOT_FOUND,
|
||||
message: '排班不存在',
|
||||
});
|
||||
}
|
||||
|
||||
return {
|
||||
...schedule,
|
||||
availableSlots: this.normalizeAvailableSlots(schedule.availableSlots),
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新排班
|
||||
*/
|
||||
async update(userId: string, id: string, updateDto: UpdateScheduleDto) {
|
||||
const schedule = await this.scheduleRepository.findOne({
|
||||
where: { id },
|
||||
});
|
||||
|
||||
if (!schedule) {
|
||||
throw new NotFoundException({
|
||||
code: ErrorCode.NOT_FOUND,
|
||||
message: '排班不存在',
|
||||
});
|
||||
}
|
||||
|
||||
// 只有创建者可以修改
|
||||
if (schedule.userId !== userId) {
|
||||
throw new ForbiddenException({
|
||||
code: ErrorCode.NO_PERMISSION,
|
||||
message: ErrorMessage[ErrorCode.NO_PERMISSION],
|
||||
});
|
||||
}
|
||||
|
||||
if (updateDto.availableSlots) {
|
||||
this.validateTimeSlots(updateDto.availableSlots);
|
||||
updateDto.availableSlots = updateDto.availableSlots as any;
|
||||
}
|
||||
|
||||
Object.assign(schedule, updateDto);
|
||||
await this.scheduleRepository.save(schedule);
|
||||
|
||||
return this.findOne(id);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除排班
|
||||
*/
|
||||
async remove(userId: string, id: string) {
|
||||
const schedule = await this.scheduleRepository.findOne({
|
||||
where: { id },
|
||||
});
|
||||
|
||||
if (!schedule) {
|
||||
throw new NotFoundException({
|
||||
code: ErrorCode.NOT_FOUND,
|
||||
message: '排班不存在',
|
||||
});
|
||||
}
|
||||
|
||||
// 只有创建者可以删除
|
||||
if (schedule.userId !== userId) {
|
||||
throw new ForbiddenException({
|
||||
code: ErrorCode.NO_PERMISSION,
|
||||
message: ErrorMessage[ErrorCode.NO_PERMISSION],
|
||||
});
|
||||
}
|
||||
|
||||
await this.scheduleRepository.remove(schedule);
|
||||
|
||||
return { message: '排班已删除' };
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找共同空闲时间
|
||||
*/
|
||||
async findCommonSlots(userId: string, findDto: FindCommonSlotsDto) {
|
||||
const { groupId, startTime, endTime, minParticipants = 2 } = findDto;
|
||||
|
||||
// 验证用户权限
|
||||
await this.checkGroupMembership(userId, groupId);
|
||||
|
||||
// 获取时间范围内的所有排班
|
||||
const schedules = await this.scheduleRepository.find({
|
||||
where: { groupId },
|
||||
relations: ['user'],
|
||||
});
|
||||
|
||||
if (schedules.length === 0) {
|
||||
return {
|
||||
commonSlots: [],
|
||||
message: '暂无排班数据',
|
||||
};
|
||||
}
|
||||
|
||||
// 解析所有时间段
|
||||
const userSlots: Map<string, TimeSlot[]> = new Map();
|
||||
schedules.forEach((schedule) => {
|
||||
const slots = this.normalizeAvailableSlots(schedule.availableSlots);
|
||||
const filteredSlots = slots.filter((slot) => {
|
||||
const slotStart = new Date(slot.startTime);
|
||||
const slotEnd = new Date(slot.endTime);
|
||||
const rangeStart = new Date(startTime);
|
||||
const rangeEnd = new Date(endTime);
|
||||
return slotStart >= rangeStart && slotEnd <= rangeEnd;
|
||||
});
|
||||
userSlots.set(schedule.userId, filteredSlots);
|
||||
});
|
||||
|
||||
// 计算时间交集
|
||||
const commonSlots = this.calculateCommonSlots(userSlots, minParticipants);
|
||||
|
||||
// 按参与人数排序
|
||||
commonSlots.sort((a, b) => b.participantCount - a.participantCount);
|
||||
|
||||
return {
|
||||
commonSlots,
|
||||
totalParticipants: schedules.length,
|
||||
minParticipants,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 计算共同空闲时间
|
||||
*/
|
||||
private calculateCommonSlots(
|
||||
userSlots: Map<string, TimeSlot[]>,
|
||||
minParticipants: number,
|
||||
): CommonSlot[] {
|
||||
const allSlots: Array<{ time: Date; userId: string; type: 'start' | 'end' }> = [];
|
||||
|
||||
// 收集所有时间点
|
||||
userSlots.forEach((slots, userId) => {
|
||||
slots.forEach((slot) => {
|
||||
allSlots.push({
|
||||
time: new Date(slot.startTime),
|
||||
userId,
|
||||
type: 'start',
|
||||
});
|
||||
allSlots.push({
|
||||
time: new Date(slot.endTime),
|
||||
userId,
|
||||
type: 'end',
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
// 按时间排序
|
||||
allSlots.sort((a, b) => a.time.getTime() - b.time.getTime());
|
||||
|
||||
// 扫描线算法计算重叠区间
|
||||
const commonSlots: CommonSlot[] = [];
|
||||
const activeUsers = new Set<string>();
|
||||
let lastTime: Date | null = null;
|
||||
|
||||
allSlots.forEach((event) => {
|
||||
if (lastTime && activeUsers.size >= minParticipants) {
|
||||
// 记录共同空闲时间段
|
||||
if (event.time.getTime() > lastTime.getTime()) {
|
||||
commonSlots.push({
|
||||
startTime: lastTime,
|
||||
endTime: event.time,
|
||||
participants: Array.from(activeUsers),
|
||||
participantCount: activeUsers.size,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
if (event.type === 'start') {
|
||||
activeUsers.add(event.userId);
|
||||
} else {
|
||||
activeUsers.delete(event.userId);
|
||||
}
|
||||
|
||||
lastTime = event.time;
|
||||
});
|
||||
|
||||
// 合并相邻的时间段
|
||||
return this.mergeAdjacentSlots(commonSlots);
|
||||
}
|
||||
|
||||
/**
|
||||
* 合并相邻的时间段
|
||||
*/
|
||||
private mergeAdjacentSlots(slots: CommonSlot[]): CommonSlot[] {
|
||||
if (slots.length === 0) return [];
|
||||
|
||||
const merged: CommonSlot[] = [];
|
||||
let current = slots[0];
|
||||
|
||||
for (let i = 1; i < slots.length; i++) {
|
||||
const next = slots[i];
|
||||
|
||||
// 如果参与者相同且时间连续，则合并
|
||||
if (
|
||||
current.endTime.getTime() === next.startTime.getTime() &&
|
||||
this.arraysEqual(current.participants, next.participants)
|
||||
) {
|
||||
current.endTime = next.endTime;
|
||||
} else {
|
||||
merged.push(current);
|
||||
current = next;
|
||||
}
|
||||
}
|
||||
merged.push(current);
|
||||
|
||||
return merged;
|
||||
}
|
||||
|
||||
/**
|
||||
* 验证时间段
|
||||
*/
|
||||
private validateTimeSlots(slots: TimeSlot[]): void {
|
||||
if (slots.length === 0) {
|
||||
throw new BadRequestException({
|
||||
code: ErrorCode.PARAM_ERROR,
|
||||
message: '至少需要一个时间段',
|
||||
});
|
||||
}
|
||||
|
||||
slots.forEach((slot, index) => {
|
||||
const start = new Date(slot.startTime);
|
||||
const end = new Date(slot.endTime);
|
||||
|
||||
if (start >= end) {
|
||||
throw new BadRequestException({
|
||||
code: ErrorCode.PARAM_ERROR,
|
||||
message: `时间段${index + 1}的结束时间必须大于开始时间`,
|
||||
});
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 标准化时间段数据
|
||||
*/
|
||||
private normalizeAvailableSlots(slots: any): TimeSlot[] {
|
||||
if (Array.isArray(slots)) {
|
||||
return slots;
|
||||
}
|
||||
return [];
|
||||
}
|
||||
|
||||
/**
|
||||
* 比较两个数组是否相同
|
||||
*/
|
||||
private async checkGroupMembership(userId: string, groupId: string) {
|
||||
const membership = await this.groupMemberRepository.findOne({
|
||||
where: { groupId, userId, isActive: true },
|
||||
});
|
||||
|
||||
if (!membership) {
|
||||
throw new ForbiddenException({
|
||||
code: ErrorCode.NOT_IN_GROUP,
|
||||
message: ErrorMessage[ErrorCode.NOT_IN_GROUP],
|
||||
});
|
||||
}
|
||||
|
||||
return membership;
|
||||
}
|
||||
|
||||
/**
|
||||
* 比较两个数组是否相同
|
||||
*/
|
||||
private arraysEqual(arr1: string[], arr2: string[]): boolean {
|
||||
if (arr1.length !== arr2.length) return false;
|
||||
const sorted1 = [...arr1].sort();
|
||||
const sorted2 = [...arr2].sort();
|
||||
return sorted1.every((val, index) => val === sorted2[index]);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user