Partially implement updates of models

This commit is contained in:
eikendev 2020-08-02 02:46:38 +02:00
parent 6a77df8373
commit d621333b6e
No known key found for this signature in database
GPG key ID: A1BDB1B28C8EF694
9 changed files with 158 additions and 19 deletions

View file

@ -3,6 +3,7 @@ package database
import (
"errors"
"github.com/eikendev/pushbits/assert"
"github.com/eikendev/pushbits/model"
"gorm.io/gorm"
@ -24,6 +25,11 @@ func (d *Database) DeleteUser(user *model.User) error {
return d.gormdb.Delete(user).Error
}
// UpdateUser updates a user.
func (d *Database) UpdateUser(user *model.User) error {
return d.gormdb.Save(user).Error
}
// GetUserByID returns the user with the given ID or nil.
func (d *Database) GetUserByID(ID uint) (*model.User, error) {
var user model.User
@ -34,6 +40,8 @@ func (d *Database) GetUserByID(ID uint) (*model.User, error) {
return nil, err
}
assert.Assert(user.ID == ID)
return &user, err
}
@ -47,6 +55,8 @@ func (d *Database) GetUserByName(name string) (*model.User, error) {
return nil, err
}
assert.Assert(user.Name == name)
return &user, err
}