ScuroNeko/YaeMikoBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

some fixes

Browse Source
This commit is contained in:
ScuroNeko
2026-02-12 14:40:53 +03:00
parent acbd39cf46
commit c5448b14f6
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
database/psql/rp.go
View File

@@ -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) {