cloudbeaver/webapp/packages/plugin-sql-editor/src/SqlExecutionState.ts

/*
 * cloudbeaver - Cloud Database Manager
 * Copyright (C) 2020 DBeaver Corp and others
 *
 * Licensed under the Apache License, Version 2.0.
 * you may not use this file except in compliance with the License.
 */

import { computed, observable, makeObservable } from 'mobx';

import { EDeferredState } from '@cloudbeaver/core-utils';

import type { SQLQueryExecutionProcess } from './SqlResultTabs/SQLQueryExecutionProcess';

export class SqlExecutionState {
  constructor() {
    makeObservable<SqlExecutionState, 'currentlyExecutingQuery'>(this, {
      isSqlExecuting: computed,
      canCancel: computed,
      currentlyExecutingQuery: observable,
    });
  }

  get isSqlExecuting(): boolean {
    return this.currentlyExecutingQuery ? this.currentlyExecutingQuery.isInProgress : false;
  }

  get canCancel(): boolean {
    return this.currentlyExecutingQuery ? this.currentlyExecutingQuery.getState() === EDeferredState.PENDING : false;
  }

  cancelSQlExecuting = () => {
    if (this.currentlyExecutingQuery) {
      this.currentlyExecutingQuery.cancel();
    }
  };

  private currentlyExecutingQuery: SQLQueryExecutionProcess | null = null;

  async setCurrentlyExecutingQuery(queryExecutionProcess: SQLQueryExecutionProcess): Promise<void> {
    if (this.currentlyExecutingQuery) {
      throw new Error('Simultaneous execution of several queries is forbidden');
    }
    this.currentlyExecutingQuery = queryExecutionProcess;

    try {
      await queryExecutionProcess.promise;
    } finally {
      this.currentlyExecutingQuery = null;
    }
  }
}