Преглед изворни кода

Remove unused dbms migration function

Tulir Asokan пре 3 година
родитељ
комит
7ccc0d005e
1 измењених фајлова са 0 додато и 200 уклоњено
  1. 0 200
      database/migrate.go

+ 0 - 200
database/migrate.go

@@ -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)
-	}
-}