@dazejs/framework/src/supports/orm/base-model.ts

Daze.js - A powerful web framework for Node.js

import { Builder } from '../database/builder';
import { Component } from '../../decorators';
import { ModelBuilder } from './builder';
import { Model } from './model';
import { Repository } from './repository';

@Component(undefined, 'entity')
@Reflect.metadata('connection', 'default')
export class BaseEntity {
  /**
     * 根据主键获取单挑记录
     * @param id
     */
  async get(id: number | string): Promise<Repository<this> & this> {
    return (new Model(this.constructor as any)).createRepository().get(id) as any;
  }

  /**
     * 创建模型的查询构建器
     */
  createQueryBuilder(): ModelBuilder<this> & Builder {
    return (new Model(this.constructor as any)).createRepository().createQueryBuilder() as any;
  }

  /**
     * 关联预加载
     * @param relations
     */
  with(relation: string, callback?: (query: ModelBuilder<this> & Builder) => void): Repository<this> & this {
    return (new Model(this.constructor as any)).createRepository().with(relation, callback) as any;
  }

  /**
     * 返回软删除数据
     * @returns 
     */
  withTrashed(): Repository<this> & this {
    return (new Model(this.constructor as {new(): any})).createRepository().withTrashed() as any;
  }

  /**
     * 返回隐私数据
     * @returns 
     */
  withSecret(): Repository<this> & this {
    return (new Model(this.constructor as {new(): any})).createRepository().withSecret() as any;
  }

  /**
     * 自动保存/新建记录
     */
  async save() {
    const repos = (new Model(this.constructor as any)).createRepository();
    repos.fill(this);
    await repos.save();
    Object.assign(this, repos.getAttributes());
    return repos;
  }

  /**
     * 创建记录
     * @param attributes
     */
  async create(attributes: Partial<Record<keyof this, any>>): Promise<Repository<this> & this> {
    const repos = (new Model(this.constructor as any)).createRepository();
    return repos.create(attributes) as any;
  }

  /**
     * 根据主键 id 删除记录
     * @param ids
     */
  async destroy(...ids: (number | string)[]) {
    const repos = (new Model(this.constructor as any)).createRepository();
    return repos.destroy(...ids);
  }
}