fabian/telegram_backup
1
0
Fork 0
mirror of https://github.com/fabianonline/telegram_backup.git synced 2024-09-28 18:05:49 +00:00
Code Issues Releases Wiki Activity

Added a method to create database backups before doing huge database update stuff.

Browse Source
This commit is contained in:
Fabian Schlenz 2016-07-08 06:58:58 +02:00
parent db52b8da47
commit 4e227e55c8
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/de/fabianonline/telegram_backup/Config.java
View File

@ -30,6 +30,7 @@ public class Config {
public static final String FILE_NAME_AUTH_KEY = "auth.dat";
public static final String FILE_NAME_DC = "dc.dat";
public static final String FILE_NAME_DB = "database.sqlite";
public static final String FILE_NAME_DB_BACKUP = "database.version_%d.backup.sqlite";
public static final String FILE_FILES_BASE = "files";
public static final String FILE_STICKER_BASE = "stickers";

17
src/main/java/de/fabianonline/telegram_backup/Database.java
View File

@ -36,6 +36,8 @@ import java.io.IOException;
import java.util.LinkedList;
import java.util.LinkedHashMap;
import java.util.HashMap;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
import de.fabianonline.telegram_backup.mediafilemanager.AbstractMediaFileManager;
import de.fabianonline.telegram_backup.mediafilemanager.FileManagerFactory;
@ -151,6 +153,7 @@ public class Database {
version = 5;
}
if (version==5) {
backupDatabase(5);
System.out.println(" Updating to version 6...");
stmt.executeUpdate(
"CREATE TABLE messages_new (\n" +
@ -239,6 +242,20 @@ public class Database {
}
}
private void backupDatabase(int currentVersion) {
String filename = String.format(Config.FILE_NAME_DB_BACKUP, currentVersion);
System.out.println(" Creating a backup of your database as " + filename);
try {
Files.copy(
new File(user_manager.getFileBase() + Config.FILE_NAME_DB).toPath(),
new File(user_manager.getFileBase() + filename).toPath(),
StandardCopyOption.REPLACE_EXISTING);
} catch (IOException e) {
e.printStackTrace();
throw new RuntimeException("Could not create backup.");
}
}
public int getTopMessageID() {
try {
ResultSet rs = stmt.executeQuery("SELECT MAX(id) FROM messages");