Skip to content

Commit 641bd56

Browse files
authored
refactor: use slices.Contains to simplify code (#1254)
Signed-off-by: CoolCu <coolcui@qq.com>
1 parent 41553e5 commit 641bd56

File tree

6 files changed

+17
-37
lines changed

6 files changed

+17
-37
lines changed

x/amm/keeper/params.go

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"github.com/elys-network/elys/x/amm/types"
88
assetprofiletypes "github.com/elys-network/elys/x/assetprofile/types"
99
ptypes "github.com/elys-network/elys/x/parameter/types"
10+
"slices"
1011
)
1112

1213
func (k Keeper) GetParams(ctx sdk.Context) (params types.Params) {
@@ -42,13 +43,7 @@ func (k Keeper) CheckBaseAssetExist(ctx sdk.Context, denom string) bool {
4243
}
4344
}
4445

45-
found := false
46-
for _, asset := range params.BaseAssets {
47-
if asset == denom {
48-
found = true
49-
break
50-
}
51-
}
46+
found := slices.Contains(params.BaseAssets, denom)
5247
return found
5348
}
5449

x/leveragelp/keeper/msg_server_open.go

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"errors"
66
"fmt"
7+
"slices"
78
"strconv"
89

910
errorsmod "cosmossdk.io/errors"
@@ -19,13 +20,7 @@ func (k msgServer) Open(goCtx context.Context, msg *types.MsgOpen) (*types.MsgOp
1920
ctx := sdk.UnwrapSDKContext(goCtx)
2021

2122
enabledPools := k.GetParams(ctx).EnabledPools
22-
found := false
23-
for _, poolId := range enabledPools {
24-
if poolId == msg.AmmPoolId {
25-
found = true
26-
break
27-
}
28-
}
23+
found := slices.Contains(enabledPools, msg.AmmPoolId)
2924
if !found {
3025
return nil, errorsmod.Wrap(types.ErrPoolNotEnabled, fmt.Sprintf("poolId: %d", msg.AmmPoolId))
3126
}

x/perpetual/keeper/msg_server_open.go

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package keeper
33
import (
44
"context"
55
"fmt"
6+
"slices"
67

78
errorsmod "cosmossdk.io/errors"
89
sdk "github.com/cosmos/cosmos-sdk/types"
@@ -13,13 +14,7 @@ func (k msgServer) Open(goCtx context.Context, msg *types.MsgOpen) (*types.MsgOp
1314
ctx := sdk.UnwrapSDKContext(goCtx)
1415

1516
enabledPools := k.GetParams(ctx).EnabledPools
16-
found := false
17-
for _, poolId := range enabledPools {
18-
if poolId == msg.PoolId {
19-
found = true
20-
break
21-
}
22-
}
17+
found := slices.Contains(enabledPools, msg.PoolId)
2318
if !found {
2419
return nil, errorsmod.Wrap(types.ErrPoolNotEnabled, fmt.Sprintf("poolId: %d", msg.PoolId))
2520
}

x/tradeshield/types/message_execute_orders.go

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55
"fmt"
66
sdk "github.com/cosmos/cosmos-sdk/types"
77
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
8+
"slices"
89
)
910

1011
var _ sdk.Msg = &MsgExecuteOrders{}
@@ -26,16 +27,12 @@ func (msg *MsgExecuteOrders) ValidateBasic() error {
2627
if len(msg.SpotOrderIds) == 0 && len(msg.PerpetualOrderIds) == 0 {
2728
return fmt.Errorf("SpotOrderIds and PerpetualOrderIds both are empty")
2829
}
29-
for _, id := range msg.SpotOrderIds {
30-
if id == 0 {
31-
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
32-
}
30+
if slices.Contains(msg.SpotOrderIds, 0) {
31+
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
3332
}
3433

35-
for _, id := range msg.PerpetualOrderIds {
36-
if id == 0 {
37-
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "perpetual order ID cannot be zero")
38-
}
34+
if slices.Contains(msg.PerpetualOrderIds, 0) {
35+
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "perpetual order ID cannot be zero")
3936
}
4037

4138
return nil

x/tradeshield/types/messages_pending_perpetual_order.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ package types
22

33
import (
44
"fmt"
5+
"slices"
56

67
errorsmod "cosmossdk.io/errors"
78
"cosmossdk.io/math"
@@ -182,10 +183,8 @@ func (msg *MsgCancelPerpetualOrders) ValidateBasic() error {
182183
if len(msg.OrderIds) == 0 {
183184
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order IDs cannot be empty")
184185
}
185-
for _, id := range msg.OrderIds {
186-
if id == 0 {
187-
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
188-
}
186+
if slices.Contains(msg.OrderIds, 0) {
187+
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
189188
}
190189

191190
return nil

x/tradeshield/types/messages_pending_spot_order.go

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55
"cosmossdk.io/math"
66
sdk "github.com/cosmos/cosmos-sdk/types"
77
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
8+
"slices"
89
)
910

1011
var _ sdk.Msg = &MsgCreateSpotOrder{}
@@ -124,10 +125,8 @@ func (msg *MsgCancelSpotOrders) ValidateBasic() error {
124125
if len(msg.SpotOrderIds) == 0 {
125126
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order IDs cannot be empty")
126127
}
127-
for _, id := range msg.SpotOrderIds {
128-
if id == 0 {
129-
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
130-
}
128+
if slices.Contains(msg.SpotOrderIds, 0) {
129+
return errorsmod.Wrap(sdkerrors.ErrInvalidRequest, "spot order ID cannot be zero")
131130
}
132131
return nil
133132
}

0 commit comments

Comments
 (0)