ZYZ/shared/db/ActivityGroup.ts

import BaseModel from './BaseModel';
import { index, getModelForClass, prop, DocumentType } from '@typegoose/typegoose';
import { CounterModel } from './Counter';
import { COUNTER } from '../consts';
import { ActivityModel } from './Activity';

/**
  * 活动合集
*/
@index({ activityId: 1 })

export default class Activity_Group extends BaseModel {
    @prop({ required: true })
    groupId: number;				// 组Id
    @prop({ required: true })
    groupName: string;				// 组名
    @prop({ required: true, type: Number })
    activities: number[];		    // 活动Id
    @prop({ required: true })
    beginTime: Date;			    // 里面的activity的最早的时间
    @prop({ required: true })
    endTime: Date;					// 里面的activity的最晚的时间
    @prop({ required: true, type: Number })
    serverIds: number[];			// 关联服务器列表


    //查询组
    public static async createGroup() {
        let groupId = await CounterModel.getNewCounter(COUNTER.ACTIVITY_GROUP);

        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId }, {
            $setOnInsert: { groupName: `活动组${groupId}`, activities: [], serverIds: [] }
        }, { new: true, upsert: true }).lean();
        return result;
    }

    //查询组
    public static async findGroupData(groupId: number) {
        let result: ActivityGroupModelType = await ActivityGroupModel.findOne({ groupId }).lean();
        return result;
    }

    //更新名字
    public static async updateGroupName(groupId: number, groupName: string) {
        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId },
            { $set: { groupName } }, { new: true }).lean(true);
        return result;
    }

    //更新询组
    public static async setActivitiesToGroupData(groupId: number, activities: number[]) {
        await this.pullByActivityIds(activities); // 先将原先的删除
        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId },
            { $set: { activities } }, { new: true }).lean(true);
        await ActivityModel.updateGroupId(activities, groupId);
        result = await this.updateTime(result.groupId, result.activities);
        return result;
    }

    //更新询组
    public static async addActivitiesToGroupData(groupId: number, activities: number[]) {
        await this.pullByActivityIds(activities); // 先将原先的删除
        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId },
            { $addToSet: { activities: { $each: activities } } }, { new: true }).lean(true);
        await ActivityModel.updateGroupId(activities, groupId);
        result = await this.updateTime(result.groupId, result.activities);
        return result;
    }

    // 根据包含的活动更新开始和结束时间
    private static async updateTime(groupId: number, aids: number[]) {
        let activities = await ActivityModel.findActivityByIds(aids);
        let beginTime = undefined, endTime = undefined;
        for(let activity of activities) {
            if(!beginTime || activity.beginTime < beginTime) beginTime = activity.beginTime;
            if(!endTime || activity.endTime > endTime) endTime = activity.endTime;
        }
        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId },
            { $set: { beginTime, endTime } }, { new: true }).lean(true);
        return result;
    }


    // 删除活动id
    public static async pullByActivityIds(activities: number[]) {
        await ActivityModel.updateGroupId(activities, 0);
        let activityGroups = await ActivityGroupModel.find({ activities: { $in: activities } });
        for(let { groupId } of activityGroups) {
            let result = await ActivityGroupModel.findOneAndUpdate({ groupId }, { $pull: { activities: { $in: activities } } }, { new: true }).lean(true);
            await this.updateTime(result.groupId, result.activities);
        }
    }


    //更新服务器
    public static async updateServerData(groupId: number, serverIds: number[]) {
        let result: ActivityGroupModelType = await ActivityGroupModel.findOneAndUpdate({ groupId },
            { $set: { serverIds } }, { new: true }).lean(true);
        return result;
    }

    //删除组
    public static async deleteGroup(groupId: number) {
        let result = await ActivityGroupModel.deleteMany({ groupId });
        return result;
    }

    //查询
    public static async findByCondition(page: number, pageSize: number, serverId: number = 0, current: boolean = false, groupId: number = 0) {
        let searchObj = {};
        if (serverId != 0) searchObj['serverIds'] = { $elemMatch:{$eq:serverId} };
        if (groupId != 0) searchObj['groupId'] = groupId;
        if (current) {
            searchObj['beginTime'] = { $lte: new Date };
            searchObj['endTime'] = { $gte: new Date };
        }
        const result: ActivityGroupModelType[] = await ActivityGroupModel.find(searchObj, { _id: 0 }).limit(pageSize).skip((page - 1) * pageSize).sort({ updatedAt: -1 }).lean();
        return result;
    }

    // 获得活动数量
    public static async countByCondition(serverId: number = 0, current: boolean = false, groupId: number = 0) {
        let searchObj = {};
        if (serverId != 0) searchObj['serverId'] = { $in: [serverId] };
        if (groupId != 0) searchObj['groupId'] = groupId;
        if (current) {
            searchObj['beginTime'] = { $lte: new Date };
            searchObj['endTime'] = { $gte: new Date };
        }
        const result = await ActivityGroupModel.count(searchObj);
        return result;
    }
}

export const ActivityGroupModel = getModelForClass(Activity_Group);

export interface ActivityGroupModelType extends Pick<DocumentType<Activity_Group>, keyof Activity_Group> { }
export type ActivityGroupModelTypeParam = Partial<ActivityGroupModelType>; // 将所有字段变成可选项