mirror of
https://github.com/element-hq/dendrite.git
synced 2025-09-14 13:22:26 +03:00
drop primary key constraint from userapi_devices.access_token
This commit is contained in:
parent
ff63e7fa98
commit
bf310d558f
5 changed files with 111 additions and 12 deletions
|
@ -283,8 +283,6 @@ func (m *MSC3861UserVerifier) getUserByAccessToken(ctx context.Context, token st
|
||||||
Msg: strings.Join([]string{"Invalid device ID in scope: ", deviceID}, ""),
|
Msg: strings.Join([]string{"Invalid device ID in scope: ", deviceID}, ""),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
logger.Debugf("deviceID is: %s", deviceID)
|
|
||||||
logger.Debugf("scope is: %+v", scopes)
|
|
||||||
|
|
||||||
userDeviceExists := false
|
userDeviceExists := false
|
||||||
{
|
{
|
||||||
|
@ -302,14 +300,13 @@ func (m *MSC3861UserVerifier) getUserByAccessToken(ctx context.Context, token st
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
logger.Debugf("userDeviceExists is: %t", userDeviceExists)
|
|
||||||
if !userDeviceExists {
|
if !userDeviceExists {
|
||||||
var rs api.PerformDeviceCreationResponse
|
var rs api.PerformDeviceCreationResponse
|
||||||
deviceDisplayName := "OIDC-native client"
|
deviceDisplayName := "OIDC-native client"
|
||||||
if err := m.userAPI.PerformDeviceCreation(ctx, &api.PerformDeviceCreationRequest{
|
if err := m.userAPI.PerformDeviceCreation(ctx, &api.PerformDeviceCreationRequest{
|
||||||
Localpart: localpart,
|
Localpart: localpart,
|
||||||
ServerName: m.serverName,
|
ServerName: m.serverName,
|
||||||
AccessToken: token,
|
AccessToken: "",
|
||||||
DeviceID: &deviceID,
|
DeviceID: &deviceID,
|
||||||
DeviceDisplayName: &deviceDisplayName,
|
DeviceDisplayName: &deviceDisplayName,
|
||||||
// TODO: Cannot add IPAddr and Useragent values here. Should we care about it here?
|
// TODO: Cannot add IPAddr and Useragent values here. Should we care about it here?
|
||||||
|
|
|
@ -0,0 +1,25 @@
|
||||||
|
package deltas
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func UpDropPrimaryKeyConstraint(ctx context.Context, tx *sql.Tx) error {
|
||||||
|
_, err := tx.ExecContext(ctx, `
|
||||||
|
ALTER TABLE userapi_devices DROP CONSTRAINT userapi_devices_pkey;`)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute upgrade: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DownDropPrimaryKeyConstraint(ctx context.Context, tx *sql.Tx) error {
|
||||||
|
_, err := tx.ExecContext(ctx, `
|
||||||
|
ALTER TABLE userapi_devices ADD CONSTRAINT userapi_devices_pkey PRIMARY KEY (access_token);`)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute downgrade: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -116,10 +116,16 @@ func NewPostgresDevicesTable(db *sql.DB, serverName spec.ServerName) (tables.Dev
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
m := sqlutil.NewMigrator(db)
|
m := sqlutil.NewMigrator(db)
|
||||||
m.AddMigrations(sqlutil.Migration{
|
m.AddMigrations(
|
||||||
Version: "userapi: add last_seen_ts",
|
sqlutil.Migration{
|
||||||
Up: deltas.UpLastSeenTSIP,
|
Version: "userapi: add last_seen_ts",
|
||||||
})
|
Up: deltas.UpLastSeenTSIP,
|
||||||
|
},
|
||||||
|
sqlutil.Migration{
|
||||||
|
Version: "userapi: drop primary key constraint",
|
||||||
|
Up: deltas.UpDropPrimaryKeyConstraint,
|
||||||
|
},
|
||||||
|
)
|
||||||
err = m.Up(context.Background())
|
err = m.Up(context.Background())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
@ -0,0 +1,65 @@
|
||||||
|
package deltas
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
func UpDropPrimaryKeyConstraint(ctx context.Context, tx *sql.Tx) error {
|
||||||
|
_, err := tx.ExecContext(ctx, `
|
||||||
|
ALTER TABLE userapi_devices RENAME TO userapi_devices_tmp;
|
||||||
|
CREATE TABLE userapi_devices (
|
||||||
|
access_token TEXT,
|
||||||
|
session_id INTEGER,
|
||||||
|
device_id TEXT ,
|
||||||
|
localpart TEXT ,
|
||||||
|
server_name TEXT NOT NULL,
|
||||||
|
created_ts BIGINT,
|
||||||
|
display_name TEXT,
|
||||||
|
last_seen_ts BIGINT,
|
||||||
|
ip TEXT,
|
||||||
|
user_agent TEXT,
|
||||||
|
UNIQUE (localpart, device_id)
|
||||||
|
);
|
||||||
|
INSERT
|
||||||
|
INTO userapi_devices (
|
||||||
|
access_token, session_id, device_id, localpart, created_ts, display_name, last_seen_ts, ip, user_agent
|
||||||
|
) SELECT
|
||||||
|
access_token, session_id, device_id, localpart, created_ts, display_name, created_ts, '', ''
|
||||||
|
FROM userapi_devices_tmp;
|
||||||
|
DROP TABLE userapi_devices_tmp;`)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute upgrade: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DownDropPrimaryKeyConstraint(ctx context.Context, tx *sql.Tx) error {
|
||||||
|
_, err := tx.ExecContext(ctx, `
|
||||||
|
ALTER TABLE userapi_devices RENAME TO userapi_devices_tmp;
|
||||||
|
CREATE TABLE userapi_devices (
|
||||||
|
access_token TEXT PRIMARY KEY,
|
||||||
|
session_id INTEGER,
|
||||||
|
device_id TEXT ,
|
||||||
|
localpart TEXT ,
|
||||||
|
server_name TEXT NOT NULL,
|
||||||
|
created_ts BIGINT,
|
||||||
|
display_name TEXT,
|
||||||
|
last_seen_ts BIGINT,
|
||||||
|
ip TEXT,
|
||||||
|
user_agent TEXT,
|
||||||
|
UNIQUE (localpart, device_id)
|
||||||
|
);
|
||||||
|
INSERT
|
||||||
|
INTO userapi_devices (
|
||||||
|
access_token, session_id, device_id, localpart, created_ts, display_name, last_seen_ts, ip, user_agent
|
||||||
|
) SELECT
|
||||||
|
access_token, session_id, device_id, localpart, created_ts, display_name, created_ts, '', ''
|
||||||
|
FROM userapi_devices_tmp;
|
||||||
|
DROP TABLE userapi_devices_tmp;`)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to execute downgrade: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -102,10 +102,16 @@ func NewSQLiteDevicesTable(db *sql.DB, serverName spec.ServerName) (tables.Devic
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
m := sqlutil.NewMigrator(db)
|
m := sqlutil.NewMigrator(db)
|
||||||
m.AddMigrations(sqlutil.Migration{
|
m.AddMigrations(
|
||||||
Version: "userapi: add last_seen_ts",
|
sqlutil.Migration{
|
||||||
Up: deltas.UpLastSeenTSIP,
|
Version: "userapi: add last_seen_ts",
|
||||||
})
|
Up: deltas.UpLastSeenTSIP,
|
||||||
|
},
|
||||||
|
sqlutil.Migration{
|
||||||
|
Version: "userapi: drop primary key constraint",
|
||||||
|
Up: deltas.UpDropPrimaryKeyConstraint,
|
||||||
|
},
|
||||||
|
)
|
||||||
if err = m.Up(context.Background()); err != nil {
|
if err = m.Up(context.Background()); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue