use swr for subscribe relaypool (only chats/channels page)

This commit is contained in:
Ren Amamiya
2023-04-22 08:09:17 +07:00
parent 2239606235
commit 340f6e4806
6 changed files with 78 additions and 26 deletions

View File

@@ -1,6 +1,6 @@
'use client';
import { ChannelMessages } from '@components/channels/messages/index';
import { ChannelMessages } from '@components/channels/messages';
import { FormChannel } from '@components/form/channel';
import { RelayContext } from '@components/relaysProvider';
@@ -14,6 +14,7 @@ import { useSetAtom } from 'jotai';
import { useResetAtom } from 'jotai/utils';
import { useSearchParams } from 'next/navigation';
import { useContext, useEffect, useRef } from 'react';
import useSWRSubscription from 'swr/subscription';
export default function Page() {
const searchParams = useSearchParams();
@@ -30,23 +31,18 @@ export default function Page() {
const muted = useRef(new Set());
const hided = useRef(new Set());
useEffect(() => {
// reset channel reply
resetChannelReply();
// reset channel messages
resetChannelMessages();
// subscribe event
useSWRSubscription(id, () => {
const unsubscribe = pool.subscribe(
[
{
authors: [activeAccount.pubkey],
kinds: [43, 44],
since: dateToUnix(hoursAgo(24, now.current)),
since: dateToUnix(hoursAgo(48, now.current)),
},
{
'#e': [id],
kinds: [42],
since: dateToUnix(hoursAgo(24, now.current)),
since: dateToUnix(hoursAgo(48, now.current)),
},
],
FULL_RELAYS,
@@ -61,7 +57,7 @@ export default function Page() {
} else if (hided.current.has(event.id)) {
console.log('hided');
} else {
setChannelMessages((messages) => [event, ...messages]);
setChannelMessages((prev) => [...prev, event]);
}
}
}
@@ -70,7 +66,22 @@ export default function Page() {
return () => {
unsubscribe();
};
}, [pool, id, activeAccount.pubkey, setChannelMessages, resetChannelReply, resetChannelMessages]);
});
useEffect(() => {
let ignore = false;
if (!ignore) {
// reset channel reply
resetChannelReply();
// reset channel messages
resetChannelMessages();
}
return () => {
ignore = true;
};
}, [resetChannelReply, resetChannelMessages]);
return (
<div className="flex h-full w-full flex-col justify-between">

View File

@@ -12,6 +12,7 @@ import { useSetAtom } from 'jotai';
import { useResetAtom } from 'jotai/utils';
import { useSearchParams } from 'next/navigation';
import { useContext, useEffect } from 'react';
import useSWRSubscription from 'swr/subscription';
export default function Page() {
const searchParams = useSearchParams();
@@ -23,10 +24,7 @@ export default function Page() {
const setChatMessages = useSetAtom(chatMessagesAtom);
const resetChatMessages = useResetAtom(chatMessagesAtom);
useEffect(() => {
// reset stored messages
resetChatMessages();
// fetch messages from relays
useSWRSubscription(pubkey, () => {
const unsubscribe = pool.subscribe(
[
{
@@ -42,14 +40,26 @@ export default function Page() {
],
FULL_RELAYS,
(event: any) => {
setChatMessages((data) => [...data, event]);
setChatMessages((prev) => [...prev, event]);
}
);
return () => {
unsubscribe();
};
}, [pubkey, activeAccount.pubkey, setChatMessages, pool]);
});
useEffect(() => {
let ignore = false;
if (!ignore) {
resetChatMessages();
}
return () => {
ignore = true;
};
}, [resetChatMessages]);
return (
<div className="flex h-full w-full flex-col justify-between">