Хэндшейк, сервисы, аннотационные блокировки в протоколе, репозитории

src/main/java/com/rosetta/im/database/Repository.java

@@ -1,6 +1,7 @@
 package com.rosetta.im.database;
 
 import java.util.HashMap;
+import java.util.List;
 
 import org.hibernate.Session;
 import org.hibernate.Transaction;
@@ -16,6 +17,11 @@ public abstract class Repository<T> {
         this.entityClass = entityClass;
     }
 
+    /**
+     * Сохранение сущности в базе данных
+     * @param entity сущность для сохранения
+     * @return сохраненная сущность
+     */
     public T save(T entity) {
         return executeInTransaction(session -> {
             session.persist(entity);
@@ -23,6 +29,11 @@ public abstract class Repository<T> {
         });
     }
 
+    /**
+     * Обновление сущности в базе данных
+     * @param entity сущность для обновления
+     * @return обновленная сущность
+     */
     public T update(T entity) {
         return executeInTransaction(session -> {
             session.merge(entity);
@@ -30,6 +41,10 @@ public abstract class Repository<T> {
         });
     }
 
+    /**
+     * Удаление сущности из базы данных
+     * @param entity сущность для удаления
+     */
     public void delete(T entity) {
         executeInTransaction(session -> {
             session.remove(entity);
@@ -76,6 +91,126 @@ public abstract class Repository<T> {
         });
     }
 
+    /**
+     * Удаление сущностей по значению одного поля
+     * @param fieldName поле
+     * @param value значение
+     */
+    public void deleteByField(String fieldName, Object value) {
+        executeInTransaction(session -> {
+            String queryString = "DELETE FROM " + entityClass.getSimpleName() + " WHERE " + fieldName + " = :value";
+            session.createQuery(queryString, entityClass)
+                    .setParameter("value", value)
+                    .executeUpdate();
+            return null;
+        });
+    }
+
+    /**
+     * Поиск всех сущностей по значению одного поля
+     * @param fieldName поле
+     * @param value значение
+     * @return список найденных сущностей
+     */
+    public List<T> findAllByField(String fieldName, Object value) {
+        return executeInSession(session -> {
+            String queryString = "FROM " + entityClass.getSimpleName() + " WHERE " + fieldName + " = :value";
+            return session.createQuery(queryString, entityClass)
+                    .setParameter("value", value)
+                    .list();
+        });
+    }
+
+    /**
+     * Поиск всех сущностей по значению набора полей
+     * @param fields карта полей и их значений
+     * @return список найденных сущностей
+     */
+    public List<T> findAllByField(HashMap<String, Object> fields) {
+        return executeInSession(session -> {
+            StringBuilder queryString = new StringBuilder("FROM " + entityClass.getSimpleName() + " WHERE ");
+            int index = 0;
+            for (String fieldName : fields.keySet()) {
+                if (index > 0) {
+                    queryString.append(" AND ");
+                }
+                queryString.append(fieldName).append(" = :").append(fieldName);
+                index++;
+            }
+            var query = session.createQuery(queryString.toString(), this.entityClass);
+            for (var entry : fields.entrySet()) {
+                query.setParameter(entry.getKey(), entry.getValue());
+            }
+            return query.list();
+        });
+    }
+
+    /**
+     * Поиск всех сущностей, тяжелый метод, лучше не выполнять без необходимости
+     * @return список всех сущностей
+     */
+    public List<T> findAll() {
+        return executeInSession(session -> {
+            String queryString = "FROM " + entityClass.getSimpleName();
+            return session.createQuery(queryString, entityClass).list();
+        });
+    }
+    
+    /**
+     * Подсчет всех сущностей в таблице
+     * @return количество сущностей
+     */
+    public long countAll() {
+        return executeInSession(session -> {
+            String queryString = "SELECT COUNT(*) FROM " + entityClass.getSimpleName();
+            return session.createQuery(queryString, Long.class).uniqueResult();
+        });
+    }
+
+    /**
+     * Подсчет сущностей по значению одного поля
+     * @param fieldName поле
+     * @param value значение
+     * @return количество сущностей
+     */
+    public long countByField(String fieldName, Object value) {
+        return executeInSession(session -> {
+            String queryString = "SELECT COUNT(*) FROM " + entityClass.getSimpleName() + " WHERE " + fieldName + " = :value";
+            return session.createQuery(queryString, Long.class)
+                    .setParameter("value", value)
+                    .uniqueResult();
+        });
+    }
+
+    /**
+     * Подсчет сущностей по набору полей
+     * @param fields карта полей и их значений
+     * @return количество сущностей
+     */
+    public long countByField(HashMap<String, Object> fields) {
+        return executeInSession(session -> {
+            StringBuilder queryString = new StringBuilder("SELECT COUNT(*) FROM " + entityClass.getSimpleName() + " WHERE ");
+            int index = 0;
+            for (String fieldName : fields.keySet()) {
+                if (index > 0) {
+                    queryString.append(" AND ");
+                }
+                queryString.append(fieldName).append(" = :").append(fieldName);
+                index++;
+            }
+            var query = session.createQuery(queryString.toString(), Long.class);
+            for (var entry : fields.entrySet()) {
+                query.setParameter(entry.getKey(), entry.getValue());
+            }
+            return query.uniqueResult();
+        });
+    }
+
+    /**
+     * Обновление полей сущности по заданным условиям
+     * @param fieldsToUpdate поля для обновления
+     * @param whereFields условия для выбора сущностей
+     */
     public void update(HashMap<String, Object> fieldsToUpdate, HashMap<String, Object> whereFields) {
         executeInTransaction(session -> {
             StringBuilder queryString = new StringBuilder("UPDATE " + entityClass.getSimpleName() + " SET ");

src/main/java/com/rosetta/im/database/entity/Device.java

@@ -7,7 +7,6 @@ import jakarta.persistence.Entity;
 import jakarta.persistence.GeneratedValue;
 import jakarta.persistence.GenerationType;
 import jakarta.persistence.Id;
-import jakarta.persistence.PrePersist;
 import jakarta.persistence.Table;
 
 @Entity
@@ -29,14 +28,9 @@ public class Device extends CreateUpdateEntity {
     /**
      * Время завершения сессии устройства
      */
-    @Column(name = "leaveTime", nullable = true)
+    @Column(name = "leaveTime", nullable = true, columnDefinition = "bigint default 0")
     private Long leaveTime;
 
-    @PrePersist
-    protected void onCreate() {
-        this.leaveTime = 0L;
-    }
-
     public Long getId() {
         return id;
     }

src/main/java/com/rosetta/im/database/repository/DeviceRepository.java

@@ -1,5 +1,25 @@
 package com.rosetta.im.database.repository;
 
-public class DeviceRepository {
+import java.util.List;
+
+import com.rosetta.im.database.Repository;
+import com.rosetta.im.database.entity.Device;
+import com.rosetta.im.database.entity.User;
+
+public class DeviceRepository extends Repository<Device> {
+
+    public DeviceRepository() {
+        super(Device.class);
+    }
+
+    /**
+     * Найти все устройства пользователя
+     * @param user пользователь
+     * @return список устройств
+     */
+    public List<Device> findAll(User user) {
+        return this.findAllByField("publicKey", user.getPublicKey());
+    }
+
 
 }