update

src/infrastructure/sqlite/migrations.rs

@@ -1,8 +1,5 @@
 use rusqlite::{Connection, Result};
 
-/// Versão atual do schema do banco de dados.
-const VERSAO_SCHEMA_ATUAL: i64 = 3;
-
 /// Aplica todas as migrations necessárias para atualizar o banco
 /// para a versão mais recente.
 pub fn aplicar_migrations(conn: &Connection) -> Result<()> {
@@ -29,26 +26,17 @@ pub fn aplicar_migrations(conn: &Connection) -> Result<()> {
         migration_v3(conn)?;
     }
 
-    if versao_atual == 0 {
-        conn.execute(
-            "INSERT INTO schema_version (versao) VALUES (?1);",
-            [VERSAO_SCHEMA_ATUAL],
-        )?;
-    } else if versao_atual < VERSAO_SCHEMA_ATUAL {
-        conn.execute(
-            "UPDATE schema_version SET versao = ?1;",
-            [VERSAO_SCHEMA_ATUAL],
-        )?;
-    }
-
     Ok(())
 }
 
 /// Migration v3: adicionar colunas de tipo de documento nos layouts.
 fn migration_v3(conn: &Connection) -> Result<()> {
     conn.execute_batch(
-        "ALTER TABLE layouts ADD COLUMN indice_documento_tipo INTEGER;
-         ALTER TABLE layouts ADD COLUMN pos_documento_tipo TEXT;",
+        "BEGIN;
+         ALTER TABLE layouts ADD COLUMN indice_documento_tipo INTEGER;
+         ALTER TABLE layouts ADD COLUMN pos_documento_tipo TEXT;
+         UPDATE schema_version SET versao = 3;
+         COMMIT;",
     )?;
     Ok(())
 }
@@ -59,12 +47,15 @@ fn migration_v3(conn: &Connection) -> Result<()> {
 /// em bancos que já possuem nomes repetidos.
 fn migration_v2(conn: &Connection) -> Result<()> {
     conn.execute_batch(
-        "UPDATE layouts
-            SET nome = nome || ' (' || id || ')'
-            WHERE id NOT IN (
-                SELECT MIN(id) FROM layouts GROUP BY nome
-            );
-         CREATE UNIQUE INDEX IF NOT EXISTS idx_layouts_nome ON layouts (nome);",
+        "BEGIN;
+         UPDATE layouts
+             SET nome = nome || ' (' || id || ')'
+             WHERE id NOT IN (
+                 SELECT MIN(id) FROM layouts GROUP BY nome
+             );
+         CREATE UNIQUE INDEX IF NOT EXISTS idx_layouts_nome ON layouts (nome);
+         UPDATE schema_version SET versao = 2;
+         COMMIT;",
     )?;
     Ok(())
 }
@@ -72,27 +63,30 @@ fn migration_v2(conn: &Connection) -> Result<()> {
 /// Migration v1: criar tabela de layouts.
 fn migration_v1(conn: &Connection) -> Result<()> {
     conn.execute_batch(
-        "CREATE TABLE IF NOT EXISTS layouts (
-            id               INTEGER PRIMARY KEY AUTOINCREMENT,
-            nome             TEXT NOT NULL,
-            tipo             TEXT NOT NULL CHECK(tipo IN ('csv', 'xlsx')),
+        "BEGIN;
+         CREATE TABLE IF NOT EXISTS layouts (
+             id               INTEGER PRIMARY KEY AUTOINCREMENT,
+             nome             TEXT NOT NULL,
+             tipo             TEXT NOT NULL CHECK(tipo IN ('csv', 'xlsx')),
 
-            -- Campos CSV
-            delimitador      TEXT,
-            encoding         TEXT,
-            linha_cabecalho  INTEGER,
-            indice_numero    INTEGER,
-            indice_serie     INTEGER,
-            indice_valor     INTEGER,
-            indice_data      INTEGER,
+             -- Campos CSV
+             delimitador      TEXT,
+             encoding         TEXT,
+             linha_cabecalho  INTEGER,
+             indice_numero    INTEGER,
+             indice_serie     INTEGER,
+             indice_valor     INTEGER,
+             indice_data      INTEGER,
 
-            -- Campos XLSX
-            aba              TEXT,
-            pos_numero       TEXT,
-            pos_serie        TEXT,
-            pos_valor        TEXT,
-            pos_data         TEXT
-        );",
+             -- Campos XLSX
+             aba              TEXT,
+             pos_numero       TEXT,
+             pos_serie        TEXT,
+             pos_valor        TEXT,
+             pos_data         TEXT
+         );
+         INSERT INTO schema_version (versao) VALUES (1);
+         COMMIT;",
     )?;
     Ok(())
 }