Fix further errors pointed by linter

This commit is contained in:
Piotr Kozimor 2021-09-14 10:43:01 +02:00
parent 5f8601d18c
commit 1fbed4194a
5 changed files with 12 additions and 78 deletions

View file

@ -15,7 +15,6 @@ import (
) )
const ( const (
sessionIdByteLength = 32
tokenByteLength = 48 tokenByteLength = 48
) )
@ -25,9 +24,10 @@ func (a *UserInternalAPI) CreateSession(ctx context.Context, req *api.CreateSess
s, err := a.ThreePidDB.GetSessionByThreePidAndSecret(ctx, req.ThreePid, req.ClientSecret) s, err := a.ThreePidDB.GetSessionByThreePidAndSecret(ctx, req.ThreePid, req.ClientSecret)
if err != nil { if err != nil {
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
token, errB := internal.GenerateBlob(tokenByteLength) var token string
if errB != nil { token, err = internal.GenerateBlob(tokenByteLength)
return errB if err != nil {
return err
} }
s = &api.Session{ s = &api.Session{
ClientSecret: req.ClientSecret, ClientSecret: req.ClientSecret,

View file

@ -40,11 +40,11 @@ func Open(dbProperties *config.DatabaseOptions) (*Database, error) {
} }
// Create the tables. // Create the tables.
if err := create(db); err != nil { if err = create(db); err != nil {
return nil, err return nil, err
} }
var sharedDb *shared.Database
sharedDb, err := shared.NewDatabase(db, sqlutil.NewDummyWriter()) sharedDb, err = shared.NewDatabase(db, sqlutil.NewDummyWriter())
if err != nil { if err != nil {
return nil, err return nil, err
} }

View file

@ -18,12 +18,7 @@ func (d *Database) InsertSession(
var sid int64 var sid int64
return sid, d.writer.Do(nil, nil, func(_ *sql.Tx) error { return sid, d.writer.Do(nil, nil, func(_ *sql.Tx) error {
err := d.stm.insertSessionStmt.QueryRowContext(ctx, clientSecret, threepid, token, nextlink, validatedAt, validated, sendAttempt).Scan(&sid) err := d.stm.insertSessionStmt.QueryRowContext(ctx, clientSecret, threepid, token, nextlink, validatedAt, validated, sendAttempt).Scan(&sid)
// _, err := d.stm.insertSessionStmt.ExecContext(ctx, clientSecret, threepid, token, nextlink, sendAttempt, validatedAt, validated)
return err return err
// if err != nil {
// return err
// }
// err = d.stm.selectSidStmt.QueryRowContext(ctx).Scan(&sid)
}) })
} }
@ -76,61 +71,3 @@ func NewDatabase(db *sql.DB, writer sqlutil.Writer) (*Database, error) {
} }
return &d, nil return &d, nil
} }
// func newSQLiteDatabase(dbProperties *config.DatabaseOptions) (*Db, error) {
// db, err := sqlutil.Open(dbProperties)
// if err != nil {
// return nil, err
// }
// writer := sqlutil.NewExclusiveWriter()
// stmt := sessionStatements{
// db: db,
// writer: writer,
// }
// // Create tables before executing migrations so we don't fail if the table is missing,
// // and THEN prepare statements so we don't fail due to referencing new columns
// if err = stmt.execSchema(db); err != nil {
// return nil, err
// }
// if err = stmt.prepare(); err != nil {
// return nil, err
// }
// handler := func(f func(tx *sql.Tx) error) error {
// return writer.Do(nil, nil, f)
// }
// return &Db{db, writer, &stmt, handler}, nil
// }
// func newPostgresDatabase(dbProperties *config.DatabaseOptions) (*Db, error) {
// db, err := sqlutil.Open(dbProperties)
// if err != nil {
// return nil, err
// }
// stmt := sessionStatements{
// db: db,
// }
// // Create tables before executing migrations so we don't fail if the table is missing,
// // and THEN prepare statements so we don't fail due to referencing new columns
// if err = stmt.execSchema(db); err != nil {
// return nil, err
// }
// if err = stmt.prepare(); err != nil {
// return nil, err
// }
// handler := func(f func(tx *sql.Tx) error) error {
// return f(nil)
// }
// return &Db{db, nil, &stmt, handler}, nil
// }
// func NewDatabase(dbProperties *config.DatabaseOptions) (Database, error) {
// switch {
// case dbProperties.ConnectionString.IsSQLite():
// return newSQLiteDatabase(dbProperties)
// case dbProperties.ConnectionString.IsPostgres():
// return newPostgresDatabase(dbProperties)
// default:
// return nil, fmt.Errorf("unexpected database type")
// }
// }

View file

@ -11,8 +11,6 @@ const (
"INSERT INTO threepid_sessions (client_secret, threepid, token, next_link, validated_at_ts, validated, send_attempt)" + "INSERT INTO threepid_sessions (client_secret, threepid, token, next_link, validated_at_ts, validated, send_attempt)" +
"VALUES ($1, $2, $3, $4, $5, $6, $7)" + "VALUES ($1, $2, $3, $4, $5, $6, $7)" +
"RETURNING sid;" "RETURNING sid;"
// selectSidSQL = "" +
// "SELECT last_insert_rowid();"
selectSessionSQL = "" + selectSessionSQL = "" +
"SELECT client_secret, threepid, token, next_link, validated_at_ts, validated, send_attempt FROM threepid_sessions WHERE sid = $1" "SELECT client_secret, threepid, token, next_link, validated_at_ts, validated, send_attempt FROM threepid_sessions WHERE sid = $1"
selectSessionByThreePidAndCLientSecretSQL = "" + selectSessionByThreePidAndCLientSecretSQL = "" +
@ -27,7 +25,6 @@ const (
type ThreePidSessionStatements struct { type ThreePidSessionStatements struct {
insertSessionStmt *sql.Stmt insertSessionStmt *sql.Stmt
// selectSidStmt *sql.Stmt
selectSessionStmt *sql.Stmt selectSessionStmt *sql.Stmt
selectSessionByThreePidAndCLientSecretStmt *sql.Stmt selectSessionByThreePidAndCLientSecretStmt *sql.Stmt
deleteSessionStmt *sql.Stmt deleteSessionStmt *sql.Stmt
@ -39,7 +36,6 @@ func PrepareThreePidSessionsTable(db *sql.DB) (*ThreePidSessionStatements, error
s := ThreePidSessionStatements{} s := ThreePidSessionStatements{}
return &s, sqlutil.StatementList{ return &s, sqlutil.StatementList{
{&s.insertSessionStmt, insertSessionSQL}, {&s.insertSessionStmt, insertSessionSQL},
// {&s.selectSidStmt, selectSidSQL},
{&s.selectSessionStmt, selectSessionSQL}, {&s.selectSessionStmt, selectSessionSQL},
{&s.selectSessionByThreePidAndCLientSecretStmt, selectSessionByThreePidAndCLientSecretSQL}, {&s.selectSessionByThreePidAndCLientSecretStmt, selectSessionByThreePidAndCLientSecretSQL},
{&s.deleteSessionStmt, deleteSessionSQL}, {&s.deleteSessionStmt, deleteSessionSQL},

View file

@ -40,11 +40,12 @@ func Open(dbProperties *config.DatabaseOptions) (*Database, error) {
} }
// Create the tables. // Create the tables.
if err := create(db); err != nil { if err = create(db); err != nil {
return nil, err return nil, err
} }
sharedDb, err := shared.NewDatabase(db, sqlutil.NewExclusiveWriter()) var sharedDb *shared.Database
sharedDb, err = shared.NewDatabase(db, sqlutil.NewExclusiveWriter())
if err != nil { if err != nil {
return nil, err return nil, err
} }