mirror of https://github.com/fantasticit/think.git
93 lines
2.3 KiB
TypeScript
93 lines
2.3 KiB
TypeScript
import { ViewEntity } from '@entities/view.entity';
|
|
import { parseUserAgent } from '@helpers/ua.helper';
|
|
import { Injectable } from '@nestjs/common';
|
|
import { InjectRepository } from '@nestjs/typeorm';
|
|
import { IDocument, IPagination, IUser } from '@think/domains';
|
|
import { Repository } from 'typeorm';
|
|
|
|
@Injectable()
|
|
export class ViewService {
|
|
constructor(
|
|
@InjectRepository(ViewEntity)
|
|
private readonly viewRepo: Repository<ViewEntity>
|
|
) {}
|
|
|
|
/**
|
|
* 创建访问记录（内部调用，无公开接口）
|
|
* @returns
|
|
*/
|
|
async create({ userId = 'public', documentId, userAgent }) {
|
|
const data = await this.viewRepo.create({
|
|
userId,
|
|
documentId,
|
|
originUserAgent: userAgent,
|
|
parsedUserAgent: parseUserAgent(userAgent).text,
|
|
});
|
|
const ret = await this.viewRepo.save(data);
|
|
return ret;
|
|
}
|
|
|
|
async deleteViews(documentId) {
|
|
const records = await this.viewRepo.find({ documentId });
|
|
await this.viewRepo.remove(records);
|
|
}
|
|
|
|
async getDocumentTotalViews(documentId) {
|
|
try {
|
|
const count = await this.viewRepo.query(
|
|
`SELECT COUNT(1)
|
|
FROM view
|
|
WHERE view.documentId = '${documentId}'
|
|
`
|
|
);
|
|
return count[0]['COUNT(1)'];
|
|
} catch (e) {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
async getDocumentViews(documentId, pagination: IPagination) {
|
|
let { page = 1, pageSize = 12 } = pagination;
|
|
if (page <= 0) {
|
|
page = 1;
|
|
}
|
|
if (pageSize <= 0) {
|
|
pageSize = 12;
|
|
}
|
|
const take = pageSize;
|
|
const skip = page === 1 ? 0 : (page - 1) * pageSize;
|
|
|
|
const [data, total] = await this.viewRepo.findAndCount({
|
|
where: { documentId },
|
|
take,
|
|
skip,
|
|
});
|
|
|
|
return { data, total };
|
|
}
|
|
|
|
async getUserRecentVisitedDocuments(userId: IUser['id']): Promise<
|
|
Array<{
|
|
documentId: IDocument['id'];
|
|
visitedAt: Date;
|
|
}>
|
|
> {
|
|
const count = 20;
|
|
const ret = await this.viewRepo.query(
|
|
`
|
|
SELECT v.documentId, v.visitedAt FROM (
|
|
SELECT ANY_VALUE(documentId) as documentId, ANY_VALUE(created_at) as visitedAt
|
|
FROM view
|
|
WHERE view.userId = '${userId}'
|
|
GROUP BY visitedAt
|
|
ORDER BY visitedAt DESC
|
|
) v
|
|
GROUP BY v.documentId
|
|
LIMIT ${count}
|
|
`
|
|
);
|
|
ret.sort((a, b) => -new Date(a.visitedAt).getTime() + new Date(b.visitedAt).getTime());
|
|
return ret;
|
|
}
|
|
}
|