90 lines
4.9 KiB
Java
90 lines
4.9 KiB
Java
package com.rosetta.im.service.dispatch;
|
||
|
||
import java.util.List;
|
||
|
||
import com.rosetta.im.Failures;
|
||
import com.rosetta.im.client.ClientManager;
|
||
import com.rosetta.im.client.tags.ECIAuthentificate;
|
||
import com.rosetta.im.database.repository.BufferRepository;
|
||
import com.rosetta.im.database.repository.GroupRepository;
|
||
import com.rosetta.im.packet.base.PacketBaseDialog;
|
||
import com.rosetta.im.service.services.BufferService;
|
||
|
||
import io.orprotocol.ProtocolException;
|
||
import io.orprotocol.client.Client;
|
||
import io.orprotocol.packet.PacketManager;
|
||
|
||
/**
|
||
* Диспетчер сообщений, который отвечает за отправку сообщений получателям, а так же за сохранение сообщений в буфер для офлайн получателей и для синхронизации
|
||
* Такой диспетчер нужен для того, чтобы не загромождать логику обработчиков сообщений, а так же для того, чтобы
|
||
* централизовать логику отправки сообщений и сохранения их в буфер
|
||
* Например, при отправке группового сообщения, диспетчер сам достает участников группы и
|
||
* отправляет сообщение каждому участнику, а так же сохраняет сообщение в буфер для каждого участника, который офлайн
|
||
*/
|
||
public class MessageDispatcher {
|
||
|
||
private final GroupRepository groupRepository = new GroupRepository();
|
||
private final ClientManager clientManager;
|
||
private final BufferRepository bufferRepository = new BufferRepository();
|
||
private final BufferService bufferService;
|
||
|
||
public MessageDispatcher(ClientManager clientManager, PacketManager packetManager) {
|
||
this.clientManager = clientManager;
|
||
this.bufferService = new BufferService(bufferRepository, packetManager);
|
||
}
|
||
|
||
/**
|
||
* Отправляет групповое сообщение всем участникам группы, кроме отправителя
|
||
* @param packet пакет с групповым сообщением
|
||
*/
|
||
public void sendGroup(PacketBaseDialog packet, Client client, ECIAuthentificate eciAuthentificate) throws ProtocolException {
|
||
String toPublicKey = packet.getToPublicKey();
|
||
List<String> groupMembersPublicKeys = this.groupRepository.findGroupMembers(toPublicKey.replace("#group:", ""));
|
||
if(groupMembersPublicKeys.isEmpty()){
|
||
/**
|
||
* Если группа не найдена или в группе нет участников, то в такую отправить
|
||
* сообщение нельзя
|
||
*/
|
||
client.disconnect(Failures.DATA_MISSMATCH);
|
||
return;
|
||
}
|
||
if(!groupMembersPublicKeys.contains(eciAuthentificate.getPublicKey())){
|
||
/**
|
||
* Если отправитель не является участником группы, то он не может отправлять
|
||
* сообщения в эту группу
|
||
*/
|
||
client.disconnect(Failures.USER_NOT_IN_GROUP);
|
||
return;
|
||
}
|
||
/**
|
||
* Отправляем всем участникам группы, кроме отправителя этот пакет, попутно не забывая проверить, а не один ли он в группе
|
||
*/
|
||
groupMembersPublicKeys.remove(eciAuthentificate.getPublicKey());
|
||
if(groupMembersPublicKeys.isEmpty()){
|
||
/**
|
||
* Если отправитель был единственным участником группы, то отправлять сообщение некуда,
|
||
* не кикаем пользователя
|
||
*/
|
||
return;
|
||
}
|
||
this.clientManager.sendPacketToPK(groupMembersPublicKeys, packet);
|
||
|
||
//TODO: Сохранить сообщение в буфер для группы, чтобы группы тоже синхронизировались
|
||
}
|
||
|
||
/**
|
||
* Отправляет личное сообщение получателю
|
||
* @param packet пакет с личным сообщением
|
||
*/
|
||
public void sendPeer(PacketBaseDialog packet, Client client) throws ProtocolException {
|
||
String fromPublicKey = packet.getFromPublicKey();
|
||
String toPublicKey = packet.getToPublicKey();
|
||
this.clientManager.sendPacketToPK(toPublicKey, packet);
|
||
/**
|
||
* Сохраняем сообщение в буфер на случай если получатель офлайн, или нам нужна будет синхронизация сообщений для получателя
|
||
*/
|
||
this.bufferService.pushPacketToBuffer(fromPublicKey, toPublicKey, packet);
|
||
}
|
||
|
||
}
|