first init.

wangzhibo

2026-07-16 bac4362a0b7726d38a23d38f0d7913f2c2bab262

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
import { RecycleDataEntity } from './../entity/data';
import { Inject, Provide, Scope, ScopeEnum } from '@midwayjs/core';
import { BaseService } from '@cool-midway/core';
import { InjectEntityModel, TypeORMDataSourceManager } from '@midwayjs/typeorm';
import { LessThan, Repository } from 'typeorm';
import * as _ from 'lodash';
import * as moment from 'moment';
import { BaseSysConfService } from '../../base/service/sys/conf';
 
/**
 * 数据回收
 */
@Provide()
@Scope(ScopeEnum.Request, { allowDowngrade: true })
export class RecycleDataService extends BaseService {
  @InjectEntityModel(RecycleDataEntity)
  recycleDataEntity: Repository<RecycleDataEntity>;
 
  @Inject()
  typeORMDataSourceManager: TypeORMDataSourceManager;
 
  @Inject()
  baseSysConfService: BaseSysConfService;
 
  /**
   * 恢复数据
   * @param ids
   */
  async restore(ids: number[]) {
    for (const id of ids) {
      const info = await this.recycleDataEntity.findOneBy({ id });
      if (!info) {
        continue;
      }
      let entityModel = this.typeORMDataSourceManager
        .getDataSource(info.entityInfo.dataSourceName)
        .getRepository(info.entityInfo.entity);
      await entityModel.save(info.data);
      await this.recycleDataEntity.delete(id);
    }
  }
 
  /**
   * 记录数据
   * @param params
   */
  async record(params) {
    const { ctx, data, entity } = params;
    if (!ctx?.req) return;
    const dataSourceName =
      this.typeORMDataSourceManager.getDataSourceNameByModel(entity.target);
    const url = ctx?.url;
    await this.recycleDataEntity.save({
      entityInfo: {
        dataSourceName,
        entity: entity.target.name,
      },
      url,
      params:
        ctx?.req.method === 'GET' ? ctx?.request.query : ctx?.request.body,
      data,
      count: data.length,
      userId: _.startsWith(url, '/admin/') ? ctx?.admin.userId : ctx?.user?.id,
    });
  }
 
  /**
   * 日志
   * @param isAll 是否清除全部
   */
  async clear(isAll?) {
    if (isAll) {
      await this.recycleDataEntity.clear();
      return;
    }
    const keepDay = await this.baseSysConfService.getValue('recycleKeep');
    if (keepDay) {
      const beforeDate = moment().add(-keepDay, 'days').startOf('day').toDate();
      await this.recycleDataEntity.delete({ createTime: LessThan(beforeDate) });
    } else {
      await this.recycleDataEntity.clear();
    }
  }
}