115 lines
3.8 KiB
TypeScript
115 lines
3.8 KiB
TypeScript
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop, DocumentType, modelOptions } from '@typegoose/typegoose';
|
|
import { MARQUEE_TYPE } from '../consts';
|
|
import { genCode } from '../pubUtils/util';
|
|
/**
|
|
* 跑马灯
|
|
**/
|
|
@modelOptions({ schemaOptions: { id: false } })
|
|
@index({ code: 1 })
|
|
export default class Marquee extends BaseModel {
|
|
@prop({ required: true })
|
|
code: string; // 跑马灯唯一标识
|
|
|
|
@prop({ required: true, type: Number })
|
|
serverIds: number[]; // 推送服务器
|
|
|
|
@prop({ required: true, enum: MARQUEE_TYPE })
|
|
type: MARQUEE_TYPE; // 推送类型
|
|
|
|
@prop({ required: true })
|
|
startTime: Date; // 活动开始时间
|
|
|
|
@prop({ required: true })
|
|
endTime: Date; // 活动结束时间
|
|
|
|
@prop({ required: true })
|
|
interval: number; // 活动结束时间
|
|
|
|
@prop({ required: true })
|
|
content: string; // 广播内容
|
|
|
|
@prop({ required: true })
|
|
isRunning: boolean; // 当前正在发布
|
|
|
|
/**
|
|
* 创建跑马灯
|
|
* @param params 参数
|
|
* @param uid 后台操作人
|
|
*/
|
|
public static async createData(params: MarqueeParam, uid = 1) {
|
|
const code = genCode(10);
|
|
const doc = new MarqueeModel();
|
|
const update = Object.assign(doc.toJSON(), params, { code, createdBy: uid, updatedBy: uid });
|
|
const result: MarqueeType = await MarqueeModel.findOneAndUpdate({ code }, { $setOnInsert: update }, { new: true, upsert: true }).lean();
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* 更新跑马灯
|
|
* @param code 唯一标识
|
|
* @param params 参数
|
|
* @param uid 后台操作人
|
|
*/
|
|
public static async updateData(code: string, params: MarqueeParam, uid = 1) {
|
|
const result: MarqueeType = await MarqueeModel.findOneAndUpdate({ code }, { $set: {...params, updatedBy: uid }}, { new: true }).lean();
|
|
return result;
|
|
}
|
|
|
|
|
|
/**
|
|
* 根据唯一code查询一条数据
|
|
* @param {String} code 唯一码
|
|
*/
|
|
public static async findByCode(code: string) {
|
|
const result: MarqueeType = await MarqueeModel.findOne({ code }).lean();
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* 查询所有生效中的跑马灯
|
|
*/
|
|
public static async findEffectiveMarque() {
|
|
const result: MarqueeType[] = await MarqueeModel.find({ type: MARQUEE_TYPE.SCHEDULE, startTime: { $lte: new Date() }, endTime: { $gte: new Date() } }).lean();
|
|
return result;
|
|
}
|
|
|
|
|
|
private static getSearchObj(form: { type?: number, current?: boolean, content?: string }) {
|
|
let searchObj = {};
|
|
if(form['type']) searchObj['type'] = form.type;
|
|
if(form['current']) searchObj['isRunning'] = form.current
|
|
if(form['content']) searchObj['content'] = { $regex: new RegExp(form.content.toString(), 'i') };
|
|
return searchObj
|
|
}
|
|
|
|
public static async findByCondition(page: number, pageSize: number, sortField: string = 'updatedAt', sortOrder: string = 'descend', form: { type?: number, current?: boolean, content?: string } = {}) {
|
|
|
|
let searchObj = this.getSearchObj(form);
|
|
let sort = {};
|
|
if(sortField && sortOrder) {
|
|
if(sortOrder == 'ascend') {
|
|
sort[sortField] = 1;
|
|
} else if (sortOrder == 'descend') {
|
|
sort[sortField] = -1;
|
|
}
|
|
}
|
|
const result: MarqueeType[] = await MarqueeModel.find(searchObj).limit(pageSize).skip((page - 1) * pageSize).sort(sort).select('+sort +showStartTime +showEndTime +serverType +isEnable').lean({ getters: true, virtuals: true });
|
|
return result;
|
|
|
|
}
|
|
|
|
public static async countByCondition(form: { type?: number, current?: boolean, content?: string } = {}) {
|
|
|
|
let searchObj = this.getSearchObj(form);
|
|
const result = await MarqueeModel.count(searchObj);
|
|
return result;
|
|
}
|
|
}
|
|
|
|
|
|
export const MarqueeModel = getModelForClass(Marquee);
|
|
|
|
export interface MarqueeType extends Pick<DocumentType<Marquee>, keyof Marquee> {};
|
|
export type MarqueeParam = Partial<MarqueeType>;
|