diff --git a/database/psql/rp.go b/database/psql/rp.go index 66c4864..49feb55 100644 --- a/database/psql/rp.go +++ b/database/psql/rp.go @@ -85,10 +85,15 @@ func (rep *RPRepository) GetUser(id int64) (*RPUser, error) { return user, err } func (rep *RPRepository) UpdateUser(user *RPUser) error { - _, err := rep.db.NamedExec( - "UPDATE rp_users SET selected_preset=:selected_preset, used_tokens=:used_tokens, user_prompt=:user_prompt, selected_model=:selected_model WHERE user_id=:user_id;", - user, + s, args, err := sqlx.In( + "UPDATE rp_users SET selected_preset=?, used_tokens=?, user_prompt=?, selected_model=? WHERE user_id=?;", + user.SelectedPreset, user.UsedTokens, user.UserPrompt, user.SelectedModel, user.UserID, ) + if err != nil { + return err + } + s = rep.db.Rebind(s) + _, err = rep.db.Exec(s, args...) return err } func (rep *RPRepository) UpdateUserPreset(user *RPUser, presetId string) (*RPPreset, error) { @@ -96,7 +101,15 @@ func (rep *RPRepository) UpdateUserPreset(user *RPUser, presetId string) (*RPPre if err != nil { return preset, err } - _, err = rep.db.Exec("UPDATE rp_users SET selected_preset=$1 WHERE user_id=$2;", presetId, user.UserID) + s, args, err := sqlx.In( + "UPDATE rp_users SET selected_preset=? WHERE user_id=?;", + presetId, user.UserID, + ) + if err != nil { + return preset, err + } + s = rep.db.Rebind(s) + _, err = rep.db.Exec(s, args...) return preset, err } func (rep *RPRepository) GetUserPreset(user *RPUser) (*RPPreset, error) {