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

use crate::{AssetIdOf, Config, LockInfo, LockerData, Timestamp, Weight};
use common::{convert_block_number_to_timestamp, Balance};
use frame_support::log;
use frame_support::traits::Get;
use sp_std::vec::Vec;

pub fn migrate<T: Config>() -> Weight {
    sp_runtime::runtime_logger::RuntimeLogger::init();
    migrate_locker_data::<T>()
}

pub fn migrate_locker_data<T: Config>() -> Weight {
    let mut weight: u64 = 0;

    let current_timestamp = Timestamp::<T>::get();
    let current_block = frame_system::Pallet::<T>::block_number();
    LockerData::<T>::translate_values::<
        Vec<(Balance, T::BlockNumber, AssetIdOf<T>, AssetIdOf<T>)>,
        _,
    >(|v| {
        Some(
            v.into_iter()
                .map(|(pool_tokens, unlocking_block, asset_a, asset_b)| {
                    weight += 1;
                    let unlocking_timestamp = convert_block_number_to_timestamp::<T>(
                        unlocking_block,
                        current_block,
                        current_timestamp,
                    );

                    LockInfo {
                        pool_tokens,
                        unlocking_timestamp,
                        asset_a,
                        asset_b,
                    }
                })
                .collect::<Vec<LockInfo<Balance, T::Moment, AssetIdOf<T>>>>(),
        )
    });

    log::info!(
        target: "runtime",
        "LockInfo migrated to new version with unlocking_timestamp field"
    );

    T::DbWeight::get().reads_writes(weight, weight)
}