Weiko
0ef7e6db85
## Context The method was very simple and deterministic, I'm moving its logic to a util so we don't have to import a service to use it.
28 lines
1.1 KiB
TypeScript
28 lines
1.1 KiB
TypeScript
import { PostgresTableSchemaColumn } from 'src/engine/metadata-modules/remote-server/types/postgres-table-schema-column';
|
|
import { getWorkspaceSchemaName } from 'src/engine/workspace-datasource/utils/get-workspace-schema-name.util';
|
|
import { WorkspaceDataSourceService } from 'src/engine/workspace-datasource/workspace-datasource.service';
|
|
|
|
export const fetchTableColumns = async (
|
|
workspaceDataSourceService: WorkspaceDataSourceService,
|
|
workspaceId: string,
|
|
tableName: string,
|
|
): Promise<PostgresTableSchemaColumn[]> => {
|
|
const schemaName = getWorkspaceSchemaName(workspaceId);
|
|
|
|
// TODO: executeRawQuery is deprecated and will throw
|
|
const res = await workspaceDataSourceService.executeRawQuery(
|
|
`SELECT column_name, data_type, udt_name
|
|
FROM information_schema.columns
|
|
WHERE table_schema = $1 AND table_name = $2`,
|
|
[schemaName, tableName],
|
|
workspaceId,
|
|
);
|
|
|
|
// @ts-expect-error legacy noImplicitAny
|
|
return res.map((column) => ({
|
|
columnName: column.column_name,
|
|
dataType: column.data_type,
|
|
udtName: column.udt_name,
|
|
}));
|
|
};
|