init

Telegram/SourceFiles/data/data_pts_waiter.h

@@ -0,0 +1,104 @@
+/*
+This file is part of Telegram Desktop,
+the official desktop application for the Telegram messaging service.
+
+For license and copyright information please follow this link:
+https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
+*/
+#pragma once
+
+namespace Api {
+class Updates;
+} // namespace Api
+
+enum PtsSkippedQueue {
+	SkippedUpdate,
+	SkippedUpdates,
+};
+
+class PtsWaiter {
+public:
+	explicit PtsWaiter(not_null<Api::Updates*> owner);
+
+	// 1s wait for skipped seq or pts in updates.
+	static constexpr auto kWaitForSkippedTimeout = 1000;
+
+	void init(int32 pts) {
+		_good = _last = _count = pts;
+		clearSkippedUpdates();
+	}
+	bool inited() const {
+		return _good > 0;
+	}
+	void setRequesting(bool isRequesting) {
+		_requesting = isRequesting;
+		if (_requesting) {
+			clearSkippedUpdates();
+		}
+	}
+	bool requesting() const {
+		return _requesting;
+	}
+	bool waitingForSkipped() const {
+		return _waitingForSkipped;
+	}
+	bool waitingForShortPoll() const {
+		return _waitingForShortPoll;
+	}
+	void setWaitingForSkipped(ChannelData *channel, crl::time ms); // < 0 - not waiting
+	void setWaitingForShortPoll(ChannelData *channel, crl::time ms); // < 0 - not waiting
+	int32 current() const{
+		return _good;
+	}
+	bool updated(
+		ChannelData *channel,
+		int32 pts,
+		int32 count,
+		const MTPUpdates &updates);
+	bool updated(
+		ChannelData *channel,
+		int32 pts,
+		int32 count,
+		const MTPUpdate &update);
+	bool updated(
+		ChannelData *channel,
+		int32 pts,
+		int32 count);
+	bool updateAndApply(
+		ChannelData *channel,
+		int32 pts,
+		int32 count,
+		const MTPUpdates &updates);
+	bool updateAndApply(
+		ChannelData *channel,
+		int32 pts,
+		int32 count,
+		const MTPUpdate &update);
+	bool updateAndApply(
+		ChannelData *channel,
+		int32 pts,
+		int32 count);
+	void applySkippedUpdates(ChannelData *channel);
+	void clearSkippedUpdates();
+
+private:
+	// Return false if need to save that update and apply later.
+	bool check(ChannelData *channel, int32 pts, int32 count);
+
+	uint64 ptsKey(PtsSkippedQueue queue, int32 pts);
+	void checkForWaiting(ChannelData *channel);
+
+	const not_null<Api::Updates*> _owner;
+	base::flat_map<uint64, PtsSkippedQueue> _queue;
+	base::flat_map<uint64, MTPUpdate> _updateQueue;
+	base::flat_map<uint64, MTPUpdates> _updatesQueue;
+	int32 _good = 0;
+	int32 _last = 0;
+	int32 _count = 0;
+	int32 _applySkippedLevel = 0;
+	bool _requesting = false;
+	bool _waitingForSkipped = false;
+	bool _waitingForShortPoll = false;
+	uint32 _skippedKey = 0;
+
+};