fabian/telegram_backup
1
0
Fork 0
mirror of https://github.com/fabianonline/telegram_backup.git synced 2024-11-22 16:56:16 +00:00
Code Issues Releases Wiki Activity

Added a database update to fix the supergroup messages that were wrongly identified as being from a channel.

Browse Source
This commit is contained in:
Fabian Schlenz 2018-02-21 06:31:39 +01:00
parent bb48a2f601
commit de354f548d
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/main/kotlin/de/fabianonline/telegram_backup/DatabaseUpdates.kt
View File

@ -10,6 +10,7 @@ import java.sql.Types
import java.sql.ResultSet import java.sql.ResultSet
import java.sql.PreparedStatement import java.sql.PreparedStatement
import com.github.badoualy.telegram.tl.api.* import com.github.badoualy.telegram.tl.api.*
import com.github.badoualy.telegram.tl.core.TLVector
import org.slf4j.LoggerFactory import org.slf4j.LoggerFactory
import org.slf4j.Logger import org.slf4j.Logger
import de.fabianonline.telegram_backup.mediafilemanager.FileManagerFactory import de.fabianonline.telegram_backup.mediafilemanager.FileManagerFactory
@ -30,6 +31,7 @@ class DatabaseUpdates(protected var conn: Connection, protected var db: Database
register(DB_Update_6(conn, db)) register(DB_Update_6(conn, db))
register(DB_Update_7(conn, db)) register(DB_Update_7(conn, db))
register(DB_Update_8(conn, db)) register(DB_Update_8(conn, db))
register(DB_Update_9(conn, db))
} }
fun doUpdates() { fun doUpdates() {
@ -369,3 +371,30 @@ internal class DB_Update_8(conn: Connection, db: Database) : DatabaseUpdate(conn
execute("CREATE UNIQUE INDEX unique_messages ON messages (source_type, source_id, message_id)") execute("CREATE UNIQUE INDEX unique_messages ON messages (source_type, source_id, message_id)")
} }
} }
internal class DB_Update_9(conn: Connection, db: Database) : DatabaseUpdate(conn, db) {
override val version: Int
get() = 9
override val needsBackup = true
@Throws(SQLException::class)
override fun _doUpdate() {
val logger = LoggerFactory.getLogger(DB_Update_9::class.java)
println(" Updating supergroup channel message data (this might take some time)...")
val rs = stmt.executeQuery("SELECT id, data, source_id FROM messages WHERE source_type='channel' and sender_id IS NULL and api_layer=53")
val messages = TLVector<TLAbsMessage>()
val messages_to_delete = mutableListOf<Int>()
var i = 0
while (rs.next()) {
i++
val msg = Database.bytesToTLMessage(rs.getBytes(2))
if (msg!!.getFromId() != null) {
messages.add(msg)
messages_to_delete.add(rs.getInt(1))
}
}
logger.info("Converted ${messages.size} of ${i} messages.")
db.saveMessages(messages, api_layer=53, source_type=MessageSource.SUPERGROUP)
execute("DELETE FROM messages WHERE id IN (" + messages_to_delete.joinToString() + ")")
}
}