Refactoring streaming connections (#4645)
parent
10e9a9a3f9
commit
ea958cae7f
5 changed files with 116 additions and 181 deletions
@ -0,0 +1,94 @@ |
|||||||
|
import createStream from '../stream'; |
||||||
|
import { |
||||||
|
updateTimeline, |
||||||
|
deleteFromTimelines, |
||||||
|
refreshHomeTimeline, |
||||||
|
connectTimeline, |
||||||
|
disconnectTimeline, |
||||||
|
} from './timelines'; |
||||||
|
import { updateNotifications, refreshNotifications } from './notifications'; |
||||||
|
import { getLocale } from '../locales'; |
||||||
|
|
||||||
|
const { messages } = getLocale(); |
||||||
|
|
||||||
|
export function connectTimelineStream (timelineId, path, pollingRefresh = null) { |
||||||
|
return (dispatch, getState) => { |
||||||
|
const streamingAPIBaseURL = getState().getIn(['meta', 'streaming_api_base_url']); |
||||||
|
const accessToken = getState().getIn(['meta', 'access_token']); |
||||||
|
const locale = getState().getIn(['meta', 'locale']); |
||||||
|
let polling = null; |
||||||
|
|
||||||
|
const setupPolling = () => { |
||||||
|
polling = setInterval(() => { |
||||||
|
pollingRefresh(dispatch); |
||||||
|
}, 20000); |
||||||
|
}; |
||||||
|
|
||||||
|
const clearPolling = () => { |
||||||
|
if (polling) { |
||||||
|
clearInterval(polling); |
||||||
|
polling = null; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
const subscription = createStream(streamingAPIBaseURL, accessToken, path, { |
||||||
|
|
||||||
|
connected () { |
||||||
|
if (pollingRefresh) { |
||||||
|
clearPolling(); |
||||||
|
} |
||||||
|
dispatch(connectTimeline(timelineId)); |
||||||
|
}, |
||||||
|
|
||||||
|
disconnected () { |
||||||
|
if (pollingRefresh) { |
||||||
|
setupPolling(); |
||||||
|
} |
||||||
|
dispatch(disconnectTimeline(timelineId)); |
||||||
|
}, |
||||||
|
|
||||||
|
received (data) { |
||||||
|
switch(data.event) { |
||||||
|
case 'update': |
||||||
|
dispatch(updateTimeline(timelineId, JSON.parse(data.payload))); |
||||||
|
break; |
||||||
|
case 'delete': |
||||||
|
dispatch(deleteFromTimelines(data.payload)); |
||||||
|
break; |
||||||
|
case 'notification': |
||||||
|
dispatch(updateNotifications(JSON.parse(data.payload), messages, locale)); |
||||||
|
break; |
||||||
|
} |
||||||
|
}, |
||||||
|
|
||||||
|
reconnected () { |
||||||
|
if (pollingRefresh) { |
||||||
|
clearPolling(); |
||||||
|
pollingRefresh(dispatch); |
||||||
|
} |
||||||
|
dispatch(connectTimeline(timelineId)); |
||||||
|
}, |
||||||
|
|
||||||
|
}); |
||||||
|
|
||||||
|
const disconnect = () => { |
||||||
|
if (subscription) { |
||||||
|
subscription.close(); |
||||||
|
} |
||||||
|
clearPolling(); |
||||||
|
}; |
||||||
|
|
||||||
|
return disconnect; |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
function refreshHomeTimelineAndNotification (dispatch) { |
||||||
|
dispatch(refreshHomeTimeline()); |
||||||
|
dispatch(refreshNotifications()); |
||||||
|
} |
||||||
|
|
||||||
|
export const connectUserStream = () => connectTimelineStream('home', 'user', refreshHomeTimelineAndNotification); |
||||||
|
export const connectCommunityStream = () => connectTimelineStream('community', 'public:local'); |
||||||
|
export const connectMediaStream = () => connectTimelineStream('community', 'public:local'); |
||||||
|
export const connectPublicStream = () => connectTimelineStream('public', 'public'); |
||||||
|
export const connectHashtagStream = (tag) => connectTimelineStream(`hashtag:${tag}`, `hashtag&tag=${tag}`); |
Loading…
Reference in new issue