fix method GetAccAndPrivilegeByEmail

This commit is contained in:
Pasha 2025-05-19 12:34:42 +03:00 committed by skeris
parent e1164644b1
commit 2dd6d69acf

@ -322,15 +322,20 @@ func (r *AccountRepository) GetAccAndPrivilegeByEmail(ctx context.Context, email
var account model.Account var account model.Account
var privileges []model.ShortPrivilege var privileges []model.ShortPrivilege
row, err := r.queries.GetAccAndPrivilegeByEmail(ctx, email) rows, err := r.queries.GetAccAndPrivilegeByEmail(ctx, email)
if err != nil { if err != nil {
return account, privileges, err return account, privileges, err
} }
account.ID = row.ID.String() if len(rows) == 0 {
account.UserID = row.UserID return account, privileges, sql.ErrNoRows
account.CreatedAt = row.CreatedAt }
account.ID = rows[0].ID.String()
account.UserID = rows[0].UserID
account.CreatedAt = rows[0].CreatedAt
for _, row := range rows {
if row.ID_2 != 0 { if row.ID_2 != 0 {
privilege := model.ShortPrivilege{ privilege := model.ShortPrivilege{
ID: fmt.Sprint(row.ID_2), ID: fmt.Sprint(row.ID_2),
@ -340,6 +345,7 @@ func (r *AccountRepository) GetAccAndPrivilegeByEmail(ctx context.Context, email
} }
privileges = append(privileges, privilege) privileges = append(privileges, privilege)
} }
}
return account, privileges, nil return account, privileges, nil
} }