webapp/packages/plugin-sql-generator/src/SqlGenerators/SqlGeneratorsResource.ts - devel - vshmidt/cloudbeaver

webapp/packages/plugin-sql-generator/src/SqlGenerators/SqlGeneratorsResource.ts

63 строки2 KB

sergeyteleshev

[CB] SQL Query generation: front-end part (#4132)

11 фев 2026, 20:16

11 фев 2026, 20:162719f53

/*
 * CloudBeaver - Cloud Database Manager
 * Copyright (C) 2020-2026 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 { injectable } from '@cloudbeaver/core-di';
import { NavNodeInfoResource } from '@cloudbeaver/core-navigation-tree';
import { CachedMapResource, isResourceAlias, type ResourceKey, resourceKeyList, ResourceKeyUtils } from '@cloudbeaver/core-resource';
import { GraphQLService, type SqlGenerateResultSetQueryQueryVariables, type SqlQueryGenerator } from '@cloudbeaver/core-sdk';

export const MAX_GENERATORS_LENGTH = 15;

@injectable(() => [GraphQLService, NavNodeInfoResource])
export class SqlGeneratorsResource extends CachedMapResource<string, SqlQueryGenerator[]> {
  constructor(
    private readonly graphQLService: GraphQLService,
    private readonly navNodeInfoResource: NavNodeInfoResource,
  ) {
    super();

    this.navNodeInfoResource.outdateResource(this);
    this.navNodeInfoResource.deleteInResource(this);
  }

  async generateEntityQuery(generatorId: string, nodePathList: string | string[]): Promise<string> {
    const result = await this.graphQLService.sdk.sqlGenerateEntityQuery({
      generatorId,
      nodePathList,
      options: {},
    });

    return result.sqlGenerateEntityQuery;
  }

  async generateResultSetSql(params: SqlGenerateResultSetQueryQueryVariables): Promise<string | null> {
    const response = await this.graphQLService.sdk.sqlGenerateResultSetQuery(params);

    return response.sqlGenerateResultSetQuery;
  }

  protected async loader(key: ResourceKey<string>): Promise<Map<string, SqlQueryGenerator[]>> {
    if (isResourceAlias(key)) {
      throw new Error('Aliases not supported by this resource.');
    }
    const values = new Map();

    await ResourceKeyUtils.forEachAsync(key, async key => {
      const { generators } = await this.graphQLService.sdk.sqlEntityQueryGenerators({
        nodePathList: key,
      });
      values.set(key, generators);
    });

    this.set(resourceKeyList(Array.from(values.keys())), Array.from(values.values()));
    return this.data;
  }

  protected validateKey(key: string): boolean {
    return typeof key === 'string';
  }
}