128 lines
		
	
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			128 lines
		
	
	
	
		
			3.1 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
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
 | 
						|
*/
 | 
						|
#include "observer_peer.h"
 | 
						|
 | 
						|
#include "base/observer.h"
 | 
						|
 | 
						|
namespace Notify {
 | 
						|
namespace {
 | 
						|
 | 
						|
using SmallUpdatesList = QVector<PeerUpdate>;
 | 
						|
NeverFreedPointer<SmallUpdatesList> SmallUpdates;
 | 
						|
using AllUpdatesList = QMap<PeerData*, PeerUpdate>;
 | 
						|
NeverFreedPointer<AllUpdatesList> AllUpdates;
 | 
						|
 | 
						|
void StartCallback() {
 | 
						|
	SmallUpdates.createIfNull();
 | 
						|
	AllUpdates.createIfNull();
 | 
						|
}
 | 
						|
void FinishCallback() {
 | 
						|
	SmallUpdates.clear();
 | 
						|
	AllUpdates.clear();
 | 
						|
}
 | 
						|
 | 
						|
base::Observable<PeerUpdate, PeerUpdatedHandler> PeerUpdatedObservable;
 | 
						|
 | 
						|
} // namespace
 | 
						|
 | 
						|
void mergePeerUpdate(PeerUpdate &mergeTo, const PeerUpdate &mergeFrom) {
 | 
						|
	if (!(mergeTo.flags & PeerUpdate::Flag::NameChanged)) {
 | 
						|
		if (mergeFrom.flags & PeerUpdate::Flag::NameChanged) {
 | 
						|
			mergeTo.oldNameFirstLetters = mergeFrom.oldNameFirstLetters;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	mergeTo.flags |= mergeFrom.flags;
 | 
						|
}
 | 
						|
 | 
						|
void peerUpdatedDelayed(const PeerUpdate &update) {
 | 
						|
	SmallUpdates.createIfNull();
 | 
						|
	AllUpdates.createIfNull();
 | 
						|
 | 
						|
	Global::RefHandleDelayedPeerUpdates().call();
 | 
						|
 | 
						|
	int existingUpdatesCount = SmallUpdates->size();
 | 
						|
	for (int i = 0; i < existingUpdatesCount; ++i) {
 | 
						|
		auto &existingUpdate = (*SmallUpdates)[i];
 | 
						|
		if (existingUpdate.peer == update.peer) {
 | 
						|
			mergePeerUpdate(existingUpdate, update);
 | 
						|
			return;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (AllUpdates->isEmpty()) {
 | 
						|
		if (existingUpdatesCount < 5) {
 | 
						|
			SmallUpdates->push_back(update);
 | 
						|
		} else {
 | 
						|
			AllUpdates->insert(update.peer, update);
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		auto it = AllUpdates->find(update.peer);
 | 
						|
		if (it != AllUpdates->cend()) {
 | 
						|
			mergePeerUpdate(it.value(), update);
 | 
						|
			return;
 | 
						|
		}
 | 
						|
		AllUpdates->insert(update.peer, update);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void peerUpdatedSendDelayed() {
 | 
						|
	if (!SmallUpdates || !AllUpdates || SmallUpdates->empty()) return;
 | 
						|
 | 
						|
	auto smallList = base::take(*SmallUpdates);
 | 
						|
	auto allList = base::take(*AllUpdates);
 | 
						|
	for (auto &update : smallList) {
 | 
						|
		PeerUpdated().notify(std::move(update), true);
 | 
						|
	}
 | 
						|
	for (auto &update : allList) {
 | 
						|
		PeerUpdated().notify(std::move(update), true);
 | 
						|
	}
 | 
						|
 | 
						|
	if (SmallUpdates->isEmpty()) {
 | 
						|
		std::swap(smallList, *SmallUpdates);
 | 
						|
		SmallUpdates->resize(0);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
base::Observable<PeerUpdate, PeerUpdatedHandler> &PeerUpdated() {
 | 
						|
	return PeerUpdatedObservable;
 | 
						|
}
 | 
						|
 | 
						|
rpl::producer<PeerUpdate> PeerUpdateViewer(
 | 
						|
		PeerUpdate::Flags flags) {
 | 
						|
	return [=](const auto &consumer) {
 | 
						|
		auto lifetime = rpl::lifetime();
 | 
						|
		lifetime.make_state<base::Subscription>(
 | 
						|
			PeerUpdated().add_subscription({ flags, [=](
 | 
						|
					const PeerUpdate &update) {
 | 
						|
				consumer.put_next_copy(update);
 | 
						|
			}}));
 | 
						|
		return lifetime;
 | 
						|
	};
 | 
						|
}
 | 
						|
 | 
						|
rpl::producer<PeerUpdate> PeerUpdateViewer(
 | 
						|
		not_null<PeerData*> peer,
 | 
						|
		PeerUpdate::Flags flags) {
 | 
						|
	return PeerUpdateViewer(
 | 
						|
		flags
 | 
						|
	) | rpl::filter([=](const PeerUpdate &update) {
 | 
						|
		return (update.peer == peer);
 | 
						|
	});
 | 
						|
}
 | 
						|
 | 
						|
rpl::producer<PeerUpdate> PeerUpdateValue(
 | 
						|
		not_null<PeerData*> peer,
 | 
						|
		PeerUpdate::Flags flags) {
 | 
						|
	auto initial = PeerUpdate(peer);
 | 
						|
	initial.flags = flags;
 | 
						|
	return rpl::single(
 | 
						|
		initial
 | 
						|
	) | rpl::then(PeerUpdateViewer(peer, flags));
 | 
						|
}
 | 
						|
 | 
						|
} // namespace Notify
 |