@ibiz-template/runtime 0.0.1-alpha.9 → 0.0.1-beta.17

package/src/service/utils/entity-cache/entity-cache.ts

@@ -0,0 +1,369 @@
+import { where, equals, clone, isNil, isEmpty } from 'ramda';
+import { createUUID } from 'qx-util';
+import {
+  isExistSessionId,
+  isExistSrfKey,
+} from '../service-exist-util/service-exist-util';
+import { Entity } from '../../entity/entity';
+
+/**
+ * 实体缓存工具类
+ *
+ * @author chitanda
+ * @date 2022-08-17 23:08:56
+ * @export
+ * @class EntityCache
+ */
+export class EntityCache {
+  /**
+   * 数据缓存
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:08
+   * @type {Map<string, Map<string, Entity>>}
+   */
+  readonly cacheMap: Map<string, Map<string, Entity>> = new Map();
+
+  /**
+   * 强制设置数据，忽略其他逻辑
+   *
+   * @author chitanda
+   * @date 2022-05-10 17:05:45
+   * @param {IContext} context
+   * @param {Entity} entity
+   */
+  forceAdd(context: IContext, entity: Entity): void {
+    const map = this.getCacheByTag(context);
+    const data = map.get(entity.srfkey!);
+    if (data) {
+      data.assign!(entity);
+      ibiz.log.warn('forceAdd', entity.srfkey, entity);
+    }
+  }
+
+  /**
+   * 强制更新数据，非合并，忽略其他逻辑
+   *
+   * @author chitanda
+   * @date 2022-05-10 17:05:27
+   * @param {IContext} context
+   * @param {Entity} entity
+   */
+  forceUpdate(context: IContext, entity: Entity): void {
+    const map = this.getCacheByTag(context);
+    map.set(entity.srfkey!, clone(entity));
+    ibiz.log.warn('forceUpdate', entity.srfkey, entity);
+  }
+
+  /**
+   * 强制删除数据，忽略其他逻辑
+   *
+   * @author chitanda
+   * @date 2022-05-10 17:05:08
+   * @param {IContext} context
+   * @param {string} srfKey
+   */
+  forceDelete(context: IContext, srfKey: string): void {
+    const map = this.getCacheByTag(context);
+    map.delete(srfKey);
+    ibiz.log.warn('forceDelete', srfKey);
+  }
+
+  /**
+   * 新增数据
+   *
+   * @param {*} context
+   * @param {Entity} entity
+   * @return {*}  {boolean}
+   * @memberof EntityCache
+   */
+  add(context: IContext, entity: Entity): Entity | null {
+    try {
+      isExistSessionId('add', context);
+      if (isNil(entity.srfkey) || isEmpty(entity.srfkey)) {
+        entity.srfkey = createUUID();
+      }
+      entity.srftempdate = new Date().getTime();
+      const map = this.getCacheByTag(context);
+      map.set(entity.srfkey, clone(entity));
+      ibiz.log.warn('add', entity.srfkey, entity);
+      return entity;
+    } catch (err) {
+      ibiz.log.error(err);
+      return null;
+    }
+  }
+
+  /**
+   * 查找数据
+   *
+   * @param {*} context
+   * @param {string} srfKey
+   * @return {*}  {Entity}
+   * @memberof EntityCache
+   */
+  get(context: IContext, srfKey: string): Entity | null {
+    try {
+      isExistSessionId('get', context);
+      const map = this.getCacheByTag(context);
+      const data = map.get(srfKey);
+      ibiz.log.warn('get', srfKey, data);
+      return clone(data)!;
+    } catch (err) {
+      ibiz.log.error(err);
+      return null;
+    }
+  }
+
+  /**
+   * 更新数据
+   *
+   * @param {IContext} context
+   * @param {Entity} entity
+   * @return {*}  {Entity}
+   * @memberof EntityCache
+   */
+  update(context: IContext, entity: Entity): Entity | null {
+    try {
+      isExistSessionId('update', context);
+      isExistSrfKey('update', entity);
+      entity.srftempdate = new Date().getTime();
+      const map = this.getCacheByTag(context);
+      const data = map.get(entity.srfkey!);
+      if (data) {
+        data.assign!(entity);
+        map.set(entity.srfkey!, data);
+        ibiz.log.warn('update', entity.srfkey, entity);
+        return clone(data);
+      }
+      throw new Error('数据不存在，无法更新!');
+    } catch (err) {
+      ibiz.log.error(err);
+      return null;
+    }
+  }
+
+  /**
+   * 删除数据
+   *
+   * @param {IContext} context
+   * @param {string} srfKey
+   * @return {*}  {(Entity | null)}
+   * @memberof EntityCache
+   */
+  delete(context: IContext, srfKey: string): Entity | null {
+    try {
+      isExistSessionId('delete', context);
+      const map = this.getCacheByTag(context);
+      const key = srfKey;
+      if (map.has(key)) {
+        const data = map.get(key)!;
+        data.srftempdate = new Date().getTime();
+        map.delete(key);
+        ibiz.log.warn('delete', key);
+        return data;
+      }
+      return null;
+    } catch (err) {
+      ibiz.log.error(err);
+      return null;
+    }
+  }
+
+  /**
+   * 批量创建临时数据
+   *
+   * @author chitanda
+   * @date 2022-03-23 11:03:52
+   * @param {IContext} context
+   * @param {Entity[]} entities
+   * @return {*}  {Entity[]}
+   */
+  createBatch(context: IContext, entities: Entity[]): Entity[] {
+    try {
+      isExistSessionId('add', context);
+      const map = this.getCacheByTag(context);
+      for (let i = 0; i < entities.length; i++) {
+        const entity = entities[i];
+        if (isNil(entity.srfkey) || isEmpty(entity.srfkey)) {
+          entity.srfkey = createUUID();
+        }
+        entity.srftempdate = new Date().getTime();
+        const data = clone(entity);
+        map.set(entity.srfkey, data);
+        entities[i] = data;
+        ibiz.log.warn('add', entity.srfkey, entity);
+      }
+      return entities;
+    } catch (err) {
+      ibiz.log.error(err);
+    }
+    return [];
+  }
+
+  /**
+   * 批量更新数据
+   *
+   * @author chitanda
+   * @date 2022-03-23 10:03:17
+   * @param {IContext} context
+   * @param {Entity[]} entities
+   * @return {*}  {Entity[]}
+   */
+  updateBatch(context: IContext, entities: Entity[]): Entity[] {
+    try {
+      isExistSessionId('update', context);
+      const map = this.getCacheByTag(context);
+      for (let i = 0; i < entities.length; i++) {
+        const entity = entities[i];
+        isExistSrfKey('update', entity);
+        entity.srftempdate = new Date().getTime();
+        const data = map.get(entity.srfkey!);
+        if (data) {
+          data.assign!(entity);
+          map.set(entity.srfkey!, data);
+          ibiz.log.warn('update', entity.srfkey, entity);
+          entities[i] = clone(data);
+        } else {
+          throw new Error(
+            `数据[${entity.srfdename}-${entity.srfmajortext}(${entity.srfkey})]不存在，无法更新!`,
+          );
+        }
+      }
+      return entities;
+    } catch (err) {
+      ibiz.log.error(err);
+      return null!;
+    }
+  }
+
+  /**
+   * 批量删除数据
+   *
+   * @author chitanda
+   * @date 2022-03-23 10:03:40
+   * @param {IContext} context 上下文
+   * @param {string[]} srfKeys 需要删除的数据主键
+   * @return {*}  {string[]} 未能删除的数据主键
+   */
+  deleteBatch(context: IContext, srfKeys: string[]): string[] {
+    try {
+      isExistSessionId('delete', context);
+    } catch (err) {
+      ibiz.log.error(err);
+      return srfKeys;
+    }
+    const map = this.getCacheByTag(context);
+    const notRemoveKeys: string[] = [];
+    for (let i = 0; i < srfKeys.length; i++) {
+      const key = srfKeys[i];
+      if (!map.has(key)) {
+        notRemoveKeys.push(key);
+      }
+    }
+    if (notRemoveKeys.length > 0) {
+      throw new Error(
+        `未找到以下数据「${notRemoveKeys.join('、')}」，无法删除!`,
+      );
+    }
+    for (let i = 0; i < srfKeys.length; i++) {
+      const key = srfKeys[i];
+      map.delete(key);
+      ibiz.log.warn('delete', key);
+    }
+    return [];
+  }
+
+  /**
+   * 检查数据是否已经存在
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:06
+   * @param {IContext} context
+   * @param {string} srfkey
+   * @return {*}  {boolean}
+   */
+  checkData(context: IContext, srfkey: string): boolean {
+    const list = this.getList(context);
+    const i = list.findIndex(item => item.srfkey === srfkey);
+    return i !== -1;
+  }
+
+  /**
+   * 根据条件查找数据，未设置条件默认返回全部
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:56
+   * @param {IContext} context
+   * @return {*}  {Entity[]}
+   */
+  getList(context: IContext): Entity[] {
+    try {
+      isExistSessionId('getList', context);
+      const map = this.getCacheByTag(context);
+      if (!map) {
+        return [];
+      }
+      const values = map.values();
+      return Array.from(values);
+    } catch (err) {
+      ibiz.log.error(err);
+      return [];
+    }
+  }
+
+  /**
+   * 根据条件生成查询
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:33
+   * @param {IParams} [params={}]
+   * @return {*}  {<U>(testObj: U) => boolean}
+   */
+  generatePred(params: IParams = {}): <U>(testObj: U) => boolean {
+    // 查询数据条件集
+    const data: IData = {};
+    if (params.srfkey) {
+      data.srfkey = equals(params.srfkey);
+    }
+    delete params.srfkey;
+    for (const key in params) {
+      if (Object.prototype.hasOwnProperty.call(params, key)) {
+        const val = params[key];
+        data[key] = equals(val);
+      }
+    }
+    return where(data);
+  }
+
+  /**
+   * 清除缓存
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:54
+   * @param {IContext} context
+   */
+  clear(context: IContext): void {
+    const tag: string = context.srfsessionid;
+    if (this.cacheMap.size > 0) {
+      this.cacheMap.delete(tag);
+    }
+  }
+
+  /**
+   * 根据标识获取map组
+   *
+   * @author chitanda
+   * @date 2022-08-17 23:08:47
+   * @private
+   * @param {IContext} context
+   * @return {*}  {Map<string, Entity>}
+   */
+  private getCacheByTag(context: IContext): Map<string, Entity> {
+    const tag: string = context.srfsessionid;
+    if (!this.cacheMap.has(tag)) {
+      this.cacheMap.set(tag, new Map());
+    }
+    return this.cacheMap.get(tag)!;
+  }
+}