Fix portfolio asset count.

PolymeshAssociation · Feb 14, 2024 · 4247f7c · 4247f7c
1 parent a461eca
commit 4247f7c
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 9 deletions.
diff --git a/pallets/asset/src/lib.rs b/pallets/asset/src/lib.rs
@@ -102,7 +102,6 @@ use pallet_base::{
     ensure_opt_string_limited, ensure_string_limited, try_next_pre, Error::CounterOverflow,
 };
 use pallet_identity::PermissionedCallOriginData;
-use pallet_portfolio::PortfolioAssetBalances;
 use polymesh_common_utilities::asset::AssetFnTrait;
 use polymesh_common_utilities::compliance_manager::ComplianceFnConfig;
 use polymesh_common_utilities::constants::*;
@@ -1402,7 +1401,7 @@ impl<T: Config> Module<T> {
         // Increase total supply.
         token.total_supply = updated_total_supply;
         BalanceOf::insert(ticker, issuer_portfolio.did, updated_to_balance);
-        PortfolioAssetBalances::insert(issuer_portfolio, ticker, updated_to_def_balance);
+        Portfolio::<T>::set_portfolio_balance(issuer_portfolio, ticker, updated_to_def_balance);
         Tokens::insert(ticker, token);
 
         Statistics::<T>::update_asset_stats(

diff --git a/pallets/portfolio/src/lib.rs b/pallets/portfolio/src/lib.rs
@@ -34,7 +34,7 @@
 //!
 //! - `default_portfolio_balance`: Returns the ticker balance of the identity's default portfolio.
 //! - `user_portfolio_balance`: Returns the ticker balance of an identity's user portfolio.
-//! - `set_default_portfolio_balance`: Sets the ticker balance of the identity's default portfolio.
+//! - `set_portfolio_balance`: Sets the ticker balance of a portfolio.
 //! - `unchecked_transfer_portfolio_balance`: Transfers funds from one portfolio to another.
 //! - `ensure_portfolio_custody`: Makes sure that the given identity has custodian access over the portfolio.
 //! - `ensure_portfolio_transfer_validity`: Makes sure that a transfer between two portfolios is valid.
@@ -447,9 +447,8 @@ impl<T: Config> Module<T> {
         Self::portfolio_asset_balances(PortfolioId::user_portfolio(did, num), ticker)
     }
 
-    /// Sets the ticker balance of the identity's default portfolio to `new`.
-    pub fn set_default_portfolio_balance(did: IdentityId, ticker: &Ticker, new: Balance) {
-        let pid = PortfolioId::default_portfolio(did);
+    /// Sets the ticker balance of the a portfolio to `new`.
+    pub fn set_portfolio_balance(pid: PortfolioId, ticker: &Ticker, new: Balance) {
         PortfolioAssetBalances::mutate(&pid, ticker, |old| {
             Self::transition_asset_count(&pid, *old, new);
             *old = new;

diff --git a/pallets/runtime/develop/src/runtime.rs b/pallets/runtime/develop/src/runtime.rs
@@ -53,7 +53,7 @@ pub const VERSION: RuntimeVersion = RuntimeVersion {
     authoring_version: 1,
     // `spec_version: aaa_bbb_ccd` should match node version v`aaa.bbb.cc`
     // N.B. `d` is unpinned from the binary version
-    spec_version: 6_001_030,
+    spec_version: 6_001_031,
     impl_version: 0,
     apis: RUNTIME_API_VERSIONS,
     transaction_version: 4,

diff --git a/pallets/runtime/mainnet/src/runtime.rs b/pallets/runtime/mainnet/src/runtime.rs
@@ -51,7 +51,7 @@ pub const VERSION: RuntimeVersion = RuntimeVersion {
     authoring_version: 1,
     // `spec_version: aaa_bbb_ccd` should match node version v`aaa.bbb.cc`
     // N.B. `d` is unpinned from the binary version
-    spec_version: 6_001_030,
+    spec_version: 6_001_031,
     impl_version: 0,
     apis: RUNTIME_API_VERSIONS,
     transaction_version: 4,

diff --git a/pallets/runtime/testnet/src/runtime.rs b/pallets/runtime/testnet/src/runtime.rs
@@ -56,7 +56,7 @@ pub const VERSION: RuntimeVersion = RuntimeVersion {
     authoring_version: 1,
     // `spec_version: aaa_bbb_ccd` should match node version v`aaa.bbb.cc`
     // N.B. `d` is unpinned from the binary version
-    spec_version: 6_001_030,
+    spec_version: 6_001_031,
     impl_version: 0,
     apis: RUNTIME_API_VERSIONS,
     transaction_version: 4,