diff --git a/src/main/java/me/youhavetrouble/noted/Storage.java b/src/main/java/me/youhavetrouble/noted/Storage.java index 176c6b3..d5f4165 100644 --- a/src/main/java/me/youhavetrouble/noted/Storage.java +++ b/src/main/java/me/youhavetrouble/noted/Storage.java @@ -18,6 +18,8 @@ import java.util.logging.Logger; public class Storage { + public static final int DATABASE_VERSION = 1; + public final List aliases = new ArrayList<>(); private final Logger logger = Logger.getLogger("Storage"); @@ -46,6 +48,14 @@ public class Storage { logger.warning("Failed to set journal mode to WAL"); } + try (Connection connection = dataSource.getConnection()) { + PreparedStatement preparedStatement = connection.prepareStatement("PRAGMA user_version = ?"); + preparedStatement.setInt(1, DATABASE_VERSION); + preparedStatement.execute(); + } catch (SQLException e) { + logger.warning("Failed to set database version"); + } + createTables(); aliases.addAll(getAliases()); }