@ttkit/daily-hot-api 0.0.0-stage → 0.1.1

package/dist/database/repositories/daily-brief.repository.js

@@ -0,0 +1,457 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DailyBriefRepository = void 0;
+const common_1 = require("@nestjs/common");
+const mongoose_1 = require("@nestjs/mongoose");
+const mongoose_2 = require("mongoose");
+const daily_brief_schema_1 = require("../schemas/daily-brief.schema");
+const INVALID_VALUES = ['', '待验证', '未知', 'N/A', '-'];
+const MARKET_PREFIXES = ['SH', 'SZ', 'BJ'];
+const MARKET_PREFIX_PATTERN = new RegExp(`^(?:${MARKET_PREFIXES.join('|')})(.{6})$`);
+const lastValidValue = (field) => ({
+    $let: {
+        vars: {
+            valid: {
+                $filter: {
+                    input: field,
+                    cond: { $not: [{ $in: ['$$this', [null, ...INVALID_VALUES]] }] },
+                },
+            },
+        },
+        in: { $ifNull: [{ $arrayElemAt: ['$$valid', -1] }, ''] },
+    },
+});
+let DailyBriefRepository = class DailyBriefRepository {
+    constructor(dailyBriefModel) {
+        this.dailyBriefModel = dailyBriefModel;
+        this.publicProjection = {
+            rawInputItems: 0,
+            searchEvidence: 0,
+            generationId: 0,
+            generationHeartbeatAt: 0,
+        };
+    }
+    async findByDate(briefDate, period = 'daily', includeDebug = false) {
+        return this.dailyBriefModel
+            .findOne({ briefDate, period }, includeDebug ? undefined : this.publicProjection)
+            .lean();
+    }
+    async findLatest(period, includeDebug = false) {
+        const query = { status: 'success' };
+        if (period) {
+            query.period = period;
+        }
+        return this.dailyBriefModel
+            .findOne(query, includeDebug ? undefined : this.publicProjection)
+            .sort({ briefDate: -1, createdAt: -1 })
+            .lean();
+    }
+    async list(options = {}) {
+        const page = Math.max(1, options.page || 1);
+        const limit = Math.min(100, Math.max(1, options.limit || 20));
+        const { status, period } = options;
+        const query = {};
+        if (status) {
+            query.status = status;
+        }
+        if (period) {
+            query.period = period;
+        }
+        const skip = (page - 1) * limit;
+        const [data, total] = await Promise.all([
+            this.dailyBriefModel.aggregate([
+                { $match: query },
+                { $sort: { briefDate: -1, createdAt: -1 } },
+                { $skip: skip },
+                { $limit: limit },
+                {
+                    $project: {
+                        _id: 0,
+                        status: 1,
+                        briefDate: 1,
+                        period: 1,
+                        analysis: {
+                            summary: { $ifNull: ['$analysis.summary', '$$REMOVE'] },
+                        },
+                        topicCount: {
+                            $cond: [
+                                { $isArray: '$analysis.topics' },
+                                { $size: '$analysis.topics' },
+                                '$$REMOVE',
+                            ],
+                        },
+                        updatedAt: 1,
+                    },
+                },
+            ]),
+            this.dailyBriefModel.countDocuments(query),
+        ]);
+        return {
+            data,
+            total,
+            page,
+            limit,
+            totalPages: Math.ceil(total / limit),
+        };
+    }
+    async getStockRanking(options = {}) {
+        const match = { status: 'success' };
+        if (options.period) {
+            match.period = options.period;
+        }
+        if (options.startDate || options.endDate) {
+            const briefDate = {};
+            if (options.startDate) {
+                briefDate.$gte = options.startDate;
+            }
+            if (options.endDate) {
+                briefDate.$lte = options.endDate;
+            }
+            match.briefDate = briefDate;
+        }
+        const stockOccurrences = [
+            {
+                $project: {
+                    briefId: '$_id',
+                    briefDate: 1,
+                    topics: {
+                        $cond: [{ $isArray: '$analysis.topics' }, '$analysis.topics', []],
+                    },
+                },
+            },
+            { $unwind: '$topics' },
+            {
+                $project: {
+                    briefId: 1,
+                    briefDate: 1,
+                    mappings: {
+                        $cond: [
+                            { $isArray: '$topics.aShareMapping' },
+                            '$topics.aShareMapping',
+                            [],
+                        ],
+                    },
+                },
+            },
+            { $unwind: '$mappings' },
+            {
+                $addFields: {
+                    company: {
+                        $trim: {
+                            input: {
+                                $convert: {
+                                    input: '$mappings.company',
+                                    to: 'string',
+                                    onError: '',
+                                    onNull: '',
+                                },
+                            },
+                        },
+                    },
+                    code: {
+                        $toUpper: {
+                            $replaceAll: {
+                                input: {
+                                    $trim: {
+                                        input: {
+                                            $convert: {
+                                                input: '$mappings.code',
+                                                to: 'string',
+                                                onError: '',
+                                                onNull: '',
+                                            },
+                                        },
+                                    },
+                                },
+                                find: ' ',
+                                replacement: '',
+                            },
+                        },
+                    },
+                },
+            },
+            {
+                $addFields: {
+                    code: {
+                        $let: {
+                            vars: {
+                                base: { $arrayElemAt: [{ $split: ['$code', '.'] }, 0] },
+                            },
+                            in: {
+                                $let: {
+                                    vars: {
+                                        matched: {
+                                            $regexFind: {
+                                                input: '$$base',
+                                                regex: MARKET_PREFIX_PATTERN,
+                                            },
+                                        },
+                                    },
+                                    in: {
+                                        $cond: [
+                                            { $ne: ['$$matched', null] },
+                                            { $arrayElemAt: ['$$matched.captures', 0] },
+                                            '$$base',
+                                        ],
+                                    },
+                                },
+                            },
+                        },
+                    },
+                },
+            },
+            {
+                $addFields: {
+                    hasValidCompany: { $not: [{ $in: ['$company', INVALID_VALUES] }] },
+                    hasValidCode: { $not: [{ $in: ['$code', INVALID_VALUES] }] },
+                },
+            },
+            {
+                $addFields: {
+                    groupKey: {
+                        $cond: [
+                            '$hasValidCompany',
+                            { $concat: ['company:', '$company'] },
+                            {
+                                $cond: ['$hasValidCode', { $concat: ['code:', '$code'] }, ''],
+                            },
+                        ],
+                    },
+                },
+            },
+            { $match: { groupKey: { $ne: '' } } },
+        ];
+        const limit = Math.min(200, Math.max(1, options.limit || 50));
+        const [[result], totalBriefs] = await Promise.all([
+            this.dailyBriefModel.aggregate([
+                { $match: match },
+                { $sort: { briefDate: 1, createdAt: 1 } },
+                ...stockOccurrences,
+                {
+                    $group: {
+                        _id: '$groupKey',
+                        companies: {
+                            $push: { $cond: ['$hasValidCompany', '$company', null] },
+                        },
+                        codes: { $push: { $cond: ['$hasValidCode', '$code', null] } },
+                        appearanceCount: { $sum: 1 },
+                        briefIds: { $addToSet: '$briefId' },
+                        firstAppearedDate: { $min: '$briefDate' },
+                        lastAppearedDate: { $max: '$briefDate' },
+                    },
+                },
+                {
+                    $addFields: {
+                        company: lastValidValue('$companies'),
+                        resolvedCode: lastValidValue('$codes'),
+                    },
+                },
+                {
+                    $addFields: {
+                        identity: {
+                            $cond: [
+                                { $eq: ['$resolvedCode', ''] },
+                                '$_id',
+                                { $concat: ['code:', '$resolvedCode'] },
+                            ],
+                        },
+                    },
+                },
+                { $sort: { lastAppearedDate: 1 } },
+                {
+                    $group: {
+                        _id: '$identity',
+                        companies: { $push: '$company' },
+                        code: { $last: '$resolvedCode' },
+                        appearanceCount: { $sum: '$appearanceCount' },
+                        briefIdGroups: { $push: '$briefIds' },
+                        firstAppearedDate: { $min: '$firstAppearedDate' },
+                        lastAppearedDate: { $max: '$lastAppearedDate' },
+                    },
+                },
+                {
+                    $addFields: {
+                        company: lastValidValue('$companies'),
+                        briefCount: {
+                            $size: {
+                                $reduce: {
+                                    input: '$briefIdGroups',
+                                    initialValue: [],
+                                    in: { $setUnion: ['$$value', '$$this'] },
+                                },
+                            },
+                        },
+                    },
+                },
+                {
+                    $project: {
+                        briefIdGroups: 0,
+                        briefIds: 0,
+                        companies: 0,
+                        codes: 0,
+                        resolvedCode: 0,
+                    },
+                },
+                {
+                    $facet: {
+                        summary: [
+                            {
+                                $group: {
+                                    _id: null,
+                                    uniqueStocks: { $sum: 1 },
+                                    totalAppearances: { $sum: '$appearanceCount' },
+                                },
+                            },
+                            { $project: { _id: 0, uniqueStocks: 1, totalAppearances: 1 } },
+                        ],
+                        rankings: [
+                            {
+                                $sort: {
+                                    appearanceCount: -1,
+                                    briefCount: -1,
+                                    lastAppearedDate: -1,
+                                    code: 1,
+                                    company: 1,
+                                },
+                            },
+                            { $limit: limit },
+                            {
+                                $project: {
+                                    _id: 0,
+                                    company: 1,
+                                    code: 1,
+                                    appearanceCount: 1,
+                                    briefCount: 1,
+                                    firstAppearedDate: 1,
+                                    lastAppearedDate: 1,
+                                },
+                            },
+                        ],
+                    },
+                },
+            ]),
+            this.dailyBriefModel.countDocuments(match),
+        ]);
+        const summary = result?.summary[0];
+        const invalidNames = new Set(INVALID_VALUES);
+        return {
+            totalBriefs,
+            uniqueStocks: summary?.uniqueStocks || 0,
+            totalAppearances: summary?.totalAppearances || 0,
+            rankings: (result?.rankings || []).map((item) => ({
+                company: invalidNames.has(item.company)
+                    ? item.code || ''
+                    : item.company,
+                code: item.code || null,
+                appearanceCount: item.appearanceCount,
+                briefCount: item.briefCount,
+                firstAppearedDate: item.firstAppearedDate,
+                lastAppearedDate: item.lastAppearedDate,
+            })),
+        };
+    }
+    async claimGeneration(data) {
+        try {
+            const claimed = await this.dailyBriefModel
+                .findOneAndUpdate({
+                briefDate: data.briefDate,
+                period: data.period,
+                $or: [
+                    { generationId: { $exists: false } },
+                    { generationId: null },
+                    { generationHeartbeatAt: { $lt: data.staleBefore } },
+                ],
+            }, {
+                $set: {
+                    generationId: data.generationId,
+                    generationHeartbeatAt: new Date(),
+                },
+                $setOnInsert: {
+                    status: 'generating',
+                    sources: data.sources,
+                    inputWindow: data.inputWindow,
+                    model: data.model,
+                    createdAt: new Date(),
+                },
+            }, { upsert: true, new: true })
+                .lean();
+            return claimed?.generationId === data.generationId;
+        }
+        catch (error) {
+            if (error.code === 11000)
+                return false;
+            throw error;
+        }
+    }
+    async heartbeatGeneration(briefDate, period, generationId) {
+        await this.dailyBriefModel.updateOne({ briefDate, period, generationId }, { $set: { generationHeartbeatAt: new Date() } });
+    }
+    async markSuccess(briefDate, period, generationId, data) {
+        return this.dailyBriefModel
+            .findOneAndUpdate({ briefDate, period, generationId }, {
+            $set: {
+                ...data,
+                status: 'success',
+                updatedAt: new Date(),
+            },
+            $unset: {
+                error: 1,
+                generationId: 1,
+                generationHeartbeatAt: 1,
+            },
+        }, { new: true })
+            .lean();
+    }
+    async markFailed(briefDate, period, generationId, error) {
+        const previousSuccess = await this.dailyBriefModel
+            .findOneAndUpdate({ briefDate, period, generationId, status: 'success' }, { $unset: { generationId: 1, generationHeartbeatAt: 1 } }, { new: true })
+            .lean();
+        if (previousSuccess)
+            return previousSuccess;
+        return this.dailyBriefModel
+            .findOneAndUpdate({ briefDate, period, generationId }, {
+            $set: {
+                status: 'failed',
+                error,
+                updatedAt: new Date(),
+            },
+            $unset: { generationId: 1, generationHeartbeatAt: 1 },
+        }, { new: true })
+            .lean();
+    }
+    async deleteByDate(briefDate, period) {
+        const query = { briefDate };
+        if (period) {
+            query.period = period;
+        }
+        return this.dailyBriefModel.deleteMany(query);
+    }
+    async deleteBeforeDate(beforeDate, period) {
+        const query = {
+            briefDate: { $lt: beforeDate },
+        };
+        if (period) {
+            query.period = period;
+        }
+        return this.dailyBriefModel.deleteMany(query);
+    }
+};
+exports.DailyBriefRepository = DailyBriefRepository;
+exports.DailyBriefRepository = DailyBriefRepository = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, mongoose_1.InjectModel)(daily_brief_schema_1.DailyBrief.name)),
+    __metadata("design:paramtypes", [mongoose_2.Model])
+], DailyBriefRepository);
+//# sourceMappingURL=daily-brief.repository.js.map

package/dist/database/repositories/daily-brief.repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"daily-brief.repository.js","sourceRoot":"","sources":["../../../src/database/repositories/daily-brief.repository.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAA4C;AAC5C,+CAA+C;AAC/C,uCAAgD;AAChD,sEAA+E;AAG/E,MAAM,cAAc,GAAG,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;AAGrD,MAAM,eAAe,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAM3C,MAAM,qBAAqB,GAAG,IAAI,MAAM,CACtC,OAAO,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAC3C,CAAC;AAMF,MAAM,cAAc,GAAG,CAAC,KAAa,EAAE,EAAE,CAAC,CAAC;IACzC,IAAI,EAAE;QACJ,IAAI,EAAE;YACJ,KAAK,EAAE;gBACL,OAAO,EAAE;oBACP,KAAK,EAAE,KAAK;oBAEZ,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,GAAG,cAAc,CAAC,CAAC,EAAE,CAAC,EAAE;iBACjE;aACF;SACF;QACD,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE;KACzD;CACF,CAAC,CAAC;AAmDI,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAQ/B,YAEE,eAA2D;QAA1C,oBAAe,GAAf,eAAe,CAA2B;QAT5C,qBAAgB,GAAG;YAClC,aAAa,EAAE,CAAC;YAChB,cAAc,EAAE,CAAC;YACjB,YAAY,EAAE,CAAC;YACf,qBAAqB,EAAE,CAAC;SACzB,CAAC;IAKC,CAAC;IAEJ,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,MAAM,GAAG,OAAO,EAAE,YAAY,GAAG,KAAK;QACxE,OAAO,IAAI,CAAC,eAAe;aACxB,OAAO,CACN,EAAE,SAAS,EAAE,MAAM,EAAE,EACrB,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CACjD;aACA,IAAI,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAe,EAAE,YAAY,GAAG,KAAK;QACpD,MAAM,KAAK,GAA4B,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC7D,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC,eAAe;aACxB,OAAO,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC;aAChE,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC;aACtC,IAAI,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,UAAiC,EAAE;QAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;QAC9D,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QACnC,MAAM,KAAK,GAA4B,EAAE,CAAC;QAE1C,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACxB,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACxB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;QAGhC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACtC,IAAI,CAAC,eAAe,CAAC,SAAS,CAAqB;gBACjD,EAAE,MAAM,EAAE,KAAK,EAAE;gBACjB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE;gBAC3C,EAAE,KAAK,EAAE,IAAI,EAAE;gBACf,EAAE,MAAM,EAAE,KAAK,EAAE;gBACjB;oBACE,QAAQ,EAAE;wBACR,GAAG,EAAE,CAAC;wBACN,MAAM,EAAE,CAAC;wBACT,SAAS,EAAE,CAAC;wBACZ,MAAM,EAAE,CAAC;wBACT,QAAQ,EAAE;4BACR,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,mBAAmB,EAAE,UAAU,CAAC,EAAE;yBACxD;wBACD,UAAU,EAAE;4BACV,KAAK,EAAE;gCACL,EAAE,QAAQ,EAAE,kBAAkB,EAAE;gCAChC,EAAE,KAAK,EAAE,kBAAkB,EAAE;gCAC7B,UAAU;6BACX;yBACF;wBACD,SAAS,EAAE,CAAC;qBACb;iBACF;aACF,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,KAAK,CAAC;SAC3C,CAAC,CAAC;QAEH,OAAO;YACL,IAAI;YACJ,KAAK;YACL,IAAI;YACJ,KAAK;YACL,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;SACrC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,UAA+B,EAAE;QACrD,MAAM,KAAK,GAA4B,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC7D,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAChC,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,SAAS,GAA2B,EAAE,CAAC;YAC7C,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACtB,SAAS,CAAC,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC;YACrC,CAAC;YACD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,SAAS,CAAC,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;YACnC,CAAC;YACD,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;QAC9B,CAAC;QAED,MAAM,gBAAgB,GAAoB;YACxC;gBACE,QAAQ,EAAE;oBACR,OAAO,EAAE,MAAM;oBACf,SAAS,EAAE,CAAC;oBACZ,MAAM,EAAE;wBACN,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,kBAAkB,EAAE,EAAE,kBAAkB,EAAE,EAAE,CAAC;qBAClE;iBACF;aACF;YACD,EAAE,OAAO,EAAE,SAAS,EAAE;YACtB;gBACE,QAAQ,EAAE;oBACR,OAAO,EAAE,CAAC;oBACV,SAAS,EAAE,CAAC;oBACZ,QAAQ,EAAE;wBACR,KAAK,EAAE;4BACL,EAAE,QAAQ,EAAE,uBAAuB,EAAE;4BACrC,uBAAuB;4BACvB,EAAE;yBACH;qBACF;iBACF;aACF;YACD,EAAE,OAAO,EAAE,WAAW,EAAE;YACxB;gBACE,UAAU,EAAE;oBACV,OAAO,EAAE;wBACP,KAAK,EAAE;4BACL,KAAK,EAAE;gCACL,QAAQ,EAAE;oCACR,KAAK,EAAE,mBAAmB;oCAC1B,EAAE,EAAE,QAAQ;oCACZ,OAAO,EAAE,EAAE;oCACX,MAAM,EAAE,EAAE;iCACX;6BACF;yBACF;qBACF;oBACD,IAAI,EAAE;wBACJ,QAAQ,EAAE;4BACR,WAAW,EAAE;gCACX,KAAK,EAAE;oCACL,KAAK,EAAE;wCACL,KAAK,EAAE;4CACL,QAAQ,EAAE;gDACR,KAAK,EAAE,gBAAgB;gDACvB,EAAE,EAAE,QAAQ;gDACZ,OAAO,EAAE,EAAE;gDACX,MAAM,EAAE,EAAE;6CACX;yCACF;qCACF;iCACF;gCACD,IAAI,EAAE,GAAG;gCACT,WAAW,EAAE,EAAE;6BAChB;yBACF;qBACF;iBACF;aACF;YAGD;gBACE,UAAU,EAAE;oBACV,IAAI,EAAE;wBACJ,IAAI,EAAE;4BACJ,IAAI,EAAE;gCACJ,IAAI,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE;6BACxD;4BACD,EAAE,EAAE;gCACF,IAAI,EAAE;oCACJ,IAAI,EAAE;wCACJ,OAAO,EAAE;4CACP,UAAU,EAAE;gDACV,KAAK,EAAE,QAAQ;gDACf,KAAK,EAAE,qBAAqB;6CAC7B;yCACF;qCACF;oCACD,EAAE,EAAE;wCACF,KAAK,EAAE;4CACL,EAAE,GAAG,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC,EAAE;4CAC5B,EAAE,YAAY,EAAE,CAAC,oBAAoB,EAAE,CAAC,CAAC,EAAE;4CAC3C,QAAQ;yCACT;qCACF;iCACF;6BACF;yBACF;qBACF;iBACF;aACF;YACD;gBACE,UAAU,EAAE;oBACV,eAAe,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,CAAC,EAAE;oBAClE,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,CAAC,EAAE;iBAC7D;aACF;YACD;gBAGE,UAAU,EAAE;oBACV,QAAQ,EAAE;wBACR,KAAK,EAAE;4BACL,kBAAkB;4BAClB,EAAE,OAAO,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC,EAAE;4BACrC;gCACE,KAAK,EAAE,CAAC,eAAe,EAAE,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC;6BAC9D;yBACF;qBACF;iBACF;aACF;YACD,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE;SACtC,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;QAG9D,MAAM,CAAC,CAAC,MAAM,CAAC,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAChD,IAAI,CAAC,eAAe,CAAC,SAAS,CAAgC;gBAC5D,EAAE,MAAM,EAAE,KAAK,EAAE;gBAEjB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE;gBACzC,GAAG,gBAAgB;gBAEnB;oBACE,MAAM,EAAE;wBACN,GAAG,EAAE,WAAW;wBAChB,SAAS,EAAE;4BACT,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,kBAAkB,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE;yBACzD;wBACD,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,eAAe,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;wBAC7D,eAAe,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;wBAC5B,QAAQ,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE;wBACnC,iBAAiB,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;wBACzC,gBAAgB,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;qBACzC;iBACF;gBACD;oBAEE,UAAU,EAAE;wBACV,OAAO,EAAE,cAAc,CAAC,YAAY,CAAC;wBACrC,YAAY,EAAE,cAAc,CAAC,QAAQ,CAAC;qBACvC;iBACF;gBACD;oBAGE,UAAU,EAAE;wBACV,QAAQ,EAAE;4BACR,KAAK,EAAE;gCACL,EAAE,GAAG,EAAE,CAAC,eAAe,EAAE,EAAE,CAAC,EAAE;gCAC9B,MAAM;gCACN,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,eAAe,CAAC,EAAE;6BACxC;yBACF;qBACF;iBACF;gBACD,EAAE,KAAK,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,EAAE;gBAElC;oBACE,MAAM,EAAE;wBACN,GAAG,EAAE,WAAW;wBAChB,SAAS,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE;wBAChC,IAAI,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE;wBAChC,eAAe,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;wBAC7C,aAAa,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE;wBACrC,iBAAiB,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE;wBACjD,gBAAgB,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE;qBAChD;iBACF;gBACD;oBACE,UAAU,EAAE;wBACV,OAAO,EAAE,cAAc,CAAC,YAAY,CAAC;wBAErC,UAAU,EAAE;4BACV,KAAK,EAAE;gCACL,OAAO,EAAE;oCACP,KAAK,EAAE,gBAAgB;oCACvB,YAAY,EAAE,EAAE;oCAChB,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE;iCACzC;6BACF;yBACF;qBACF;iBACF;gBAED;oBACE,QAAQ,EAAE;wBACR,aAAa,EAAE,CAAC;wBAChB,QAAQ,EAAE,CAAC;wBACX,SAAS,EAAE,CAAC;wBACZ,KAAK,EAAE,CAAC;wBACR,YAAY,EAAE,CAAC;qBAChB;iBACF;gBACD;oBACE,MAAM,EAAE;wBACN,OAAO,EAAE;4BACP;gCACE,MAAM,EAAE;oCACN,GAAG,EAAE,IAAI;oCACT,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;oCACzB,gBAAgB,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;iCAC/C;6BACF;4BACD,EAAE,QAAQ,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,EAAE;yBAC/D;wBACD,QAAQ,EAAE;4BACR;gCACE,KAAK,EAAE;oCACL,eAAe,EAAE,CAAC,CAAC;oCACnB,UAAU,EAAE,CAAC,CAAC;oCACd,gBAAgB,EAAE,CAAC,CAAC;oCACpB,IAAI,EAAE,CAAC;oCACP,OAAO,EAAE,CAAC;iCACX;6BACF;4BACD,EAAE,MAAM,EAAE,KAAK,EAAE;4BACjB;gCACE,QAAQ,EAAE;oCACR,GAAG,EAAE,CAAC;oCACN,OAAO,EAAE,CAAC;oCACV,IAAI,EAAE,CAAC;oCACP,eAAe,EAAE,CAAC;oCAClB,UAAU,EAAE,CAAC;oCACb,iBAAiB,EAAE,CAAC;oCACpB,gBAAgB,EAAE,CAAC;iCACpB;6BACF;yBACF;qBACF;iBACF;aACF,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,cAAc,CAAC,KAAK,CAAC;SAC3C,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,CAAC;QAC7C,OAAO;YACL,WAAW;YACX,YAAY,EAAE,OAAO,EAAE,YAAY,IAAI,CAAC;YACxC,gBAAgB,EAAE,OAAO,EAAE,gBAAgB,IAAI,CAAC;YAChD,QAAQ,EAAE,CAAC,MAAM,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CACpC,CAAC,IAAI,EAAoB,EAAE,CAAC,CAAC;gBAC3B,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC;oBACrC,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE;oBACjB,CAAC,CAAC,IAAI,CAAC,OAAO;gBAChB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;gBACvB,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;gBACzC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;aACxC,CAAC,CACH;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAQrB;QACC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe;iBACvC,gBAAgB,CACf;gBACE,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,GAAG,EAAE;oBACH,EAAE,YAAY,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE;oBACpC,EAAE,YAAY,EAAE,IAAI,EAAE;oBACtB,EAAE,qBAAqB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE;iBACrD;aACF,EACD;gBACE,IAAI,EAAE;oBACJ,YAAY,EAAE,IAAI,CAAC,YAAY;oBAC/B,qBAAqB,EAAE,IAAI,IAAI,EAAE;iBAClC;gBACD,YAAY,EAAE;oBACZ,MAAM,EAAE,YAAY;oBACpB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB;aACF,EACD,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,CAC5B;iBACA,IAAI,EAAE,CAAC;YACV,OAAO,OAAO,EAAE,YAAY,KAAK,IAAI,CAAC,YAAY,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,IAAK,KAA2B,CAAC,IAAI,KAAK,KAAK;gBAAE,OAAO,KAAK,CAAC;YAC9D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,SAAiB,EACjB,MAAc,EACd,YAAoB;QAEpB,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAClC,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,EACnC,EAAE,IAAI,EAAE,EAAE,qBAAqB,EAAE,IAAI,IAAI,EAAE,EAAE,EAAE,CAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CACf,SAAiB,EACjB,MAAc,EACd,YAAoB,EACpB,IAUC;QAED,OAAO,IAAI,CAAC,eAAe;aACxB,gBAAgB,CACf,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,EACnC;YACE,IAAI,EAAE;gBACJ,GAAG,IAAI;gBACP,MAAM,EAAE,SAAS;gBACjB,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB;YACD,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC;gBACR,YAAY,EAAE,CAAC;gBACf,qBAAqB,EAAE,CAAC;aACzB;SACF,EACD,EAAE,GAAG,EAAE,IAAI,EAAE,CACd;aACA,IAAI,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,UAAU,CACd,SAAiB,EACjB,MAAc,EACd,YAAoB,EACpB,KAAa;QAEb,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,eAAe;aAC/C,gBAAgB,CACf,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,EACtD,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,qBAAqB,EAAE,CAAC,EAAE,EAAE,EACzD,EAAE,GAAG,EAAE,IAAI,EAAE,CACd;aACA,IAAI,EAAE,CAAC;QACV,IAAI,eAAe;YAAE,OAAO,eAAe,CAAC;QAE5C,OAAO,IAAI,CAAC,eAAe;aACxB,gBAAgB,CACf,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,EACnC;YACE,IAAI,EAAE;gBACJ,MAAM,EAAE,QAAQ;gBAChB,KAAK;gBACL,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB;YACD,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,qBAAqB,EAAE,CAAC,EAAE;SACtD,EACD,EAAE,GAAG,EAAE,IAAI,EAAE,CACd;aACA,IAAI,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,SAAiB,EACjB,MAAe;QAEf,MAAM,KAAK,GAA4B,EAAE,SAAS,EAAE,CAAC;QACrD,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,UAAkB,EAClB,MAAe;QAEf,MAAM,KAAK,GAA4B;YACrC,SAAS,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE;SAC/B,CAAC;QACF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACxB,CAAC;QAED,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;CACF,CAAA;AA/fY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;IAUR,WAAA,IAAA,sBAAW,EAAC,+BAAU,CAAC,IAAI,CAAC,CAAA;qCACK,gBAAK;GAV9B,oBAAoB,CA+fhC"}

package/dist/database/repositories/hot-item.repository.d.ts

@@ -0,0 +1,52 @@
+import { OnApplicationBootstrap } from '@nestjs/common';
+import { Model } from 'mongoose';
+import { HotItem, HotItemDocument } from '../schemas/hot-item.schema';
+import { HotListItem } from '../../host-lists/interfaces/hot-list.interface';
+export interface SearchHistoryOptions {
+    source?: string;
+    keyword?: string;
+    startDate?: Date;
+    endDate?: Date;
+    page?: number;
+    limit?: number;
+    sortBy?: string;
+    sortOrder?: 'asc' | 'desc';
+}
+export interface SearchHistoryResult {
+    data: HotItem[];
+    total: number;
+    page: number;
+    limit: number;
+    totalPages: number;
+}
+type HistoryStats = {
+    totalCount: number;
+    totalSources: number;
+    sources: {
+        _id: string;
+        count: number;
+        latestTimestamp: number;
+    }[];
+};
+export declare class HotItemRepository implements OnApplicationBootstrap {
+    private hotItemModel;
+    private readonly logger;
+    private statsCache?;
+    private statsInFlight?;
+    private statsVersion;
+    constructor(hotItemModel: Model<HotItemDocument>);
+    onApplicationBootstrap(): void;
+    private verifyUniqueIndex;
+    private isSourceUrlIndex;
+    isDuplicate(source: string, url: string): Promise<boolean>;
+    getDuplicateUrls(source: string, urls: string[]): Promise<Set<string>>;
+    private validateAndCleanItem;
+    saveHotItems(items: HotListItem[], source: string): Promise<number>;
+    searchHistory(options: SearchHistoryOptions): Promise<SearchHistoryResult>;
+    getSources(): Promise<string[]>;
+    getStats(): Promise<HistoryStats>;
+    private invalidateStats;
+    private loadStats;
+    getDataByTimeRange(source: string, startTime: number, endTime: number): Promise<HotItem[]>;
+}
+export {};