Method setClass

packages/core/src/metadata/EntitySchema.ts:349–378

Sets or replaces the entity class associated with this schema.

(cls: Class)

  /** Sets or replaces the entity class associated with this schema. */
  setClass(cls: Class) {
    const oldClass = this._meta.class;
    const sameClass = this._meta.class === cls;
    this._meta.class = cls;
    this._meta.prototype = cls.prototype;
    this._meta.className = this._meta.name ?? cls.name;

    if (!sameClass || !this._meta.constructorParams) {
      this._meta.constructorParams = Utils.getConstructorParams(cls) as EntityKey<Entity>[];
    }

    if (!this.internal) {
      // Remove old class from registry if it's being replaced with a different class
      if (oldClass && oldClass !== cls && EntitySchema.REGISTRY.get(oldClass) === this) {
        EntitySchema.REGISTRY.delete(oldClass);
      }

      EntitySchema.REGISTRY.set(cls, this);
    }

    const base = Object.getPrototypeOf(cls);

    // Only set extends if the parent is NOT the auto-generated class for this same entity.
    // When the user extends the auto-generated class (from defineEntity without a class option)
    // and registers their custom class via setClass, we don't want to discover the
    // auto-generated class as a separate parent entity.
    if (base !== BaseEntity && base.name !== this._meta.className) {
      this._meta.extends ??= base.name ? base : undefined;
    }
  }

  /** Returns the underlying EntityMetadata. */
  get meta(): EntityMetadata<Entity, Class> {