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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
| package ibc
import (
sdk "github.com/cosmos/cosmos-sdk/types"
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
)
type IBCTransferKeeper struct {
bankKeeper banktypes.Keeper
channelKeeper ChannelKeeper
portKeeper PortKeeper
}
// 跨链转账
func (k IBCTransferKeeper) Transfer(
ctx sdk.Context,
sourcePort string,
sourceChannel string,
token sdk.Coin,
sender sdk.AccAddress,
receiver string,
timeoutHeight uint64,
) error {
// 1. 验证参数
if token.Amount.IsZero() {
return fmt.Errorf("amount must be positive")
}
// 2. 从发送者账户扣除代币
if err := k.bankKeeper.SendCoins(
ctx,
sender,
accountAddr,
sdk.NewCoins(token),
); err != nil {
return err
}
// 3. 创建IBC数据包
packetData := TransferData{
Sender: sender.String(),
Receiver: receiver,
Amount: token,
Memo: "",
}
packetBz := modulecdc.MustMarshalJSON(&packetData)
packet := channeltypes.Packet{
Data: packetBz,
TimeoutHeight: clienttypes.Height{RevisionNumber: 0, RevisionHeight: timeoutHeight},
TimeoutTimestamp: 0,
}
// 4. 发送数据包
_, err := k.channelKeeper.SendPacket(ctx, packet)
if err != nil {
return err
}
ctx.EventManager().EmitEvents(
sdk.Events{
sdk.NewEvent(
"ibc_transfer",
sdk.NewAttribute("sender", sender.String()),
sdk.NewAttribute("receiver", receiver),
sdk.NewAttribute("amount", token.String()),
),
},
)
return nil
}
// 接收跨链代币
func (k IBCTransferKeeper) OnRecvPacket(
ctx sdk.Context,
packet channeltypes.Packet,
relayer sdk.AccAddress,
) exported.Acknowledgement {
var data TransferData
if err := modulecdc.UnmarshalJSON(packet.Data, &data); err != nil {
return channeltypes.NewErrorAcknowledgement(err)
}
// 解析接收者地址
receiver, err := sdk.AccAddressFromBech32(data.Receiver)
if err != nil {
return channeltypes.NewErrorAcknowledgement(err)
}
// 铸造代币
coins := sdk.NewCoins(data.Amount)
if err := k.bankKeeper.MintCoins(ctx, coins); err != nil {
return channeltypes.NewErrorAcknowledgement(err)
}
// 发送给接收者
if err := k.bankKeeper.SendCoins(
ctx,
accountAddr,
receiver,
coins,
); err != nil {
return channeltypes.NewErrorAcknowledgement(err)
}
// 记录事件
ctx.EventManager().EmitEvents(
sdk.Events{
sdk.NewEvent(
"ibc_receive",
sdk.NewAttribute("receiver", receiver.String()),
sdk.NewAttribute("amount", data.Amount.String()),
),
},
)
return channeltypes.NewResultAcknowledgement([]byte{0x01})
}
|