139 lines
2.6 KiB
TypeScript
139 lines
2.6 KiB
TypeScript
import { connectionMongo, type Model } from '../../../common/mongo';
|
|
const { Schema, model, models } = connectionMongo;
|
|
import { TeamCollectionName } from '@fastgpt/global/support/user/team/constant';
|
|
import {
|
|
standardSubLevelMap,
|
|
subModeMap,
|
|
subStatusMap,
|
|
subTypeMap
|
|
} 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.keys(subTypeMap),
|
|
required: true
|
|
},
|
|
status: {
|
|
// active: continue sub; canceled: canceled sub;
|
|
type: String,
|
|
enum: Object.keys(subStatusMap),
|
|
required: true
|
|
},
|
|
mode: {
|
|
type: String,
|
|
enum: Object.keys(subModeMap),
|
|
required: true
|
|
},
|
|
startTime: {
|
|
type: Date,
|
|
default: () => new Date()
|
|
},
|
|
expiredTime: {
|
|
type: Date,
|
|
required: true
|
|
},
|
|
price: {
|
|
// last sub pay price(total price)
|
|
type: Number,
|
|
required: true
|
|
},
|
|
|
|
// sub content
|
|
currentSubLevel: {
|
|
type: String,
|
|
enum: Object.keys(standardSubLevelMap)
|
|
},
|
|
nextSubLevel: {
|
|
type: String,
|
|
enum: Object.keys(standardSubLevelMap)
|
|
},
|
|
|
|
currentExtraDatasetSize: {
|
|
type: Number
|
|
},
|
|
nextExtraDatasetSize: {
|
|
type: Number
|
|
},
|
|
|
|
currentExtraPoints: {
|
|
type: Number
|
|
},
|
|
nextExtraPoints: {
|
|
type: Number
|
|
},
|
|
|
|
// standard sub limit
|
|
maxTeamMember: {
|
|
type: Number
|
|
},
|
|
maxAppAmount: {
|
|
type: Number
|
|
},
|
|
maxDatasetAmount: {
|
|
type: Number
|
|
},
|
|
chatHistoryStoreDuration: {
|
|
// n day
|
|
type: Number
|
|
},
|
|
maxDatasetSize: {
|
|
type: Number
|
|
},
|
|
trainingWeight: {
|
|
// 0 1 2 3
|
|
type: Number
|
|
},
|
|
customApiKey: {
|
|
type: Boolean
|
|
},
|
|
customCopyright: {
|
|
type: Boolean
|
|
},
|
|
exportDatasetInterval: {
|
|
// hours
|
|
type: Number
|
|
},
|
|
websiteSyncInterval: {
|
|
// hours
|
|
type: Number
|
|
},
|
|
reRankWeight: {
|
|
// 0 1 2 3
|
|
type: Number
|
|
},
|
|
totalPoints: {
|
|
// record standard sub points
|
|
type: Number
|
|
},
|
|
|
|
surplusPoints: {
|
|
// standard sub / extra points sub
|
|
type: Number
|
|
},
|
|
|
|
// abandon
|
|
renew: Boolean, //决定是否续费
|
|
datasetStoreAmount: Number
|
|
});
|
|
|
|
try {
|
|
SubSchema.index({ teamId: 1 });
|
|
SubSchema.index({ status: 1 });
|
|
SubSchema.index({ type: 1 });
|
|
SubSchema.index({ expiredTime: -1 });
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
|
|
export const MongoTeamSub: Model<TeamSubSchema> =
|
|
models[subCollectionName] || model(subCollectionName, SubSchema);
|