72 lines
1.7 KiB
TypeScript
72 lines
1.7 KiB
TypeScript
// Next.js API route support: https://nextjs.org/docs/api-routes/introduction
|
|
import type { NextApiRequest, NextApiResponse } from 'next';
|
|
import { jsonRes } from '@/service/response';
|
|
import { connectToDatabase, Data, DataItem } from '@/service/mongo';
|
|
import { authToken } from '@/service/utils/tools';
|
|
import type { DataListItem } from '@/types/data';
|
|
import mongoose from 'mongoose';
|
|
|
|
export default async function handler(req: NextApiRequest, res: NextApiResponse) {
|
|
try {
|
|
const { authorization } = req.headers;
|
|
|
|
if (!authorization) {
|
|
throw new Error('缺少登录凭证');
|
|
}
|
|
|
|
const userId = await authToken(authorization);
|
|
|
|
await connectToDatabase();
|
|
|
|
const datalist = await Data.aggregate<DataListItem>([
|
|
{
|
|
$match: {
|
|
userId: new mongoose.Types.ObjectId(userId),
|
|
isDeleted: false
|
|
}
|
|
},
|
|
{
|
|
$sort: { createTime: -1 } // 按照创建时间倒序排列
|
|
},
|
|
{
|
|
$lookup: {
|
|
from: 'dataitems',
|
|
localField: '_id',
|
|
foreignField: 'dataId',
|
|
as: 'items'
|
|
}
|
|
},
|
|
{
|
|
$addFields: {
|
|
totalData: {
|
|
$size: '$items' // 统计dataItem的总数
|
|
},
|
|
trainingData: {
|
|
$size: {
|
|
$filter: {
|
|
input: '$items',
|
|
as: 'item',
|
|
cond: { $ne: ['$$item.status', 0] } // 统计 status 不为0的数量
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
$project: {
|
|
items: 0 // 不返回 items 字段
|
|
}
|
|
}
|
|
]);
|
|
|
|
jsonRes(res, {
|
|
data: datalist
|
|
});
|
|
} catch (err) {
|
|
jsonRes(res, {
|
|
code: 500,
|
|
error: err
|
|
});
|
|
}
|
|
}
|