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

// Copyright 2023 Centrifuge Foundation (centrifuge.io).
// This file is part of Centrifuge chain project.

// Centrifuge is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version (see http://www.gnu.org/licenses).

// Centrifuge is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
use frame_support::weights::{constants::RocksDbWeight, Weight};

pub trait WeightInfo {
	fn add_pool() -> Weight;
	fn add_tranche() -> Weight;
	fn update_token_price() -> Weight;
	fn update_member() -> Weight;
	fn transfer() -> Weight;
	fn set_domain_router() -> Weight;
	fn add_currency() -> Weight;
	fn allow_investment_currency() -> Weight;
	fn disallow_investment_currency() -> Weight;
	fn schedule_upgrade() -> Weight;
	fn cancel_upgrade() -> Weight;
	fn update_tranche_token_metadata() -> Weight;
	fn freeze_investor() -> Weight;
	fn unfreeze_investor() -> Weight;
	fn update_tranche_hook() -> Weight;
}

// NOTE: We use temporary weights here. `execute_epoch` is by far our heaviest
//       extrinsic. N denotes the number of tranches. 4 is quite heavy and
//       should be enough.
const N: u64 = 4;

/// NOTE: Defensive hardcoded weight taken from pool_system::execute_epoch. Will
/// be replaced with real benchmark soon.

fn default_defensive_weight() -> Weight {
	Weight::from_parts(124_979_771, 19974)
		.saturating_add(Weight::from_parts(58_136_652, 0).saturating_mul(N))
		.saturating_add(RocksDbWeight::get().reads(8))
		.saturating_add(RocksDbWeight::get().reads((7_u64).saturating_mul(N)))
		.saturating_add(RocksDbWeight::get().writes(8))
		.saturating_add(RocksDbWeight::get().writes((6_u64).saturating_mul(N)))
		.saturating_add(Weight::from_parts(0, 17774).saturating_mul(N))
}

impl WeightInfo for () {
	fn set_domain_router() -> Weight {
		default_defensive_weight()
	}

	fn add_pool() -> Weight {
		default_defensive_weight()
	}

	fn add_tranche() -> Weight {
		default_defensive_weight()
	}

	fn update_token_price() -> Weight {
		default_defensive_weight()
	}

	fn update_member() -> Weight {
		default_defensive_weight()
	}

	fn transfer() -> Weight {
		default_defensive_weight()
	}

	fn schedule_upgrade() -> Weight {
		default_defensive_weight()
	}

	fn cancel_upgrade() -> Weight {
		default_defensive_weight()
	}

	fn update_tranche_token_metadata() -> Weight {
		default_defensive_weight()
	}

	fn freeze_investor() -> Weight {
		default_defensive_weight()
	}

	fn unfreeze_investor() -> Weight {
		default_defensive_weight()
	}

	fn add_currency() -> Weight {
		// Reads: 2x AssetRegistry
		// Writes: MessageNonceStore, MessageQueue
		RocksDbWeight::get().reads_writes(2, 2)
	}

	fn allow_investment_currency() -> Weight {
		// Reads: 2x AssetRegistry
		// Writes: MessageNonceStore, MessageQueue
		RocksDbWeight::get().reads_writes(2, 2)
	}

	fn disallow_investment_currency() -> Weight {
		// Reads: 2x AssetRegistry
		// Writes: MessageNonceStore, MessageQueue
		RocksDbWeight::get().reads_writes(2, 2)
	}

	fn update_tranche_hook() -> Weight {
		// Reads: Pool, Tranche, Permissions
		// Writes: MessageNonceStore, MessageQueue
		RocksDbWeight::get().reads_writes(3, 2)
	}
}