mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-21 05:43:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

Use RETURNING because LastInsertID is not supported by postgres

Browse source
This commit is contained in:
Neil Alexander 2020-06-30 14:10:24 +01:00
parent 693928396d
commit a6aaa5aa52
2 changed files with 8 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
federationsender/storage/postgres/queue_json_table.go
View file

@ -39,7 +39,7 @@ CREATE TABLE IF NOT EXISTS federationsender_queue_json (
const insertJSONSQL = "" + const insertJSONSQL = "" +
"INSERT INTO federationsender_queue_json (json_body)" + "INSERT INTO federationsender_queue_json (json_body)" +
" VALUES ($1)" + " VALUES ($1)" +
" ON CONFLICT DO NOTHING" " RETURNING json_nid"
const deleteJSONSQL = "" + const deleteJSONSQL = "" +
"DELETE FROM federationsender_queue_json WHERE json_nid = ANY($1)" "DELETE FROM federationsender_queue_json WHERE json_nid = ANY($1)"
@ -75,12 +75,11 @@ func (s *queueJSONStatements) insertQueueJSON(
ctx context.Context, txn *sql.Tx, json string, ctx context.Context, txn *sql.Tx, json string,
) (int64, error) { ) (int64, error) {
stmt := sqlutil.TxStmt(txn, s.insertJSONStmt) stmt := sqlutil.TxStmt(txn, s.insertJSONStmt)
res, err := stmt.ExecContext(ctx, json) var lastid int64
if err != nil { if err := stmt.QueryRowContext(ctx, json).Scan(&lastid); err != nil {
return 0, err return 0, err
} }
lastid, err := res.LastInsertId() return lastid, nil
return lastid, err
} }
func (s *queueJSONStatements) deleteQueueJSON( func (s *queueJSONStatements) deleteQueueJSON(

9
federationsender/storage/sqlite3/queue_json_table.go
View file

@ -39,7 +39,7 @@ CREATE TABLE IF NOT EXISTS federationsender_queue_json (
const insertJSONSQL = "" + const insertJSONSQL = "" +
"INSERT INTO federationsender_queue_json (json_body)" + "INSERT INTO federationsender_queue_json (json_body)" +
" VALUES ($1)" + " VALUES ($1)" +
" ON CONFLICT DO NOTHING" " RETURNING json_nid"
const deleteJSONSQL = "" + const deleteJSONSQL = "" +
"DELETE FROM federationsender_queue_json WHERE json_nid = ANY($1)" "DELETE FROM federationsender_queue_json WHERE json_nid = ANY($1)"
@ -75,12 +75,11 @@ func (s *queueJSONStatements) insertQueueJSON(
ctx context.Context, txn *sql.Tx, json string, ctx context.Context, txn *sql.Tx, json string,
) (int64, error) { ) (int64, error) {
stmt := sqlutil.TxStmt(txn, s.insertJSONStmt) stmt := sqlutil.TxStmt(txn, s.insertJSONStmt)
res, err := stmt.ExecContext(ctx, json) var lastid int64
if err != nil { if err := stmt.QueryRowContext(ctx, json).Scan(&lastid); err != nil {
return 0, err return 0, err
} }
lastid, err := res.LastInsertId() return lastid, nil
return lastid, err
} }
func (s *queueJSONStatements) deleteQueueJSON( func (s *queueJSONStatements) deleteQueueJSON(