-
Notifications
You must be signed in to change notification settings - Fork 199
/
msg_server_freeze.go
49 lines (38 loc) · 1.92 KB
/
msg_server_freeze.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
package keeper
import (
"context"
"fmt"
"strconv"
"strings"
sdk "github.com/cosmos/cosmos-sdk/types"
commontypes "github.com/lavanet/lava/common/types"
"github.com/lavanet/lava/utils"
"github.com/lavanet/lava/x/pairing/types"
)
func (k msgServer) FreezeProvider(goCtx context.Context, msg *types.MsgFreezeProvider) (*types.MsgFreezeProviderResponse, error) {
ctx := sdk.UnwrapSDKContext(goCtx)
err := k.Keeper.FreezeProvider(ctx, msg.GetCreator(), msg.GetChainIds(), msg.Reason)
return &types.MsgFreezeProviderResponse{}, err
}
func (k Keeper) FreezeProvider(ctx sdk.Context, provider string, chainIDs []string, reason string) error {
providerAddr, err := sdk.AccAddressFromBech32(provider)
if err != nil {
return utils.LavaFormatWarning("Freeze_get_provider_address", err, utils.Attribute{Key: "providerAddress", Value: provider})
}
if !commontypes.ValidateString(reason, commontypes.DESCRIPTION_RESTRICTIONS, nil) {
return utils.LavaFormatWarning("Freeze_invalid_reason", fmt.Errorf("invalid string"),
utils.LogAttr("reason", reason),
)
}
for _, chainId := range chainIDs {
stakeEntry, found, index := k.epochStorageKeeper.GetStakeEntryByAddressCurrent(ctx, chainId, providerAddr)
if !found {
return utils.LavaFormatWarning("Freeze_cant_get_stake_entry", types.FreezeStakeEntryNotFoundError, []utils.Attribute{{Key: "chainID", Value: chainId}, {Key: "providerAddress", Value: provider}}...)
}
// freeze the provider by making the StakeAppliedBlock be max. This will remove the provider from the pairing list in the next epoch
stakeEntry.Freeze()
k.epochStorageKeeper.ModifyStakeEntryCurrent(ctx, chainId, stakeEntry, index)
}
utils.LogLavaEvent(ctx, ctx.Logger(), "freeze_provider", map[string]string{"providerAddress": providerAddr.String(), "chainIDs": strings.Join(chainIDs, ","), "freezeRequestBlock": strconv.FormatInt(ctx.BlockHeight(), 10), "freezeReason": reason}, "Provider Freeze")
return nil
}