@@ -6,6 +6,8 @@ import { useEffect } from 'react';
66import { useNetworkType } from '@/context/network-type' ;
77import { logger } from '@/utils' ;
88
9+ import { useTransactionsCountQueryKey } from './use-transactions-count' ;
10+
911const TRANSFERS_COUNT_SUBSCRIPTION = `
1012 subscription TransfersCountSubscription {
1113 transferCount
@@ -16,17 +18,16 @@ function useTransactionsCountSubscription() {
1618 const alert = useAlert ( ) ;
1719 const queryClient = useQueryClient ( ) ;
1820 const { NETWORK_PRESET } = useNetworkType ( ) ;
21+ const queryKey = useTransactionsCountQueryKey ( ) ;
1922
2023 useEffect ( ( ) => {
21- const wsClient = createClient ( { url : NETWORK_PRESET . INDEXER_ADDRESS } ) ;
24+ const wsClient = createClient ( { url : NETWORK_PRESET . INDEXER_ADDRESS . replace ( 'http' , 'ws' ) } ) ;
2225
2326 const unsubscribe = wsClient . subscribe (
2427 { query : TRANSFERS_COUNT_SUBSCRIPTION } ,
2528 {
2629 next : ( result ) => {
27- queryClient . setQueryData ( [ 'transactionsCount' , undefined ] , {
28- allTransfers : { totalCount : result . data ?. transferCount } ,
29- } ) ;
30+ queryClient . setQueryData ( queryKey , { allTransfers : { totalCount : result . data ?. transferCount } } ) ;
3031 } ,
3132 error : ( error : Error ) => {
3233 logger . error ( 'Transaction count subscription' , error ) ;
@@ -41,7 +42,7 @@ function useTransactionsCountSubscription() {
4142 void wsClient . dispose ( ) ;
4243 } ;
4344 // eslint-disable-next-line react-hooks/exhaustive-deps
44- } , [ NETWORK_PRESET . INDEXER_ADDRESS ] ) ;
45+ } , [ NETWORK_PRESET . INDEXER_ADDRESS , queryKey ] ) ;
4546}
4647
4748export { useTransactionsCountSubscription } ;
0 commit comments