Почта v2: ReceiveOutcomingMessage без авторизации и атомарная вставка пары

shine-server-db/src/main/java/shine/db/dao/SignedMessagesV2DAO.java

@@ -6,6 +6,7 @@ import shine.db.entities.SignedMessageV2Entry;
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
+import java.sql.SQLException;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -54,6 +55,69 @@ public final class SignedMessagesV2DAO {
         }
     }
 
+    /**
+     * Атомарная вставка пары блоков: либо вставляются оба, либо не вставляется ни один.
+     * Возвращает true только если обе записи добавлены в БД.
+     * Если хотя бы одна запись уже существует (или конфликтует по уникальности), возвращает false.
+     */
+    public boolean insertPairBothOrNothing(SignedMessageV2Entry first, SignedMessageV2Entry second) throws Exception {
+        try (Connection c = db.getConnection()) {
+            boolean prevAutoCommit = c.getAutoCommit();
+            c.setAutoCommit(false);
+            try {
+                int insertedFirst = insertStrict(c, first);
+                int insertedSecond = insertStrict(c, second);
+                if (insertedFirst == 1 && insertedSecond == 1) {
+                    c.commit();
+                    return true;
+                }
+                c.rollback();
+                return false;
+            } catch (SQLException sqlEx) {
+                try { c.rollback(); } catch (Exception ignored) {}
+                if (isConstraintViolation(sqlEx)) {
+                    return false;
+                }
+                throw sqlEx;
+            } finally {
+                c.setAutoCommit(prevAutoCommit);
+            }
+        }
+    }
+
+    private int insertStrict(Connection c, SignedMessageV2Entry e) throws SQLException {
+        String sql = """
+            INSERT INTO signed_messages_v2 (
+                message_key, base_key, target_login, from_login, to_login,
+                time_ms, nonce, message_type, raw_block, created_at_ms,
+                source_api, origin_session_id, receipt_ref_base_key, receipt_ref_type
+            ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            """;
+        try (PreparedStatement ps = c.prepareStatement(sql)) {
+            ps.setString(1, e.getMessageKey());
+            ps.setString(2, e.getBaseKey());
+            ps.setString(3, e.getTargetLogin());
+            ps.setString(4, e.getFromLogin());
+            ps.setString(5, e.getToLogin());
+            ps.setLong(6, e.getTimeMs());
+            ps.setLong(7, e.getNonce());
+            ps.setInt(8, e.getMessageType());
+            ps.setBytes(9, e.getRawBlock());
+            ps.setLong(10, e.getCreatedAtMs());
+            ps.setString(11, e.getSourceApi());
+            ps.setString(12, e.getOriginSessionId());
+            ps.setString(13, e.getReceiptRefBaseKey());
+            if (e.getReceiptRefType() == null) ps.setObject(14, null);
+            else ps.setInt(14, e.getReceiptRefType());
+            return ps.executeUpdate();
+        }
+    }
+
+    private boolean isConstraintViolation(SQLException ex) {
+        String msg = String.valueOf(ex.getMessage()).toLowerCase();
+        return msg.contains("constraint") || msg.contains("unique") || msg.contains("primary key");
+    }
+
     public SignedMessageV2Entry getByMessageKey(String messageKey) throws Exception {
         try (Connection c = db.getConnection()) {
             String sql = """