Store attributes in masternode view (#1032)

DeFiCh · Jan 18, 2022 · ac20f8e · ac20f8e
1 parent 8401b96
commit ac20f8e
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/masternodes/govvariables/attributes.cpp b/src/masternodes/govvariables/attributes.cpp
@@ -182,6 +182,6 @@ Res ATTRIBUTES::Validate(const CCustomCSView & view) const
 
 Res ATTRIBUTES::Apply(CCustomCSView & mnview, const uint32_t height)
 {
-    // On implemenation of features that rely on ATTRIBUTES store values in mnview where they are needed.
+    mnview.SetAttributes(attributes);
     return Res::Ok();
 }
diff --git a/src/masternodes/masternodes.cpp b/src/masternodes/masternodes.cpp
@@ -544,6 +544,16 @@ uint16_t CMasternodesView::GetTimelock(const uint256& nodeId, const CMasternode&
     return 0;
 }
 
+std::map<std::string, std::string> CMasternodesView::GetAttributes() const {
+    std::map<std::string, std::string> attributes;
+    Read(Attrs::prefix(), attributes);
+    return attributes;
+}
+
+void CMasternodesView::SetAttributes(const std::map<std::string, std::string>& attributes) {
+    Write(Attrs::prefix(), attributes);
+}
+
 std::vector<int64_t> CMasternodesView::GetBlockTimes(const CKeyID& keyID, const uint32_t blockHeight, const int32_t creationHeight, const uint16_t timelock)
 {
     // Get last block time for non-subnode staking

diff --git a/src/masternodes/masternodes.h b/src/masternodes/masternodes.h
@@ -222,6 +222,9 @@ class CMasternodesView : public virtual CStorageView
 
     uint16_t GetTimelock(const uint256& nodeId, const CMasternode& node, const uint64_t height) const;
 
+    std::map<std::string, std::string> GetAttributes() const;
+    void SetAttributes(const std::map<std::string, std::string>& attributes);
+
     // tags
     struct ID       { static constexpr uint8_t prefix() { return 'M'; } };
     struct Operator { static constexpr uint8_t prefix() { return 'o'; } };
@@ -233,6 +236,9 @@ class CMasternodesView : public virtual CStorageView
 
     // Store long term time lock
     struct Timelock { static constexpr uint8_t prefix() { return 'K'; } };
+
+    // Store attributes map
+    struct Attrs { static constexpr uint8_t prefix() { return 's'; } };
 };
 
 class CLastHeightView : public virtual CStorageView
@@ -357,7 +363,7 @@ class CCustomCSView
     void CheckPrefixes()
     {
         CheckPrefix<
-            CMasternodesView        ::  ID, Operator, Owner, Staker, SubNode, Timelock,
+            CMasternodesView        ::  ID, Operator, Owner, Staker, SubNode, Timelock, Attrs,
             CLastHeightView         ::  Height,
             CTeamView               ::  AuthTeam, ConfirmTeam, CurrentTeam,
             CFoundationsDebtView    ::  Debt,