Skip to content

Commit 0dde76b

Browse files
committed
fix: review fixes
1 parent a301a76 commit 0dde76b

File tree

3 files changed

+25
-61
lines changed

3 files changed

+25
-61
lines changed

src/containers/Tenant/Query/QuerySettingsDialog/QuerySettingsDialog.tsx

Lines changed: 24 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,22 @@ function QuerySettingsForm({initialValues, onSubmit, onClose}: QuerySettingsForm
101101
},
102102
);
103103

104+
const resourcePoolOptions = React.useMemo(
105+
() => [
106+
{
107+
value: RESOURCE_POOL_NO_OVERRIDE_VALUE,
108+
content: i18n('form.resource-pool.no-override'),
109+
text: i18n('form.resource-pool.no-override'),
110+
},
111+
...resourcePools.map((name) => ({
112+
value: name,
113+
content: name,
114+
text: name,
115+
})),
116+
],
117+
[resourcePools],
118+
);
119+
104120
const timeout = watch('timeout');
105121
const queryMode = watch('queryMode');
106122

@@ -127,6 +143,13 @@ function QuerySettingsForm({initialValues, onSubmit, onClose}: QuerySettingsForm
127143
} else if (mode === 'query') {
128144
setValue('timeout', null);
129145
}
146+
147+
if (mode === QUERY_MODES.pg) {
148+
setValue(
149+
'resourcePool',
150+
RESOURCE_POOL_NO_OVERRIDE_VALUE,
151+
);
152+
}
130153
}}
131154
settingOptions={QUERY_SETTINGS_FIELD_SETTINGS.queryMode.options}
132155
/>
@@ -152,18 +175,7 @@ function QuerySettingsForm({initialValues, onSubmit, onClose}: QuerySettingsForm
152175
queryMode === QUERY_MODES.pg
153176
}
154177
onUpdateSetting={(value) => field.onChange(value)}
155-
settingOptions={[
156-
{
157-
value: RESOURCE_POOL_NO_OVERRIDE_VALUE,
158-
content: i18n('form.resource-pool.no-override'),
159-
text: i18n('form.resource-pool.no-override'),
160-
},
161-
...resourcePools.map((name) => ({
162-
value: name,
163-
content: name,
164-
text: name,
165-
})),
166-
]}
178+
settingOptions={resourcePoolOptions}
167179
/>
168180
)}
169181
/>
Lines changed: 1 addition & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import {applyResourcePoolPragma, prepareQueryWithPragmas} from '../utils';
1+
import {prepareQueryWithPragmas} from '../utils';
22

33
describe('prepareQueryWithPragmas', () => {
44
test('Should prepend pragmas correctly', () => {
@@ -27,32 +27,4 @@ describe('prepareQueryWithPragmas', () => {
2727

2828
expect(expectedResult).toBe('PRAGMA OrderedColumns;\n\nSELECT * FROM table;');
2929
});
30-
31-
test('applyResourcePoolPragma should prepend pragma for yql syntax', () => {
32-
const query = 'SELECT * FROM table;';
33-
const result = applyResourcePoolPragma(query, 'olap', 'yql_v1');
34-
35-
expect(result).toBe('PRAGMA ResourcePool = "olap";\n\nSELECT * FROM table;');
36-
});
37-
38-
test('applyResourcePoolPragma should not change query when no resource pool is set', () => {
39-
const query = 'SELECT * FROM table;';
40-
const result = applyResourcePoolPragma(query, undefined, 'yql_v1');
41-
42-
expect(result).toBe(query);
43-
});
44-
45-
test('applyResourcePoolPragma should not change query for pg syntax', () => {
46-
const query = 'SELECT * FROM table;';
47-
const result = applyResourcePoolPragma(query, 'olap', 'pg');
48-
49-
expect(result).toBe(query);
50-
});
51-
52-
test('applyResourcePoolPragma should not duplicate existing ResourcePool pragma', () => {
53-
const query = 'PRAGMA ResourcePool = "analytics";\n\nSELECT * FROM table;';
54-
const result = applyResourcePoolPragma(query, 'olap', 'yql_v1');
55-
56-
expect(result).toBe(query);
57-
});
5830
});

src/store/reducers/query/utils.ts

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -78,23 +78,3 @@ export const prepareQueryWithPragmas = (query: string, pragmas?: string): string
7878

7979
return `${trimmedPragmas}${separator}${query}`;
8080
};
81-
82-
const RESOURCE_POOL_PRAGMA_REGEXP = /PRAGMA\s+ResourcePool\b/i;
83-
84-
export const applyResourcePoolPragma = (
85-
query: string,
86-
resourcePool: string | undefined,
87-
syntax: QuerySyntax,
88-
): string => {
89-
if (!resourcePool || syntax !== 'yql_v1') {
90-
return query;
91-
}
92-
93-
if (RESOURCE_POOL_PRAGMA_REGEXP.test(query)) {
94-
return query;
95-
}
96-
97-
const pragmaLine = `PRAGMA ResourcePool = "${resourcePool}";`;
98-
99-
return `${pragmaLine}\n\n${query}`;
100-
};

0 commit comments

Comments
 (0)