Rosetta/rosetta-wss
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Исправление ошибок и новые уведомления #16

Merged
Royce59 merged 3 commits from dev into main 2026-03-20 18:13:00 +00:00
Conversation 0 Commits 3 Files Changed 5 +75 -38
5 changed files with 75 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitea/workflows/build.yaml
View File

@@ -1,4 +1,5 @@
name: Build rosetta-wss
run-name: Build and Deploy WSS
on:
push:
branches:

10
src/main/java/im/rosetta/client/ClientManager.java
View File

@@ -1,7 +1,7 @@
package im.rosetta.client;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import im.rosetta.client.tags.ECIAuthentificate;
@@ -33,7 +33,7 @@ public class ClientManager {
}
public boolean isClientConnected(String publicKey) {
HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
if(clients == null){
/**
* Нет клиентов с таким публичным ключом
@@ -59,7 +59,7 @@ public class ClientManager {
* @throws ProtocolException если произошла ошибка при отправке пакета клиенту
*/
public void sendPacketToAuthorizedPK(String publicKey, Packet packet) throws ProtocolException {
HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
if(clients == null){
/**
* Нет клиентов с таким публичным ключом, значит отправлять некому
@@ -91,7 +91,7 @@ public class ClientManager {
*/
public void retranslate(Client client, Packet packet) throws ProtocolException{
ECIAuthentificate eciAuthentificate = client.getTag(ECIAuthentificate.class);
HashSet<Client> clients = this.clientIndexer
Set<Client> clients = this.clientIndexer
.getClients(ECIAuthentificate.class, "publicKey", eciAuthentificate.getPublicKey());
if(clients == null){
/**
@@ -129,7 +129,7 @@ public class ClientManager {
* @return список клиентов с таким публичным ключом, может быть пустым, если клиентов с таким публичным ключом нет
*/
public List<Client> getPKClients(String publicKey) {
HashSet<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
Set<Client> clients = this.clientIndexer.getClients(ECIAuthentificate.class, "publicKey", publicKey);
if(clients == null){
/**
* Нет клиентов с таким публичным ключом

50
src/main/java/im/rosetta/service/dispatch/FirebaseDispatcher.java
View File

@@ -9,6 +9,9 @@ import java.util.concurrent.Executors;
import com.google.auth.oauth2.GoogleCredentials;
import com.google.firebase.FirebaseApp;
import com.google.firebase.FirebaseOptions;
import com.google.firebase.messaging.ApnsConfig;
import com.google.firebase.messaging.Aps;
import com.google.firebase.messaging.ApsAlert;
import com.google.firebase.messaging.FirebaseMessaging;
import com.google.firebase.messaging.Message;
import com.google.firebase.messaging.Notification;
@@ -51,13 +54,7 @@ public class FirebaseDispatcher {
}
}
/**
* Отправляет push-уведомление пользователю с данным публичным ключом (асинхронно)
* @param publicKey публичный ключ пользователя
* @param title заголовок уведомления
* @param messageText текст уведомления
*/
public void sendPushNotification(String publicKey, String title, String messageText) {
public void sendPushNotification(String publicKey, String title, String messageText, String senderPublicKey) {
executor.submit(() -> {
try {
List<String> tokens = userService.getNotificationsTokens(publicKey);
@@ -72,6 +69,17 @@ public class FirebaseDispatcher {
.setTitle(title)
.setBody(messageText)
.build())
.setApnsConfig(ApnsConfig.builder()
.setAps(Aps.builder()
.setMutableContent(true)
.setSound("default")
.setAlert(ApsAlert.builder()
.setTitle(title)
.setBody(messageText)
.build())
.build())
.build())
.putData("sender_public_key", senderPublicKey)
.setToken(token)
.build();
@@ -86,6 +94,30 @@ public class FirebaseDispatcher {
});
}
/**
* Отправляет push-уведомление пользователю с данным публичным ключом (асинхронно)
* @param publicKey публичный ключ пользователя
* @param title заголовок уведомления
* @param messageText текст уведомления
*/
public void sendPushNotification(String publicKey, String title, String messageText) {
sendPushNotification(publicKey, title, messageText, null);
}
/**
* Отправляет push-уведомление нескольким пользователям (асинхронно)
* @param publicKeys список публичных ключей пользователей
* @param title заголовок уведомления
* @param messageText текст уведомления
*/
public void sendPushNotification(List<String> publicKeys, String title, String messageText, String senderPublicKey) {
executor.submit(() -> {
for (String publicKey : publicKeys) {
sendPushNotificationSync(publicKey, title, messageText, senderPublicKey);
}
});
}
/**
* Отправляет push-уведомление нескольким пользователям (асинхронно)
* @param publicKeys список публичных ключей пользователей
@@ -95,12 +127,12 @@ public class FirebaseDispatcher {
public void sendPushNotification(List<String> publicKeys, String title, String messageText) {
executor.submit(() -> {
for (String publicKey : publicKeys) {
sendPushNotificationSync(publicKey, title, messageText);
sendPushNotificationSync(publicKey, title, messageText, null);
}
});
}
private void sendPushNotificationSync(String publicKey, String title, String messageText) {
private void sendPushNotificationSync(String publicKey, String title, String messageText, String senderPublicKey) {
try {
List<String> tokens = userService.getNotificationsTokens(publicKey);
if (tokens == null || tokens.isEmpty()) {

18
src/main/java/im/rosetta/service/dispatch/MessageDispatcher.java
View File

@@ -4,13 +4,15 @@ import java.util.List;
import im.rosetta.client.ClientManager;
import im.rosetta.client.tags.ECIAuthentificate;
import im.rosetta.database.entity.User;
import im.rosetta.database.repository.BufferRepository;
import im.rosetta.database.repository.GroupRepository;
import im.rosetta.database.repository.UserRepository;
import im.rosetta.packet.Packet11Typeing;
import im.rosetta.packet.Packet7Read;
import im.rosetta.packet.base.PacketBaseDialog;
import im.rosetta.service.services.BufferService;
import im.rosetta.service.services.UserService;
import io.orprotocol.ProtocolException;
import io.orprotocol.client.Client;
import io.orprotocol.packet.PacketManager;
@@ -29,6 +31,8 @@ public class MessageDispatcher {
private final BufferRepository bufferRepository = new BufferRepository();
private final BufferService bufferService;
private final FirebaseDispatcher firebaseDispatcher = new FirebaseDispatcher();
private final UserRepository userRepository = new UserRepository();
private final UserService userService = new UserService(userRepository);
public MessageDispatcher(ClientManager clientManager, PacketManager packetManager) {
this.clientManager = clientManager;
@@ -101,7 +105,7 @@ public class MessageDispatcher {
/**
* Отправляем PUSH уведомление
*/
this.firebaseDispatcher.sendPushNotification(groupMembersPublicKeys, "Rosetta", "New message in group");
this.firebaseDispatcher.sendPushNotification(groupMembersPublicKeys, "Rosetta", "New message in group", toPublicKey.replace("#group:", ""));
}
/**
@@ -115,7 +119,13 @@ public class MessageDispatcher {
public void sendPeer(PacketBaseDialog packet, Client client, boolean bufferizationNeed) throws ProtocolException {
String fromPublicKey = packet.getFromPublicKey();
String toPublicKey = packet.getToPublicKey();
User user = this.userService.fromClient(client);
if(user == null){
/**
* Если пользователь не найден, то не отправляем сообщение, так как у нас нет информации о том, кто отправляет сообщение
*/
return;
}
/**
* Ретранслируем сообщение ВСЕМ авторизованным сессиям отправителя КРОМЕ текущей,
* чтобы синхронизировать отправленные сообщения
@@ -154,7 +164,7 @@ public class MessageDispatcher {
/**
* Отправляем PUSH уведомление получателю
*/
this.firebaseDispatcher.sendPushNotification(toPublicKey, "Rosetta", "New message");
this.firebaseDispatcher.sendPushNotification(toPublicKey, user.getTitle(), "New message");
}
/**

34
src/main/java/io/orprotocol/index/ClientIndexer.java
View File

@@ -1,7 +1,7 @@
package io.orprotocol.index;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import io.orprotocol.client.Client;
@@ -19,14 +19,8 @@ public class ClientIndexer {
* Ключ третьего уровня - значение поля.
* Значение - клиент.
* Структура: tagClass -> indexName -> indexValue -> Client
*
*
* В качестве клиента используем HashSet потому что он очень хорош для операции contains,
* в отличие от ArrayList, который при поиске перебирает элементы один за другим,
* HashSet использует хэш таблицу и находит нужный объект почти
* мгновенно, независимо от того, 10 там элементов или миллион.
*/
private final Map<Class<?>, Map<String, Map<Object, HashSet<Client>>>> indices
private final ConcurrentHashMap<Class<?>, ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>>> indices
= new ConcurrentHashMap<>();
public <T extends ECITag> void indexTag(Client client, Class<T> tagClass, T tag) {
@@ -53,7 +47,7 @@ public class ClientIndexer {
* Если использовать putIfAbsent, то он вернет null если значение там уже есть,
* что не подходит
*/
Map<String, Map<Object, HashSet<Client>>> tagIndices = this.indices.computeIfAbsent(
ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = this.indices.computeIfAbsent(
tagClass,
k -> new ConcurrentHashMap<>());
@@ -76,7 +70,7 @@ public class ClientIndexer {
/**
* Инициализируем имя индекса, почему compute, а не put - написано выше
*/
Map<Object, HashSet<Client>> index = tagIndices.computeIfAbsent(
Map<Object, Set<Client>> index = tagIndices.computeIfAbsent(
indexName,
k -> new ConcurrentHashMap<>()
);
@@ -86,7 +80,7 @@ public class ClientIndexer {
* с одинковыми индексами, хотя такого лучше избегать, желательно чтобы индексы
* были уникальными, тогда обработка будет быстрее всего
*/
HashSet<Client> clients = index.computeIfAbsent(indexValue, k -> new HashSet<>());
Set<Client> clients = index.computeIfAbsent(indexValue, k -> ConcurrentHashMap.newKeySet());
/**
* Добавляем клиента в инициализиованный индекс
*/
@@ -117,7 +111,7 @@ public class ClientIndexer {
* @internal
*/
public <T extends ECITag> void removeTagIndex(Client client, Class<T> tagClass) {
Map<String, Map<Object, HashSet<Client>>> tagIndices = indices.get(tagClass);
ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = indices.get(tagClass);
if (tagIndices == null) {
/**
* Индекса и так не было, удалять нечего
@@ -128,12 +122,12 @@ public class ClientIndexer {
/**
* Удаляем все ключи indexName по tagClass если Client == client
*/
for (Map<Object, HashSet<Client>> index : tagIndices.values()) {
for (ConcurrentHashMap<Object, Set<Client>> index : tagIndices.values()) {
/**
* contains всегда использует переопределенный equals, по этому
* обьекты клиентов сравниваются нормально
*/
for(HashSet<Client> clients : index.values()){
for(Set<Client> clients : index.values()){
if(!clients.contains(client)){
continue;
}
@@ -147,9 +141,9 @@ public class ClientIndexer {
* @internal
*/
public void removeClientFromIndex(Client client) {
for(Map<String, Map<Object, HashSet<Client>>> tagIndices : this.indices.values()){
for(Map<Object, HashSet<Client>> index : tagIndices.values()){
for(HashSet<Client> clients : index.values()){
for(Map<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices : this.indices.values()){
for(Map<Object, Set<Client>> index : tagIndices.values()){
for(Set<Client> clients : index.values()){
/**
* Этот тройной цикл не такой страшный, так как мы всего лишь
* проходим по всем тегам (их немного), дальше идем по всем значениям в тегах
@@ -173,7 +167,7 @@ public class ClientIndexer {
* @param indexValue значение в теге
* @return список клиентов с заданными значениями
*/
public <T extends ECITag> HashSet<Client> getClients(Class<T> tagClass, String indexName, Object indexValue) {
public <T extends ECITag> Set<Client> getClients(Class<T> tagClass, String indexName, Object indexValue) {
if(indexName == null || indexValue == null){
return null;
}
@@ -181,12 +175,12 @@ public class ClientIndexer {
/**
* Получение по индексу простое, так как каждое из заданных значений и есть ключ
*/
Map<String, Map<Object, HashSet<Client>>> tagIndices = indices.get(tagClass);
ConcurrentHashMap<String, ConcurrentHashMap<Object, Set<Client>>> tagIndices = indices.get(tagClass);
if (tagIndices == null) {
return null;
}
Map<Object, HashSet<Client>> index = tagIndices.get(indexName);
ConcurrentHashMap<Object, Set<Client>> index = tagIndices.get(indexName);
if (index == null) {
return null;
}