fusenapi/model/gmodel/fs_address_logic.go

package gmodel

import (
	"context"
	"time"

	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
)

func (a *FsAddressModel) GetOne(ctx context.Context, addressId int64, userId int64) (resp *FsAddress, err error) {
	err = a.db.WithContext(ctx).Model(&FsAddress{}).Where("`address_id` = ? and `user_id` = ? and `status` = ? ", addressId, userId, 1).Take(&resp).Error
	return resp, err
}

func (a *FsAddressModel) GetUserAllAddress(ctx context.Context, userId int64) (resp []FsAddress, err error) {
	err = a.db.WithContext(ctx).Model(&FsAddress{}).Where("`user_id` = ? and `status` = 1", userId).Order("`ctime` DESC").Find(&resp).Error
	if err != nil {
		return nil, err
	}
	return
}

func (a *FsAddressModel) CreateOne(ctx context.Context, address *FsAddress) (result *FsAddress, err error) {

	err = a.db.WithContext(ctx).Model(&FsAddress{}).Transaction(func(tx *gorm.DB) error {
		now := time.Now().UTC()
		result = &FsAddress{
			UserId:    address.UserId,
			FirstName: address.FirstName,
			LastName:  address.LastName,
			Mobile:    address.Mobile,
			Street:    address.Street,
			Suite:     address.Suite,
			City:      address.City,
			State:     address.State,
			Country:   address.Country,
			ZipCode:   address.ZipCode,
			Status:    address.Status,
			IsDefault: address.IsDefault,
			Ctime:     &now,
			Utime:     &now,
			Ltime:     &now,
		}
		return tx.Model(&FsAddress{}).Create(result).Error
	})

	if err != nil {
		return nil, err
	}

	return result, nil
}

func (a *FsAddressModel) UpdateAddress(ctx context.Context, address *FsAddress) (err error) {
	err = a.db.WithContext(ctx).Model(&FsAddress{}).Transaction(func(tx *gorm.DB) error {
		err = tx.
			Where("address_id = ?  and user_id = ?  and status = 1 ", address.AddressId, address.UserId).
			Updates(address).Error
		if err != nil {
			return err
		}
		return err
	})
	return err
}

func (a *FsAddressModel) SettingUserDefaultAddressMutex(ctx context.Context, userId int64, addressId int64, isDefault int64) (err error) {

	err = a.db.WithContext(ctx).Model(&FsAddress{}).Transaction(func(tx *gorm.DB) error {

		logx.Info("address_id:", addressId, " set default ", isDefault)
		now := time.Now().UTC()

		err = tx.Where("`user_id` = ? and `status` = 1 and `is_default` = 1", userId).
			UpdateColumn("is_default", 0).
			UpdateColumn("utime", now).Error
		if err != nil {
			return err
		}

		err = tx.Where("`address_id` = ? and `user_id` = ? and `status` = 1", addressId, userId).
			UpdateColumn("is_default", isDefault).
			UpdateColumn("ltime", now).
			UpdateColumn("utime", now).Error
		if err != nil {
			return err
		}

		return nil
	})

	return err
}

func (a *FsAddressModel) SettingUserDefaultAddress(ctx context.Context, userId int64, addressId int64, isDefault int64) (err error) {

	err = a.db.WithContext(ctx).Model(&FsAddress{}).Transaction(func(tx *gorm.DB) error {

		logx.Info("address_id:", addressId, " set default ", isDefault)
		now := time.Now().UTC()
		err = tx.Where("`address_id` = ? and `user_id` = ? and `status` = 1", addressId, userId).
			UpdateColumn("is_default", isDefault).
			UpdateColumn("ltime", now).
			UpdateColumn("utime", now).Error
		if err != nil {
			return err
		}

		return nil
	})

	return err
}

func (a *FsAddressModel) DeleteOne(ctx context.Context, addressId int64, userId int64) (err error) {
	err = a.db.WithContext(ctx).Model(&FsAddress{}).
		Where("`address_id` = ? and `user_id` = ? and `status` = 1 ", addressId, userId).
		UpdateColumn("status", 0).Error
	return err
}

// UpdateUsedAddress 当每次订单成功后, 更新一次地址使用时间
func (a *FsAddressModel) UpdateUsedAddress(ctx context.Context, addressId int64, userId int64) (err error) {
	err = a.db.WithContext(ctx).Model(&FsAddress{}).Transaction(func(tx *gorm.DB) error {
		logx.Info("address_id:", addressId, " update used")
		now := time.Now().UTC()
		err = tx.Where("`address_id` = ? and `user_id` = ? and `status` = 1", addressId, userId).
			UpdateColumn("utime", now).
			UpdateColumn("ltime", now).Error
		if err != nil {
			return err
		}
		return nil
	})
	return err
}