ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-17 02:51:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

Allow copying from multiple columns (#6509)

Browse Source 
* Allow copying from multiple columns

* Explicitly skip over undefined rows.
This commit is contained in:
Amir Omidi
2019-07-31 13:26:05 -07:00
committed by GitHub
parent bc4b527de0
commit 22bd1fdc55
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/sql/platform/query/common/gridDataProvider.ts
View File

@@ -43,7 +43,7 @@ export interface IGridDataProvider {
} }
export async function getResultsString(provider: IGridDataProvider, selection: Slick.Range[], includeHeaders?: boolean): Promise<string> { export async function getResultsString(provider: IGridDataProvider, selection: Slick.Range[], includeHeaders?: boolean): Promise<string> {
let copyString = ''; let copyTable: string[][] = [];
const eol = provider.getEolString(); const eol = provider.getEolString();
// create a mapping of the ranges to get promises // create a mapping of the ranges to get promises
@@ -52,13 +52,13 @@ export async function getResultsString(provider: IGridDataProvider, selection: S
const result = await provider.getRowData(range.fromRow, range.toRow - range.fromRow + 1); const result = await provider.getRowData(range.fromRow, range.toRow - range.fromRow + 1);
// If there was a previous selection separate it with a line break. Currently // If there was a previous selection separate it with a line break. Currently
// when there are multiple selections they are never on the same line // when there are multiple selections they are never on the same line
if (i > 0) {
copyString += eol;
}
if (provider.shouldIncludeHeaders(includeHeaders)) { if (provider.shouldIncludeHeaders(includeHeaders)) {
let columnHeaders = provider.getColumnHeaders(range); let columnHeaders = provider.getColumnHeaders(range);
if (columnHeaders !== undefined) { if (columnHeaders !== undefined) {
copyString += columnHeaders.join('\t') + eol; if (copyTable[0] === undefined) {
copyTable[0] = [];
}
copyTable[0].push(...columnHeaders);
} }
} }
// Iterate over the rows to paste into the copy string // Iterate over the rows to paste into the copy string
@@ -69,10 +69,12 @@ export async function getResultsString(provider: IGridDataProvider, selection: S
let cells = provider.shouldRemoveNewLines() let cells = provider.shouldRemoveNewLines()
? cellObjects.map(x => removeNewLines(x.displayValue)) ? cellObjects.map(x => removeNewLines(x.displayValue))
: cellObjects.map(x => x.displayValue); : cellObjects.map(x => x.displayValue);
copyString += cells.join('\t');
if (rowIndex < result.resultSubset.rows.length - 1) { let idx = rowIndex + 1;
copyString += eol; if (copyTable[idx] === undefined) {
copyTable[idx] = [];
} }
copyTable[idx].push(...cells);
} }
}; };
}); });
@@ -84,6 +86,15 @@ export async function getResultsString(provider: IGridDataProvider, selection: S
} }
await p; await p;
} }
let copyString = '';
copyTable.forEach((row) => {
if (row === undefined) {
return;
}
copyString = copyString.concat(row.join('\t').concat(eol));
});
return copyString; return copyString;
} }