Множество улучшенийv0.0.2

1 files changed, 8 insertions, 1 deletions

diff --git a/repository/user.go b/repository/user.go
index ec3b702..5c3cce6 100644
--- a/repository/user.go
+++ b/repository/user.go
@@ -19,7 +19,7 @@ func NewUser(db *bun.DB) *User {
 	}
 }
 
-func (u *User) Create(ctx context.Context, email, password, username string) (int, error) {
+func (u *User) Create(ctx context.Context, email, password, username string, role models.UserRole) (int, error) {
 
 	hpassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
 	if err != nil {
@@ -30,6 +30,7 @@ func (u *User) Create(ctx context.Context, email, password, username string) (in
 		Email:    email,
 		Password: string(hpassword),
 		Username: username,
+		Role:     role,
 	}
 
 	if _, err := u.db.NewInsert().Model(user).Returning("id").Exec(ctx); err != nil {
@@ -52,3 +53,9 @@ func (u *User) Login(ctx context.Context, email, password string) (*models.User,
 
 	return user, nil
 }
+
+func (u *User) List(ctx context.Context) ([]*models.User, error) {
+	ret := make([]*models.User, 0)
+
+	return ret, u.db.NewSelect().Model(&ret).Scan(ctx)
+}