Files
FastGPT/packages/service/support/wallet/sub/schema.ts
Archer f2749cbb00 4.8.11 perf (#2832)
* save toast

* perf: surya ocr

* perf: remove same model name

* fix: indexes

* perf: ip check

* feat: Fixed the version number of the subapplication

* feat: simple app get latest child version

* perf: update child dispatch variables

* feat: variables update doc
2024-09-28 15:31:25 +08:00

97 lines
2.1 KiB
TypeScript

/*
user sub plan
1. type=standard: There will only be 1, and each team will have one
2. type=extraDatasetSize/extraPoints: Can buy multiple
*/
import { connectionMongo, getMongoModel } from '../../../common/mongo';
const { Schema } = connectionMongo;
import { TeamCollectionName } from '@fastgpt/global/support/user/team/constant';
import {
StandardSubLevelEnum,
SubModeEnum,
SubTypeEnum
} from '@fastgpt/global/support/wallet/sub/constants';
import type { TeamSubSchema } from '@fastgpt/global/support/wallet/sub/type';
export const subCollectionName = 'team_subscriptions';
const SubSchema = new Schema({
teamId: {
type: Schema.Types.ObjectId,
ref: TeamCollectionName,
required: true
},
type: {
type: String,
enum: Object.values(SubTypeEnum),
required: true
},
startTime: {
type: Date,
default: () => new Date()
},
expiredTime: {
type: Date,
required: true
},
// standard sub
currentMode: {
type: String,
enum: Object.values(SubModeEnum)
},
nextMode: {
type: String,
enum: Object.values(SubModeEnum)
},
currentSubLevel: {
type: String,
enum: Object.values(StandardSubLevelEnum)
},
nextSubLevel: {
type: String,
enum: Object.values(StandardSubLevelEnum)
},
// stand sub and extra points sub. Plan total points
totalPoints: {
type: Number
},
surplusPoints: {
// plan surplus points
type: Number
},
// extra dataset size
currentExtraDatasetSize: {
type: Number
}
});
try {
// Get plan by expiredTime
SubSchema.index({ expiredTime: -1, currentSubLevel: 1 });
// Get team plan
SubSchema.index({ teamId: 1, type: 1, expiredTime: -1 });
// timer task. Get standard plan;Get free plan;Clear expired extract plan
SubSchema.index({ type: 1, expiredTime: -1, currentSubLevel: 1 });
// 修改后的唯一索引
SubSchema.index(
{
teamId: 1,
type: 1,
currentSubLevel: 1
},
{
unique: true,
partialFilterExpression: { type: SubTypeEnum.standard }
}
);
} catch (error) {
console.log(error);
}
export const MongoTeamSub = getMongoModel<TeamSubSchema>(subCollectionName, SubSchema);