codyml commented on code in PR #22043:
URL: https://github.com/apache/superset/pull/22043#discussion_r1095213114
##########
superset-frontend/src/views/CRUD/data/hooks.ts:
##########
@@ -81,35 +83,97 @@ export function useQueryPreviewState<D extends
BaseQueryObject = any>({
/**
* Retrieves all pages of dataset results
*/
-export const UseGetDatasetsList = async (filters: object[]) => {
- let results: DatasetObject[] = [];
- let page = 0;
- let count;
-
- // If count is undefined or less than results, we need to
- // asynchronously retrieve a page of dataset results
- while (count === undefined || results.length < count) {
- const queryParams = rison.encode_uri({ filters, page });
- try {
- // eslint-disable-next-line no-await-in-loop
- const response = await SupersetClient.get({
- endpoint: `/api/v1/dataset/?q=${queryParams}`,
- });
-
- // Reassign local count to response's count
- ({ count } = response.json);
-
- const {
- json: { result },
- } = response;
-
- results = [...results, ...result];
-
- page += 1;
- } catch (error) {
- addDangerToast(t('There was an error fetching dataset'));
- logging.error(t('There was an error fetching dataset'), error);
+export const useDatasetsList = (
+ db:
+ | (DatabaseObject & {
+ owners: [number];
+ })
+ | undefined,
+ schema: string | null | undefined,
+) => {
+ const [datasets, setDatasets] = useState<DatasetObject[]>([]);
+ const encodedSchema = schema ? encodeURIComponent(schema) : undefined;
+
+ const getDatasetsList = useCallback(async (filters: object[]) => {
+ let results: DatasetObject[] = [];
+ let page = 0;
+ let count;
+
+ // If count is undefined or less than results, we need to
+ // asynchronously retrieve a page of dataset results
+ while (count === undefined || results.length < count) {
+ const queryParams = rison.encode_uri({ filters, page });
+ try {
+ // eslint-disable-next-line no-await-in-loop
+ const response = await SupersetClient.get({
+ endpoint: `/api/v1/dataset/?q=${queryParams}`,
+ });
+
+ // Reassign local count to response's count
+ ({ count } = response.json);
+
+ const {
+ json: { result },
+ } = response;
+
+ results = [...results, ...result];
+
+ page += 1;
+ } catch (error) {
+ addDangerToast(t('There was an error fetching dataset'));
+ logging.error(t('There was an error fetching dataset'), error);
+ }
}
- }
- return results;
+
+ setDatasets(results);
+ return results;
Review Comment:
```suggestion
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
