PenaSide/customer
3
0
Fork 0
generated from PenaSide/GolangTemplate
Code Issues 1 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Update 2 files

Browse Source 
- /internal/models/account.go
- /internal/interface/swagger/api.2.go
This commit is contained in:
Mikhail 2024-04-12 23:31:56 +00:00
parent be04e78880
commit e9fe718226
2 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
internal/interface/swagger/api.2.go
View File

@ -181,11 +181,15 @@ func (api *API2) AddAccount(ctx echo.Context) error {
return api.noauth(ctx) return api.noauth(ctx)
} }
var quizFrom string var quizFrom, quizUser string
cookie, er := ctx.Request().Cookie("quizFrom") cookie, er := ctx.Request().Cookie("quizFrom")
if er == nil { if er == nil {
quizFrom = cookie.Value quizFrom = cookie.Value
} }
cookie, er = ctx.Request().Cookie("quizUser")
if er == nil {
quizUser = cookie.Value
}
account, err := api.account.FindByUserID(ctx.Request().Context(), userID) account, err := api.account.FindByUserID(ctx.Request().Context(), userID)
if err != nil && err.Type() != errors.ErrNotFound { if err != nil && err.Type() != errors.ErrNotFound {
@ -201,7 +205,8 @@ func (api *API2) AddAccount(ctx echo.Context) error {
return api.errorOld(ctx, err) return api.errorOld(ctx, err)
} }
account, err = api.account.Insert(ctx.Request().Context(), &models.Account{UserID: user.ID, Wallet: models.Wallet{Currency: defaultCurrency}, From: quizFrom}) account, err = api.account.Insert(ctx.Request().Context(), &models.Account{
UserID: user.ID, Wallet: models.Wallet{Currency: defaultCurrency}, From: quizFrom, Partner: quizUser})
if err != nil { if err != nil {
return api.errorOld(ctx, err) return api.errorOld(ctx, err)
} }

3
internal/models/account.go
View File

@ -14,6 +14,7 @@ type Account struct {
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt"` UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt"`
DeletedAt *time.Time `json:"deletedAt,omitempty" bson:"deletedAt,omitempty"` DeletedAt *time.Time `json:"deletedAt,omitempty" bson:"deletedAt,omitempty"`
From string `json:"from" bson:"from"` From string `json:"from" bson:"from"`
Partner string `json:"partner" bson:"partner"`
} }
func (receiver *Account) Sanitize() *Account { func (receiver *Account) Sanitize() *Account {
@ -26,6 +27,8 @@ func (receiver *Account) Sanitize() *Account {
receiver.Status = DefaultAccountStatus receiver.Status = DefaultAccountStatus
receiver.CreatedAt = now receiver.CreatedAt = now
receiver.UpdatedAt = now receiver.UpdatedAt = now
receiver.From = ""
receiver.Partner = ""
receiver.DeletedAt = nil receiver.DeletedAt = nil
receiver.Deleted = false receiver.Deleted = false