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` = ?", userId, 1).Order("`ltime` 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,
			AddressName: address.AddressName,
			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,
		}

		// lastOne := &FsAddress{}
		// err = tx.Where("user_id = ?", lastOne.UserId).Order("ltime ASC").Take(&lastOne).Error
		// if err == gorm.ErrRecordNotFound {
		// 	result.Ltime = &now
		// 	return tx.Model(&FsAddress{}).Create(result).Error
		// }
		// if err != nil {
		// 	return err
		// }

		// // 根据lastOne处理时间

		// ltime := (*lastOne.Ltime).Add(-time.Second)
		// result.Ltime = &ltime
		return tx.Model(&FsAddress{}).Create(result).Error
	})

	if err != nil {
		return nil, err
	}

	return result, nil
}

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

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

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

		now := time.Now().UTC()

		err = tx.Model(&FsAddress{}).Where(" `user_id` = ? and `status` = ? and `address_id = ?` ", userId, 1, addressId).
			UpdateColumn("ltime", now.Add(time.Hour*24*365*100)).
			UpdateColumn("utime", now).Error
		if err != nil {
			return err
		}

		err = tx.Where(" `user_id` = ? and `status` = ? and `address_id != ? and `ltime` > ?` ", userId, 1, addressId, now.Add(time.Hour*24*365)).
			UpdateColumn("ltime", now).Error
		if err != nil {
			logx.Error(err)
		}

		return nil
	})

	return
}

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