/
market_order.go
71 lines (66 loc) · 2.4 KB
/
market_order.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package contract
import (
"context"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/fibonacci-chain/core/x/dex/keeper"
dexkeeperabci "github.com/fibonacci-chain/core/x/dex/keeper/abci"
"github.com/fibonacci-chain/core/x/dex/types"
dexutils "github.com/fibonacci-chain/core/x/dex/utils"
"go.opentelemetry.io/otel/attribute"
otrace "go.opentelemetry.io/otel/trace"
)
func PrepareCancelUnfulfilledMarketOrders(
ctx sdk.Context,
typedContractAddr types.ContractAddress,
typedPairStr types.PairString,
orderIDToSettledQuantities map[uint64]sdk.Dec,
) {
dexutils.GetMemState(ctx.Context()).ClearCancellationForPair(ctx, typedContractAddr, typedPairStr)
for _, marketOrderID := range getUnfulfilledPlacedMarketOrderIds(ctx, typedContractAddr, typedPairStr, orderIDToSettledQuantities) {
dexutils.GetMemState(ctx.Context()).GetBlockCancels(ctx, typedContractAddr, typedPairStr).Add(&types.Cancellation{
Id: marketOrderID,
Initiator: types.CancellationInitiator_USER,
})
}
}
func getUnfulfilledPlacedMarketOrderIds(
ctx sdk.Context,
typedContractAddr types.ContractAddress,
typedPairStr types.PairString,
orderIDToSettledQuantities map[uint64]sdk.Dec,
) []uint64 {
res := []uint64{}
for _, order := range dexutils.GetMemState(ctx.Context()).GetBlockOrders(ctx, typedContractAddr, typedPairStr).Get() {
if order.Status == types.OrderStatus_FAILED_TO_PLACE {
continue
}
if order.OrderType == types.OrderType_MARKET || order.OrderType == types.OrderType_FOKMARKET {
if settledQuantity, ok := orderIDToSettledQuantities[order.Id]; !ok || settledQuantity.LT(order.Quantity) {
res = append(res, order.Id)
}
} else if order.OrderType == types.OrderType_FOKMARKETBYVALUE {
// cancel market order by nominal if zero quantity is executed
if _, ok := orderIDToSettledQuantities[order.Id]; !ok {
res = append(res, order.Id)
}
}
}
return res
}
func CancelUnfulfilledMarketOrders(
ctx context.Context,
sdkCtx sdk.Context,
contractAddr string,
dexkeeper *keeper.Keeper,
registeredPairs []types.Pair,
tracer *otrace.Tracer,
) error {
spanCtx, span := (*tracer).Start(ctx, "CancelUnfulfilledMarketOrders")
span.SetAttributes(attribute.String("contract", contractAddr))
defer span.End()
abciWrapper := dexkeeperabci.KeeperWrapper{Keeper: dexkeeper}
if err := abciWrapper.HandleEBCancelOrders(spanCtx, sdkCtx, tracer, contractAddr, registeredPairs); err != nil {
return err
}
return nil
}