Auto-fill timeline gaps when getting re-connecting to Websocket/EventSource stream (#17987)
This commit is contained in:
		| @@ -7,6 +7,10 @@ import { | ||||
|   expandHomeTimeline, | ||||
|   connectTimeline, | ||||
|   disconnectTimeline, | ||||
|   fillHomeTimelineGaps, | ||||
|   fillPublicTimelineGaps, | ||||
|   fillCommunityTimelineGaps, | ||||
|   fillListTimelineGaps, | ||||
| } from './timelines'; | ||||
| import { updateNotifications, expandNotifications } from './notifications'; | ||||
| import { updateConversations } from './conversations'; | ||||
| @@ -35,6 +39,7 @@ const randomUpTo = max => | ||||
|  * @param {Object.<string, string>} params | ||||
|  * @param {Object} options | ||||
|  * @param {function(Function, Function): void} [options.fallback] | ||||
|  * @param {function(): void} [options.fillGaps] | ||||
|  * @param {function(object): boolean} [options.accept] | ||||
|  * @return {function(): void} | ||||
|  */ | ||||
| @@ -61,6 +66,10 @@ export const connectTimelineStream = (timelineId, channelName, params = {}, opti | ||||
|           clearTimeout(pollingId); | ||||
|           pollingId = null; | ||||
|         } | ||||
|  | ||||
|         if (options.fillGaps) { | ||||
|           dispatch(options.fillGaps()); | ||||
|         } | ||||
|       }, | ||||
|  | ||||
|       onDisconnect() { | ||||
| @@ -119,7 +128,7 @@ const refreshHomeTimelineAndNotification = (dispatch, done) => { | ||||
|  * @return {function(): void} | ||||
|  */ | ||||
| export const connectUserStream = () => | ||||
|   connectTimelineStream('home', 'user', {}, { fallback: refreshHomeTimelineAndNotification }); | ||||
|   connectTimelineStream('home', 'user', {}, { fallback: refreshHomeTimelineAndNotification, fillGaps: fillHomeTimelineGaps }); | ||||
|  | ||||
| /** | ||||
|  * @param {Object} options | ||||
| @@ -127,7 +136,7 @@ export const connectUserStream = () => | ||||
|  * @return {function(): void} | ||||
|  */ | ||||
| export const connectCommunityStream = ({ onlyMedia } = {}) => | ||||
|   connectTimelineStream(`community${onlyMedia ? ':media' : ''}`, `public:local${onlyMedia ? ':media' : ''}`); | ||||
|   connectTimelineStream(`community${onlyMedia ? ':media' : ''}`, `public:local${onlyMedia ? ':media' : ''}`, {}, { fillGaps: () => (fillCommunityTimelineGaps({ onlyMedia })) }); | ||||
|  | ||||
| /** | ||||
|  * @param {Object} options | ||||
| @@ -136,7 +145,7 @@ export const connectCommunityStream = ({ onlyMedia } = {}) => | ||||
|  * @return {function(): void} | ||||
|  */ | ||||
| export const connectPublicStream = ({ onlyMedia, onlyRemote } = {}) => | ||||
|   connectTimelineStream(`public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`, `public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`); | ||||
|   connectTimelineStream(`public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`, `public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`, {}, { fillGaps: () => fillPublicTimelineGaps({ onlyMedia, onlyRemote }) }); | ||||
|  | ||||
| /** | ||||
|  * @param {string} columnId | ||||
| @@ -159,4 +168,4 @@ export const connectDirectStream = () => | ||||
|  * @return {function(): void} | ||||
|  */ | ||||
| export const connectListStream = listId => | ||||
|   connectTimelineStream(`list:${listId}`, 'list', { list: listId }); | ||||
|   connectTimelineStream(`list:${listId}`, 'list', { list: listId }, { fillGaps: () => fillListTimelineGaps(listId) }); | ||||
|   | ||||
| @@ -124,6 +124,22 @@ export function expandTimeline(timelineId, path, params = {}, done = noOp) { | ||||
|   }; | ||||
| }; | ||||
|  | ||||
| export function fillTimelineGaps(timelineId, path, params = {}, done = noOp) { | ||||
|   return (dispatch, getState) => { | ||||
|     const timeline = getState().getIn(['timelines', timelineId], ImmutableMap()); | ||||
|     const items = timeline.get('items'); | ||||
|     const nullIndexes = items.map((statusId, index) => statusId === null ? index : null); | ||||
|     const gaps = nullIndexes.map(index => index > 0 ? items.get(index - 1) : null); | ||||
|  | ||||
|     // Only expand at most two gaps to avoid doing too many requests | ||||
|     done = gaps.take(2).reduce((done, maxId) => { | ||||
|       return (() => dispatch(expandTimeline(timelineId, path, { ...params, maxId }, done))); | ||||
|     }, done); | ||||
|  | ||||
|     done(); | ||||
|   }; | ||||
| } | ||||
|  | ||||
| export const expandHomeTimeline            = ({ maxId } = {}, done = noOp) => expandTimeline('home', '/api/v1/timelines/home', { max_id: maxId }, done); | ||||
| export const expandPublicTimeline          = ({ maxId, onlyMedia, onlyRemote } = {}, done = noOp) => expandTimeline(`public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`, '/api/v1/timelines/public', { remote: !!onlyRemote, max_id: maxId, only_media: !!onlyMedia }, done); | ||||
| export const expandCommunityTimeline       = ({ maxId, onlyMedia } = {}, done = noOp) => expandTimeline(`community${onlyMedia ? ':media' : ''}`, '/api/v1/timelines/public', { local: true, max_id: maxId, only_media: !!onlyMedia }, done); | ||||
| @@ -141,6 +157,11 @@ export const expandHashtagTimeline         = (hashtag, { maxId, tags, local } = | ||||
|   }, done); | ||||
| }; | ||||
|  | ||||
| export const fillHomeTimelineGaps      = (done = noOp) => fillTimelineGaps('home', '/api/v1/timelines/home', {}, done); | ||||
| export const fillPublicTimelineGaps    = ({ onlyMedia, onlyRemote } = {}, done = noOp) => fillTimelineGaps(`public${onlyRemote ? ':remote' : ''}${onlyMedia ? ':media' : ''}`, '/api/v1/timelines/public', { remote: !!onlyRemote, only_media: !!onlyMedia }, done); | ||||
| export const fillCommunityTimelineGaps = ({ onlyMedia } = {}, done = noOp) => fillTimelineGaps(`community${onlyMedia ? ':media' : ''}`, '/api/v1/timelines/public', { local: true, only_media: !!onlyMedia }, done); | ||||
| export const fillListTimelineGaps      = (id, done = noOp) => fillTimelineGaps(`list:${id}`, `/api/v1/timelines/list/${id}`, {}, done); | ||||
|  | ||||
| export function expandTimelineRequest(timeline, isLoadingMore) { | ||||
|   return { | ||||
|     type: TIMELINE_EXPAND_REQUEST, | ||||
|   | ||||
		Reference in New Issue
	
	Block a user