batch.go

package keeper

import (
	sdk "github.com/cosmos/cosmos-sdk/types"

	"github.com/cosmosquad-labs/squad/v3/x/liquidity/types"
)

// ExecuteRequests executes all orders, deposit requests and withdraw requests.
// ExecuteRequests also handles order expiration.
func (k Keeper) ExecuteRequests(ctx sdk.Context) {
	if err := k.IterateAllPairs(ctx, func(pair types.Pair) (stop bool, err error) {
		if err := k.ExecuteMatching(ctx, pair); err != nil {
			return false, err
		}
		return false, nil
	}); err != nil {
		panic(err)
	}
	if err := k.IterateAllOrders(ctx, func(order types.Order) (stop bool, err error) {
		if order.Status.CanBeExpired() && order.ExpiredAt(ctx.BlockTime()) {
			if err := k.FinishOrder(ctx, order, types.OrderStatusExpired); err != nil {
				return false, err
			}
		} else if types.IsTooSmallOrderAmount(order.OpenAmount, order.Price) {
			// TODO: should we introduce new order status for this type of expiration?
			if err := k.FinishOrder(ctx, order, types.OrderStatusExpired); err != nil {
				return false, err
			}
		}
		return false, nil
	}); err != nil {
		panic(err)
	}
	if err := k.IterateAllDepositRequests(ctx, func(req types.DepositRequest) (stop bool, err error) {
		if req.Status == types.RequestStatusNotExecuted {
			if err := k.ExecuteDepositRequest(ctx, req); err != nil {
				return false, err
			}
		}
		return false, nil
	}); err != nil {
		panic(err)
	}
	if err := k.IterateAllWithdrawRequests(ctx, func(req types.WithdrawRequest) (stop bool, err error) {
		if req.Status == types.RequestStatusNotExecuted {
			if err := k.ExecuteWithdrawRequest(ctx, req); err != nil {
				return false, err
			}
		}
		return false, nil
	}); err != nil {
		panic(err)
	}
}

// DeleteOutdatedRequests deletes outdated(should be deleted) requests.
// Determining if a request should be deleted is based on its status.
func (k Keeper) DeleteOutdatedRequests(ctx sdk.Context) {
	_ = k.IterateAllDepositRequests(ctx, func(req types.DepositRequest) (stop bool, err error) {
		if req.Status.ShouldBeDeleted() {
			k.DeleteDepositRequest(ctx, req)
		}
		return false, nil
	})
	_ = k.IterateAllWithdrawRequests(ctx, func(req types.WithdrawRequest) (stop bool, err error) {
		if req.Status.ShouldBeDeleted() {
			k.DeleteWithdrawRequest(ctx, req)
		}
		return false, nil
	})
	_ = k.IterateAllOrders(ctx, func(order types.Order) (stop bool, err error) {
		if order.Status.ShouldBeDeleted() {
			k.DeleteOrder(ctx, order)
		}
		return false, nil
	})
}