ErfanTech
/
niki
forked from ebhomengo/niki
1
0
Fork 0
Code Issues Pull Requests 1 Actions Activity

feat(repository): configured is_main field for CreateBenefactorAddress

This commit is contained in:
Fatemeh Javadi 2024-05-21 23:16:06 +03:30
parent ba4ebbcb84
commit 53017a31a5
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
repository/mysql/address/create.go
View File

@ -2,6 +2,7 @@ package mysqladdress
import ( import (
"context" "context"
"database/sql"
"git.gocasts.ir/ebhomengo/niki/entity" "git.gocasts.ir/ebhomengo/niki/entity"
errmsg "git.gocasts.ir/ebhomengo/niki/pkg/err_msg" errmsg "git.gocasts.ir/ebhomengo/niki/pkg/err_msg"
@ -11,16 +12,31 @@ import (
func (d *DB) CreateBenefactorAddress(ctx context.Context, address entity.Address) (entity.Address, error) { func (d *DB) CreateBenefactorAddress(ctx context.Context, address entity.Address) (entity.Address, error) {
const op = "mysqlbenefactor.createBenefactorAddress" const op = "mysqlbenefactor.createBenefactorAddress"
res, err := d.conn.Conn().ExecContext(ctx, `insert into addresses(postal_code, address, lat, lon,province_id,city_id,benefactor_id) values(?, ?, ?,?,?,?,?)`, // Check if the user already has a main address
address.PostalCode, address.Address, address.Lat, address.Lon, address.ProvinceID, address.CityID, address.BenefactorID) var count int
err := d.conn.Conn().QueryRowContext(ctx, `SELECT COUNT(*) FROM addresses WHERE is_main = true AND benefactor_id = ?`, address.BenefactorID).Scan(&count)
if err != nil && err != sql.ErrNoRows {
return entity.Address{}, richerror.New(op).WithErr(err).
WithMessage("error querying for existing main address").WithKind(richerror.KindUnexpected)
}
// Set is_main field based on the count
address.IsMain = count == 0
// Insert the new address
res, err := d.conn.Conn().ExecContext(ctx, `INSERT INTO addresses (postal_code, address, lat, lon, is_main, city_id, province_id, benefactor_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
address.PostalCode, address.Address, address.Lat, address.Lon, address.IsMain, address.CityID, address.ProvinceID, address.BenefactorID)
if err != nil { if err != nil {
return entity.Address{}, richerror.New(op).WithErr(err). return entity.Address{}, richerror.New(op).WithErr(err).
WithMessage(errmsg.ErrorMsgNotFound).WithKind(richerror.KindUnexpected) WithMessage(errmsg.ErrorMsgNotFound).WithKind(richerror.KindUnexpected)
} }
//nolint // Get the ID of the newly inserted record
// error is always nil id, err := res.LastInsertId()
id, _ := res.LastInsertId() if err != nil {
return entity.Address{}, richerror.New(op).WithErr(err).
WithMessage("error retrieving last insert id").WithKind(richerror.KindUnexpected)
}
address.ID = uint(id) address.ID = uint(id)
return address, nil return address, nil