chore: lib-server

2026-04-24 04:17:25 +08:00 · 2024-10-03 22:03:49 +08:00
parent a13203fb3f
commit a4e2cc54e6
101 changed files with 936 additions and 223 deletions
@@ -0,0 +1,212 @@
+import { ValidateException } from './exception/index.js';
+import * as _ from 'lodash-es';
+import { PermissionException } from './exception/index.js';
+import { Repository } from 'typeorm';
+import { Inject } from '@midwayjs/core';
+import { TypeORMDataSourceManager } from '@midwayjs/typeorm';
+import { EntityManager } from 'typeorm/entity-manager/EntityManager.js';
+
+/**
+ * 服务基类
+ */
+export abstract class BaseService<T> {
+  @Inject()
+  dataSourceManager: TypeORMDataSourceManager;
+
+  abstract getRepository(): Repository<T>;
+
+  async transaction(callback: (entityManager: EntityManager) => Promise<any>) {
+    const dataSource = this.dataSourceManager.getDataSource('default');
+    await dataSource.transaction(callback as any);
+  }
+
+  /**
+   * 获得单个ID
+   * @param id ID
+   * @param infoIgnoreProperty 忽略返回属性
+   */
+  async info(id, infoIgnoreProperty?): Promise<T | null> {
+    if (!id) {
+      throw new ValidateException('id不能为空');
+    }
+    const info = await this.getRepository().findOneBy({ id } as any);
+    if (info && infoIgnoreProperty) {
+      for (const property of infoIgnoreProperty) {
+        delete info[property];
+      }
+    }
+    return info;
+  }
+
+  /**
+   * 非分页查询
+   * @param options
+   */
+  async find(options) {
+    return await this.getRepository().find(options);
+  }
+
+  /**
+   * 删除
+   * @param ids 删除的ID集合 如：[1,2,3] 或者 1,2,3
+   */
+  async delete(ids) {
+    if (ids instanceof Array) {
+      await this.getRepository().delete(ids);
+    } else if (typeof ids === 'string') {
+      await this.getRepository().delete(ids.split(','));
+    } else {
+      //ids是一个condition
+      await this.getRepository().delete(ids);
+    }
+    await this.modifyAfter(ids);
+  }
+
+  /**
+   * 新增|修改
+   * @param param 数据
+   */
+  async addOrUpdate(param) {
+    await this.getRepository().save(param);
+  }
+
+  /**
+   * 新增
+   * @param param 数据
+   */
+  async add(param) {
+    const now = new Date();
+    param.createTime = now;
+    param.updateTime = now;
+    await this.addOrUpdate(param);
+    await this.modifyAfter(param);
+    return {
+      id: param.id,
+    };
+  }
+
+  /**
+   * 修改
+   * @param param 数据
+   */
+  async update(param) {
+    if (!param.id) throw new ValidateException('no id');
+    param.updateTime = new Date();
+    await this.addOrUpdate(param);
+    await this.modifyAfter(param);
+  }
+
+  /**
+   * 新增|修改|删除 之后的操作
+   * @param data 对应数据
+   */
+  async modifyAfter(data) {}
+
+  /**
+   * 分页查询
+   * @param query 查询条件 bean
+   * @param page
+   * @param order
+   * @param buildQuery
+   */
+  async page(query, page = { offset: 0, limit: 20 }, order, buildQuery) {
+    if (page.offset == null) {
+      page.offset = 0;
+    }
+    if (page.limit == null) {
+      page.limit = 20;
+    }
+    const qb = this.getRepository().createQueryBuilder('main');
+    if (order && order.prop) {
+      qb.addOrderBy('main.' + order.prop, order.asc ? 'ASC' : 'DESC');
+    }
+    qb.addOrderBy('id', 'DESC');
+    qb.offset(page.offset).limit(page.limit);
+    //根据bean query
+    if (query) {
+      let whereSql = '';
+      let index = 0;
+      _.forEach(query, (value, key) => {
+        if (!value) {
+          return;
+        }
+        if (index !== 0) {
+          whereSql += ' and ';
+        }
+        whereSql += ` main.${key} = :${key} `;
+        index++;
+      });
+      if (index > 0) {
+        qb.where(whereSql, query);
+      }
+    }
+    //自定义query
+    if (buildQuery) {
+      buildQuery(qb);
+    }
+    const list = await qb.getMany();
+    const total = await qb.getCount();
+    return {
+      records: list,
+      total,
+      offset: page.offset,
+      limit: page.limit,
+    };
+  }
+
+  /**
+   * 分页查询
+   * @param query 查询条件 bean
+   * @param order
+   * @param buildQuery
+   */
+  async list(query, order, buildQuery) {
+    const qb = this.getRepository().createQueryBuilder('main');
+    if (order && order.prop) {
+      qb.orderBy('main.' + order.prop, order.asc ? 'ASC' : 'DESC');
+    } else {
+      qb.orderBy('id', 'DESC');
+    }
+    //根据bean query
+    if (query) {
+      let whereSql = '';
+      let index = 0;
+      _.forEach(query, (value, key) => {
+        if (!value) {
+          return;
+        }
+        if (index !== 0) {
+          whereSql += ' and ';
+        }
+        whereSql += ` main.${key} = :${key} `;
+        index++;
+      });
+      if (index > 0) {
+        qb.where(whereSql, query);
+      }
+    }
+    //自定义query
+    if (buildQuery) {
+      buildQuery(qb);
+    }
+    return await qb.getMany();
+  }
+
+  async checkUserId(id: any = 0, userId, userKey = 'userId') {
+    // @ts-ignore
+    const res = await this.getRepository().findOne({
+      // @ts-ignore
+      select: { [userKey]: true },
+      // @ts-ignore
+      where: {
+        // @ts-ignore
+        id,
+      },
+    });
+    // @ts-ignore
+    if (!res || res[userKey] === userId) {
+      return;
+    }
+    throw new PermissionException('权限不足');
+  }
+}