all database switched to repository model. some other fixes and features
This commit is contained in:
@@ -25,7 +25,8 @@ type RPUser struct {
|
||||
UserID int64 `db:"user_id"`
|
||||
UserPrompt string `db:"user_prompt"`
|
||||
SelectedPreset string `db:"selected_preset"`
|
||||
UsedTokens int64 `db:"used_tokens"`
|
||||
Preset *RPGeneralPreset
|
||||
UsedTokens int64 `db:"used_tokens"`
|
||||
}
|
||||
|
||||
type RPRepository struct {
|
||||
@@ -51,6 +52,10 @@ func (rep *RPRepository) CreateUser(id int64) (*RPUser, error) {
|
||||
func (rep *RPRepository) GetUser(id int64) (*RPUser, error) {
|
||||
user := new(RPUser)
|
||||
err := rep.db.Get(user, "SELECT * FROM rp_users WHERE user_id=$1", id)
|
||||
if err != nil {
|
||||
return user, err
|
||||
}
|
||||
user.Preset, err = rep.GetPreset(user.SelectedPreset)
|
||||
return user, err
|
||||
}
|
||||
func (rep *RPRepository) UpdateUser(user *RPUser) error {
|
||||
|
||||
Reference in New Issue
Block a user