mirror of
https://github.com/matrix-org/dendrite
synced 2024-12-14 23:32:45 +00:00
a6f995eb45
* Move Updater structs to shared and use it for postgres * Add constructors for NewXXXUpdater and a useTxns flag In sqlite, we set useTxns=false and comment why. * Handle nil txn * Handle nil in transaction * Missed one * Close the txn at the right time * Don't close the transaction as we reuse it between calls
183 lines
4.6 KiB
Go
183 lines
4.6 KiB
Go
package shared
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/matrix-org/dendrite/roomserver/storage/tables"
|
|
"github.com/matrix-org/dendrite/roomserver/types"
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
)
|
|
|
|
type membershipUpdater struct {
|
|
transaction
|
|
d *Database
|
|
roomNID types.RoomNID
|
|
targetUserNID types.EventStateKeyNID
|
|
membership tables.MembershipState
|
|
}
|
|
|
|
func NewMembershipUpdater(
|
|
ctx context.Context, d *Database, roomID, targetUserID string,
|
|
targetLocal bool, roomVersion gomatrixserverlib.RoomVersion,
|
|
useTxns bool,
|
|
) (types.MembershipUpdater, error) {
|
|
txn, err := d.DB.Begin()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
succeeded := false
|
|
defer func() {
|
|
if !succeeded {
|
|
txn.Rollback() // nolint: errcheck
|
|
}
|
|
}()
|
|
|
|
roomNID, err := d.assignRoomNID(ctx, txn, roomID, roomVersion)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
targetUserNID, err := d.assignStateKeyNID(ctx, txn, targetUserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
updater, err := d.membershipUpdaterTxn(ctx, txn, roomNID, targetUserNID, targetLocal)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
succeeded = true
|
|
if !useTxns {
|
|
txn.Commit() // nolint: errcheck
|
|
updater.transaction.txn = nil
|
|
}
|
|
return updater, nil
|
|
}
|
|
|
|
func (d *Database) membershipUpdaterTxn(
|
|
ctx context.Context,
|
|
txn *sql.Tx,
|
|
roomNID types.RoomNID,
|
|
targetUserNID types.EventStateKeyNID,
|
|
targetLocal bool,
|
|
) (*membershipUpdater, error) {
|
|
|
|
if err := d.MembershipTable.InsertMembership(ctx, txn, roomNID, targetUserNID, targetLocal); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
membership, err := d.MembershipTable.SelectMembershipForUpdate(ctx, txn, roomNID, targetUserNID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &membershipUpdater{
|
|
transaction{ctx, txn}, d, roomNID, targetUserNID, membership,
|
|
}, nil
|
|
}
|
|
|
|
// IsInvite implements types.MembershipUpdater
|
|
func (u *membershipUpdater) IsInvite() bool {
|
|
return u.membership == tables.MembershipStateInvite
|
|
}
|
|
|
|
// IsJoin implements types.MembershipUpdater
|
|
func (u *membershipUpdater) IsJoin() bool {
|
|
return u.membership == tables.MembershipStateJoin
|
|
}
|
|
|
|
// IsLeave implements types.MembershipUpdater
|
|
func (u *membershipUpdater) IsLeave() bool {
|
|
return u.membership == tables.MembershipStateLeaveOrBan
|
|
}
|
|
|
|
// SetToInvite implements types.MembershipUpdater
|
|
func (u *membershipUpdater) SetToInvite(event gomatrixserverlib.Event) (bool, error) {
|
|
senderUserNID, err := u.d.assignStateKeyNID(u.ctx, u.txn, event.Sender())
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
inserted, err := u.d.InvitesTable.InsertInviteEvent(
|
|
u.ctx, u.txn, event.EventID(), u.roomNID, u.targetUserNID, senderUserNID, event.JSON(),
|
|
)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if u.membership != tables.MembershipStateInvite {
|
|
if err = u.d.MembershipTable.UpdateMembership(
|
|
u.ctx, u.txn, u.roomNID, u.targetUserNID, senderUserNID, tables.MembershipStateInvite, 0,
|
|
); err != nil {
|
|
return false, err
|
|
}
|
|
}
|
|
return inserted, nil
|
|
}
|
|
|
|
// SetToJoin implements types.MembershipUpdater
|
|
func (u *membershipUpdater) SetToJoin(senderUserID string, eventID string, isUpdate bool) ([]string, error) {
|
|
var inviteEventIDs []string
|
|
|
|
senderUserNID, err := u.d.assignStateKeyNID(u.ctx, u.txn, senderUserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// If this is a join event update, there is no invite to update
|
|
if !isUpdate {
|
|
inviteEventIDs, err = u.d.InvitesTable.UpdateInviteRetired(
|
|
u.ctx, u.txn, u.roomNID, u.targetUserNID,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// Look up the NID of the new join event
|
|
nIDs, err := u.d.EventNIDs(u.ctx, []string{eventID})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if u.membership != tables.MembershipStateJoin || isUpdate {
|
|
if err = u.d.MembershipTable.UpdateMembership(
|
|
u.ctx, u.txn, u.roomNID, u.targetUserNID, senderUserNID,
|
|
tables.MembershipStateJoin, nIDs[eventID],
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return inviteEventIDs, nil
|
|
}
|
|
|
|
// SetToLeave implements types.MembershipUpdater
|
|
func (u *membershipUpdater) SetToLeave(senderUserID string, eventID string) ([]string, error) {
|
|
senderUserNID, err := u.d.assignStateKeyNID(u.ctx, u.txn, senderUserID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
inviteEventIDs, err := u.d.InvitesTable.UpdateInviteRetired(
|
|
u.ctx, u.txn, u.roomNID, u.targetUserNID,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Look up the NID of the new leave event
|
|
nIDs, err := u.d.EventNIDs(u.ctx, []string{eventID})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if u.membership != tables.MembershipStateLeaveOrBan {
|
|
if err = u.d.MembershipTable.UpdateMembership(
|
|
u.ctx, u.txn, u.roomNID, u.targetUserNID, senderUserNID,
|
|
tables.MembershipStateLeaveOrBan, nIDs[eventID],
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return inviteEventIDs, nil
|
|
}
|