mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-13 03:25:20 +00:00
55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package v0_16
|
|
|
|
import (
|
|
v015swap "github.com/0glabs/0g-chain/x/swap/legacy/v0_15"
|
|
v016swap "github.com/0glabs/0g-chain/x/swap/types"
|
|
)
|
|
|
|
func migrateParams(params v015swap.Params) v016swap.Params {
|
|
allowedPools := make(v016swap.AllowedPools, len(params.AllowedPools))
|
|
for i, pool := range params.AllowedPools {
|
|
allowedPools[i] = v016swap.AllowedPool{
|
|
TokenA: pool.TokenA,
|
|
TokenB: pool.TokenB,
|
|
}
|
|
}
|
|
return v016swap.Params{
|
|
AllowedPools: allowedPools,
|
|
SwapFee: params.SwapFee,
|
|
}
|
|
}
|
|
|
|
func migratePoolRecords(oldRecords v015swap.PoolRecords) v016swap.PoolRecords {
|
|
newRecords := make(v016swap.PoolRecords, len(oldRecords))
|
|
for i, oldRecord := range oldRecords {
|
|
newRecords[i] = v016swap.PoolRecord{
|
|
PoolID: oldRecord.PoolID,
|
|
ReservesA: oldRecord.ReservesA,
|
|
ReservesB: oldRecord.ReservesB,
|
|
TotalShares: oldRecord.TotalShares,
|
|
}
|
|
}
|
|
return newRecords
|
|
}
|
|
|
|
func migrateShareRecords(oldRecords v015swap.ShareRecords) v016swap.ShareRecords {
|
|
newRecords := make(v016swap.ShareRecords, len(oldRecords))
|
|
for i, oldRecord := range oldRecords {
|
|
newRecords[i] = v016swap.ShareRecord{
|
|
Depositor: oldRecord.Depositor,
|
|
PoolID: oldRecord.PoolID,
|
|
SharesOwned: oldRecord.SharesOwned,
|
|
}
|
|
}
|
|
return newRecords
|
|
}
|
|
|
|
// Migrate converts v0.15 swap state and returns it in v0.16 format
|
|
func Migrate(oldState v015swap.GenesisState) *v016swap.GenesisState {
|
|
return &v016swap.GenesisState{
|
|
Params: migrateParams(oldState.Params),
|
|
PoolRecords: migratePoolRecords(oldState.PoolRecords),
|
|
ShareRecords: migrateShareRecords(oldState.ShareRecords),
|
|
}
|
|
}
|