68 lines
2.5 KiB
TypeScript
68 lines
2.5 KiB
TypeScript
import BaseModel from './BaseModel';
|
|
import { index, getModelForClass, prop } from '@typegoose/typegoose';
|
|
|
|
@index({ roleId: 1, id: 1 })
|
|
@index({ seqId: 1 })
|
|
export default class Item extends BaseModel {
|
|
@prop({ required: true, default: '' })
|
|
roleId: string; // 角色 id
|
|
@prop({ required: true, default: '' })
|
|
roleName: string; // 角色名称
|
|
|
|
@prop({ required: true, default: '' })
|
|
id: number; // 道具 id
|
|
@prop({ required: true, default: '' })
|
|
type: number; // 道具类型
|
|
@prop({ required: true, default: '' })
|
|
itemName: string; // 道具名称
|
|
@prop({ required: true, default: 0 })
|
|
hid: number; // 将魂：武将id，其他：0
|
|
@prop({ required: true })
|
|
count: number; // 道具数量
|
|
|
|
|
|
public static async findbyRole(roleId: string, lean = true) {
|
|
const items = await ItemModel.find({ roleId }).select('id count type').lean(lean);
|
|
return items;
|
|
}
|
|
|
|
public static async findbyRoleAndGidAndCount(roleId: string, id: number, count: number, lean = true) {
|
|
const items = await ItemModel.findOne({ roleId, id, count: {$gte: count} }).select('id count type').lean(lean);
|
|
return items;
|
|
}
|
|
|
|
public static async increaseItem(roleId: string, id: number, count: number, itemInfo: {roleId: string, roleName: string, id: number, itemName: string, type: number, hid?: number }, lean = true) {
|
|
const doc = new ItemModel();
|
|
const setOnInsert = Object.assign(doc.toJSON(), itemInfo);
|
|
const equips = await ItemModel.findOneAndUpdate({roleId, id },{$setOnInsert: setOnInsert, $inc: { count }}, {new: true, upsert: true}).lean(lean);
|
|
return equips;
|
|
}
|
|
|
|
public static async decreaseItems(roleId: string, items: Array<{id: number, count: number}>, lean = true) {
|
|
let updateItems = new Array<{id: number, count: number}>(), hasError: boolean = false;
|
|
for(let {id, count} of items) {
|
|
const rec = await ItemModel.findOneAndUpdate({roleId, id, count: {$gte: count} },{ $inc: { count: -1 * count }}, {new: true}).lean(lean);
|
|
if(rec) {
|
|
updateItems.push({id, count});
|
|
} else {
|
|
hasError = true; break;
|
|
}
|
|
}
|
|
if(hasError) { // 数量不足
|
|
for(let {id, count} of updateItems) {
|
|
await ItemModel.findOneAndUpdate({roleId, id },{ $inc: { count }}, {new: true}).lean(lean);
|
|
}
|
|
return false
|
|
} else {
|
|
return true
|
|
}
|
|
}
|
|
|
|
public static async deleteAccount(roleId: string, lean = true) {
|
|
let result = await ItemModel.deleteMany({roleId}).lean(lean);
|
|
return result||{};
|
|
}
|
|
}
|
|
|
|
export const ItemModel = getModelForClass(Item);
|