|
@@ -22,8 +22,9 @@ use crate::{
|
|
|
};
|
|
|
|
|
|
use crate::{
|
|
|
- ContentWorkingGroupInstance, OperationsWorkingGroupInstanceAlpha,
|
|
|
- OperationsWorkingGroupInstanceBeta, OperationsWorkingGroupInstanceGamma,
|
|
|
+ ContentWorkingGroupInstance, DistributionWorkingGroupInstance,
|
|
|
+ OperationsWorkingGroupInstanceAlpha, OperationsWorkingGroupInstanceBeta,
|
|
|
+ OperationsWorkingGroupInstanceGamma,
|
|
|
};
|
|
|
use frame_support::weights::Weight;
|
|
|
|
|
@@ -73,10 +74,16 @@ pub(crate) type OperationsWorkingGroupGamma<T> =
|
|
|
|
|
|
pub(crate) type ContentWorkingGroup<T> = working_group::Module<T, ContentWorkingGroupInstance>;
|
|
|
|
|
|
+pub(crate) type DistributionWorkingGroup<T> =
|
|
|
+ working_group::Module<T, DistributionWorkingGroupInstance>;
|
|
|
+
|
|
|
/// Custom runtime upgrade handler.
|
|
|
pub struct CustomOnRuntimeUpgrade;
|
|
|
impl OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
|
|
|
fn on_runtime_upgrade() -> Weight {
|
|
|
+ // initialize content module
|
|
|
+ content::Module::<Runtime>::on_runtime_upgrade();
|
|
|
+
|
|
|
// Initialize new groups
|
|
|
let default_text_constraint = crate::working_group::default_text_constraint();
|
|
|
|
|
@@ -117,6 +124,14 @@ impl OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
|
|
|
default_content_working_group_mint_capacity,
|
|
|
);
|
|
|
|
|
|
+ DistributionWorkingGroup::<Runtime>::initialize_working_group(
|
|
|
+ default_text_constraint,
|
|
|
+ default_text_constraint,
|
|
|
+ default_text_constraint,
|
|
|
+ default_storage_size_constraint,
|
|
|
+ default_content_working_group_mint_capacity,
|
|
|
+ );
|
|
|
+
|
|
|
10_000_000 // TODO: adjust weight
|
|
|
}
|
|
|
}
|