objection/lib/queryBuilder/operations/UpsertGraphAndFetchOperation.js

An SQL-friendly ORM for Node.js

'use strict';

const { DelegateOperation } = require('./DelegateOperation');
const { UpsertGraphOperation } = require('./UpsertGraphOperation');
const { RelationExpression } = require('../RelationExpression');

class UpsertGraphAndFetchOperation extends DelegateOperation {
  constructor(name, opt) {
    super(name, opt);

    if (!this.delegate.is(UpsertGraphOperation)) {
      throw new Error('Invalid delegate');
    }
  }

  get models() {
    return this.delegate.models;
  }

  get isArray() {
    return this.delegate.isArray;
  }

  async onAfter3(builder) {
    if (this.models.length === 0) {
      return this.isArray ? [] : null;
    }

    const eager = RelationExpression.fromModelGraph(this.models);
    const modelClass = this.models[0].constructor;
    const ids = this.models.map((model) => model.$id());

    const models = await modelClass
      .query()
      .childQueryOf(builder)
      .findByIds(ids)
      .withGraphFetched(eager);

    return this.isArray ? models : models[0] || null;
  }
}

module.exports = {
  UpsertGraphAndFetchOperation,
};