forked from MirrorHub/mautrix-whatsapp
Remove unused dbms migration function
This commit is contained in:
Tulir Asokan
parent
08d5b10902
commit
7ccc0d005e
1 changed files with 0 additions and 200 deletions
|
|
@ -1,200 +0,0 @@
|
|||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"math"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func countRows(db *Database, table string) (int, error) {
|
||||
countRow := db.QueryRow(fmt.Sprintf("SELECT COUNT(*) FROM %s", table))
|
||||
var count int
|
||||
err := countRow.Scan(&count)
|
||||
return count, err
|
||||
}
|
||||
|
||||
const VariableCountLimit = 512
|
||||
|
||||
func migrateTable(old *Database, new *Database, table string, columns ...string) error {
|
||||
columnNames := strings.Join(columns, ",")
|
||||
fmt.Printf("Migrating %s: ", table)
|
||||
rowCount, err := countRows(old, table)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Print("found ", rowCount, " rows of data, ")
|
||||
rows, err := old.Query(fmt.Sprintf("SELECT %s FROM \"%s\"", columnNames, table))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
serverColNames, err := rows.Columns()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
colCount := len(serverColNames)
|
||||
valueStringFormat := strings.Repeat("$%d, ", colCount)
|
||||
valueStringFormat = fmt.Sprintf("(%s)", valueStringFormat[:len(valueStringFormat)-2])
|
||||
cols := make([]interface{}, colCount)
|
||||
colPtrs := make([]interface{}, colCount)
|
||||
for i := 0; i < colCount; i++ {
|
||||
colPtrs[i] = &cols[i]
|
||||
}
|
||||
batchSize := VariableCountLimit / colCount
|
||||
values := make([]interface{}, batchSize*colCount)
|
||||
valueStrings := make([]string, batchSize)
|
||||
var inserted int64
|
||||
batchCount := int(math.Ceil(float64(rowCount) / float64(batchSize)))
|
||||
tx, err := new.Begin()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Printf("migrating in %d batches: ", batchCount)
|
||||
for rowCount > 0 {
|
||||
var i int
|
||||
for ; rows.Next() && i < batchSize; i++ {
|
||||
colPtrs := make([]interface{}, colCount)
|
||||
valueStringArgs := make([]interface{}, colCount)
|
||||
for j := 0; j < colCount; j++ {
|
||||
pos := i*colCount + j
|
||||
colPtrs[j] = &values[pos]
|
||||
valueStringArgs[j] = pos + 1
|
||||
}
|
||||
valueStrings[i] = fmt.Sprintf(valueStringFormat, valueStringArgs...)
|
||||
err = rows.Scan(colPtrs...)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
slicedValues := values
|
||||
slicedValueStrings := valueStrings
|
||||
if i < len(valueStrings) {
|
||||
slicedValueStrings = slicedValueStrings[:i]
|
||||
slicedValues = slicedValues[:i*colCount]
|
||||
}
|
||||
if len(slicedValues) == 0 {
|
||||
break
|
||||
}
|
||||
res, err := tx.Exec(fmt.Sprintf("INSERT INTO \"%s\" (%s) VALUES %s", table, columnNames, strings.Join(slicedValueStrings, ",")), slicedValues...)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
count, _ := res.RowsAffected()
|
||||
inserted += count
|
||||
rowCount -= batchSize
|
||||
fmt.Print("#")
|
||||
}
|
||||
err = tx.Commit()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Println(" -- done with", inserted, "rows inserted")
|
||||
return nil
|
||||
}
|
||||
|
||||
func Migrate(old *Database, new *Database) {
|
||||
err := migrateTable(old, new, "portal", "jid", "receiver", "mxid", "name", "topic", "avatar", "avatar_url", "encrypted", "first_event_id", "next_batch_id", "relay_user_id")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "user", "mxid", "management_room", "username", "agent", "device")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "puppet", "username", "avatar", "displayname", "name_quality", "custom_mxid", "access_token", "next_batch", "avatar_url", "enable_presence", "enable_receipts")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "message", "chat_jid", "chat_receiver", "jid", "mxid", "sender", "timestamp", "sent", "decryption_error")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "mx_registrations", "user_id")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "mx_user_profile", "room_id", "user_id", "membership", "displayname", "avatar_url")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "mx_room_state", "room_id", "power_levels")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_account", "account_id", "device_id", "shared", "sync_token", "account")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_message_index", "sender_key", "session_id", `"index"`, "event_id", "timestamp")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_tracked_user", "user_id")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_device", "user_id", "device_id", "identity_key", "signing_key", "trust", "deleted", "name")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_olm_session", "account_id", "session_id", "sender_key", "session", "created_at", "last_used")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_megolm_inbound_session", "account_id", "session_id", "sender_key", "signing_key", "room_id", "session", "forwarding_chains", "withheld_code", "withheld_reason")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_megolm_outbound_session", "account_id", "room_id", "session_id", "session", "shared", "max_messages", "message_count", "max_age", "created_at", "last_used")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_cross_signing_keys", "user_id", "usage", "key")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "crypto_cross_signing_signatures", "signed_user_id", "signed_key", "signer_user_id", "signer_key", "signature")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
// Migrate whatsmeow tables.
|
||||
err = migrateTable(old, new, "whatsmeow_device", "jid", "registration_id", "noise_key", "identity_key", "signed_pre_key", "signed_pre_key_id", "signed_pre_key_sig", "adv_key", "adv_details", "adv_account_sig", "adv_device_sig", "platform", "business_name", "push_name")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_identity_keys", "our_jid", "their_id", "identity")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_pre_keys", "jid", "key_id", "key", "uploaded")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_sessions", "our_jid", "their_id", "session")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_sender_keys", "our_jid", "chat_id", "sender_id", "sender_key")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_app_state_sync_keys", "jid", "key_id", "key_data", "timestamp", "fingerprint")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_app_state_version", "jid", "name", "version", "hash")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_app_state_mutation_macs", "jid", "name", "version", "index_mac", "value_mac")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_contacts", "our_jid", "their_jid", "first_name", "full_name", "push_name", "business_name")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
err = migrateTable(old, new, "whatsmeow_chat_settings", "our_jid", "chat_jid", "muted_until", "pinned", "archived")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue