mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-09 15:13:12 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update migrations to set "correct" history visibility

Browse source
This commit is contained in:
Till Faelligen 2022-07-19 13:30:43 +02:00
parent 56d8a2371f
commit 503afceccb
2 changed files with 74 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
syncapi/storage/postgres/deltas/2022061412000000_history_visibility_column.go
View file

@ -16,9 +16,11 @@ package deltas
import (
"database/sql"
"encoding/json"
"fmt"
"github.com/matrix-org/dendrite/internal/sqlutil"
"github.com/matrix-org/gomatrixserverlib"
)
func LoadAddHistoryVisibilityColumn(m *sqlutil.Migrations) {
@ -35,6 +37,41 @@ func UpAddHistoryVisibilityColumn(tx *sql.Tx) error {
if err != nil {
return fmt.Errorf("failed to execute upgrade: %w", err)
}
// get the current room history visibility
rows, err := tx.Query(`SELECT DISTINCT room_id, headered_event_json FROM syncapi_current_room_state
WHERE type = 'm.room.history_visibility' AND state_key = '';
`)
if err != nil {
return fmt.Errorf("failed to query current room state: %w", err)
}
var eventBytes []byte
var roomID string
var event gomatrixserverlib.HeaderedEvent
var hisVis gomatrixserverlib.HistoryVisibility
historyVisibilities := make(map[string]gomatrixserverlib.HistoryVisibility)
for rows.Next() {
if err = rows.Scan(&roomID, &eventBytes); err != nil {
return fmt.Errorf("failed to scan row: %w", err)
}
if err = json.Unmarshal(eventBytes, &event); err != nil {
return fmt.Errorf("failed to unmarshal event: %w", err)
}
historyVisibilities[roomID] = gomatrixserverlib.HistoryVisibilityShared
if hisVis, err = event.HistoryVisibility(); err == nil {
historyVisibilities[roomID] = hisVis
}
}
// update the history visibility
for roomID, hisVis = range historyVisibilities {
_, err = tx.Exec(`UPDATE syncapi_output_room_events SET history_visibility = $1
WHERE type IN ('m.room.message', 'm.room.encrypted') AND room_id = $2`, hisVis, roomID)
if err != nil {
return fmt.Errorf("failed to update history visibility: %w", err)
}
}
return nil
}

37
syncapi/storage/sqlite3/deltas/2022061412000000_history_visibility_column.go
View file

@ -16,9 +16,11 @@ package deltas
import (
"database/sql"
"encoding/json"
"fmt"
"github.com/matrix-org/dendrite/internal/sqlutil"
"github.com/matrix-org/gomatrixserverlib"
)
func LoadAddHistoryVisibilityColumn(m *sqlutil.Migrations) {
@ -51,6 +53,41 @@ func UpAddHistoryVisibilityColumn(tx *sql.Tx) error {
if err != nil {
return fmt.Errorf("failed to execute upgrade: %w", err)
}
// get the current room history visibility
rows, err := tx.Query(`SELECT DISTINCT room_id, headered_event_json FROM syncapi_current_room_state
WHERE type = 'm.room.history_visibility' AND state_key = '';
`)
if err != nil {
return fmt.Errorf("failed to query current room state: %w", err)
}
var eventBytes []byte
var roomID string
var event gomatrixserverlib.HeaderedEvent
var hisVis gomatrixserverlib.HistoryVisibility
historyVisibilities := make(map[string]gomatrixserverlib.HistoryVisibility)
for rows.Next() {
if err = rows.Scan(&roomID, &eventBytes); err != nil {
return fmt.Errorf("failed to scan row: %w", err)
}
if err = json.Unmarshal(eventBytes, &event); err != nil {
return fmt.Errorf("failed to unmarshal event: %w", err)
}
historyVisibilities[roomID] = gomatrixserverlib.HistoryVisibilityShared
if hisVis, err = event.HistoryVisibility(); err == nil {
historyVisibilities[roomID] = hisVis
}
}
// update the history visibility
for roomID, hisVis = range historyVisibilities {
_, err = tx.Exec(`UPDATE syncapi_output_room_events SET history_visibility = $1
WHERE type IN ('m.room.message', 'm.room.encrypted') AND room_id = $2`, hisVis, roomID)
if err != nil {
return fmt.Errorf("failed to update history visibility: %w", err)
}
}
return nil
}