Skip to content

Commit

Permalink
feat: performance improvement
Browse files Browse the repository at this point in the history
  • Loading branch information
kawamataryo committed Feb 15, 2025
1 parent 23e6d6c commit dc417da
Showing 1 changed file with 20 additions and 4 deletions.
24 changes: 20 additions & 4 deletions src/hooks/useRetrieveBskyUsers.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import type { AtpSessionData } from "@atproto/api";
import type { ProfileView } from "@atproto/api/dist/client/types/app/bsky/actor/defs";
import { sendToBackground } from "@plasmohq/messaging";
import { Storage } from "@plasmohq/storage";
import { useStorage } from "@plasmohq/storage/hook";
Expand Down Expand Up @@ -92,13 +93,12 @@ export const useRetrieveBskyUsers = () => {
messageName: (typeof MESSAGE_NAMES)[keyof typeof MESSAGE_NAMES];
}>(null);

const retrieveBskyUsers = React.useCallback(
const searchAndStoreBskyUser = React.useCallback(
async (
usersData: CrawledUserInfo[],
userData: CrawledUserInfo,
processExtractedData: (user: CrawledUserInfo) => Promise<CrawledUserInfo>,
) => {
for (const userData of usersData) {
setScannedUserCount((prev) => prev + 1);
try {
const searchResult = await searchBskyUser({
client: bskyClient.current,
userData,
Expand Down Expand Up @@ -146,11 +146,27 @@ export const useRetrieveBskyUsers = () => {
];
});
}
} catch (error) {
consola.error(error);
}
},
[setUsers],
);

const retrieveBskyUsers = React.useCallback(
async (
usersData: CrawledUserInfo[],
processExtractedData: (user: CrawledUserInfo) => Promise<CrawledUserInfo>,
) => {
for (const userData of usersData) {
setScannedUserCount((prev) => prev + 1);
await wait(100);
searchAndStoreBskyUser(userData, processExtractedData);
}
},
[searchAndStoreBskyUser],
);

const abortControllerRef = React.useRef<AbortController | null>(null);
const startRetrieveLoop = React.useCallback(
async (service: IService) => {
Expand Down

0 comments on commit dc417da

Please sign in to comment.