Auto-fill timeline gaps when getting re-connecting to Websocket/EventSource stream (#17987)
This commit is contained in:
		
							parent
							
								
									3906dd67ed
								
							
						
					
					
						commit
						a39bf04fe6
					
				
					 2 changed files with 34 additions and 4 deletions
				
			
		|  | @ -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, | ||||
|  |  | |||
		Loading…
	
		Reference in a new issue