Update rewards manager

Modified by: wagerr-builder <[email protected]> - Fix database functions for stake split threshold and auto combine dust functions - Rename autocombinerewards to autocombinedust - Remove change output from autocombinedust - Cleanup scheduling and initialization
watto-engineer · Dec 30, 2022 · 4cb673a · 4cb673a
1 parent dc60283
commit 4cb673a
Show file tree

Hide file tree

Showing 11 changed files with 77 additions and 37 deletions.
diff --git a/src/init.cpp b/src/init.cpp
@@ -41,7 +41,6 @@
 #include <policy/policy.h>
 #include <policy/settings.h>
 #include <pos/staking-manager.h>
-#include <reward-manager.h>
 #include <rpc/blockchain.h>
 #include <rpc/register.h>
 #include <rpc/server.h>
@@ -359,7 +358,6 @@ void PrepareShutdown(NodeContext& node)
         evoDb.reset();
         zerocoinDB.reset();
         pTokenDB.reset();
-        rewardManager.reset();
         tokenGroupManager.reset();
     }
     for (const auto& client : node.chain_clients) {
@@ -2062,8 +2060,6 @@ bool AppInitMain(const util::Ref& context, NodeContext& node, interfaces::BlockA
                 pTokenDB.reset(new CTokenDB(0, false, fReset || fReindexChainState));
                 tokenGroupManager.reset();
                 tokenGroupManager.reset(new CTokenGroupManager());
-                rewardManager.reset();
-                rewardManager.reset(new CRewardManager());
                 llmq::quorumSnapshotManager.reset();
                 llmq::quorumSnapshotManager.reset(new llmq::CQuorumSnapshotManager(*evoDb));
 
@@ -2382,6 +2378,7 @@ bool AppInitMain(const util::Ref& context, NodeContext& node, interfaces::BlockA
     // ********************************************************* Step 10b: setup Staking
 
     g_wallet_init_interface.InitStaking();
+    g_wallet_init_interface.InitRewardsManagement();
 
     // ********************************************************* Step 10c: Load cache data
 

diff --git a/src/reward-manager.cpp b/src/reward-manager.cpp
@@ -16,7 +16,7 @@
 std::shared_ptr<CRewardManager> rewardManager;
 
 CRewardManager::CRewardManager() :
-        fEnableRewardManager(false), fEnableAutoCombineRewards(false), nAutoCombineAmountThreshold(0), nAutoCombineNThreshold(10) {
+        fEnableRewardManager(false), nAutoCombineNThreshold(10) {
 }
 
 bool CRewardManager::IsReady() {
@@ -36,15 +36,25 @@ bool CRewardManager::IsReady() {
     return true;
 }
 
-bool CRewardManager::IsCombining()
+bool CRewardManager::IsAutoCombineEnabled()
 {
-    return IsReady() && IsAutoCombineEnabled();
+    bool fEnable;
+    CAmount nAutoCombineAmountThreshold;
+    pwallet->GetAutoCombineSettings(fEnable, nAutoCombineAmountThreshold);
+    return fEnable;
 }
 
-void CRewardManager::AutoCombineSettings(bool fEnable, CAmount nAutoCombineAmountThresholdIn) {
-    LOCK(cs);
-    fEnableAutoCombineRewards = fEnable;
-    nAutoCombineAmountThreshold = nAutoCombineAmountThresholdIn;
+CAmount CRewardManager::GetAutoCombineThresholdAmount()
+{
+    bool fEnable;
+    CAmount nAutoCombineAmountThreshold;
+    pwallet->GetAutoCombineSettings(fEnable, nAutoCombineAmountThreshold);
+    return nAutoCombineAmountThreshold;
+}
+
+bool CRewardManager::IsCombining()
+{
+    return IsReady() && IsAutoCombineEnabled();
 }
 
 // TODO: replace with pwallet->FilterCoins()
@@ -67,7 +77,12 @@ std::map<CTxDestination, std::vector<COutput> > CRewardManager::AvailableCoinsBy
     return mapCoins;
 }
 
-void CRewardManager::AutoCombineRewards() {
+void CRewardManager::AutocombineDust() {
+    bool fEnable;
+    CAmount nAutoCombineAmountThreshold;
+    pwallet->GetAutoCombineSettings(fEnable, nAutoCombineAmountThreshold);
+    if (!fEnable) return;
+
     std::map<CTxDestination, std::vector<COutput> > mapCoinsByAddress = AvailableCoinsByAddress(true, nAutoCombineAmountThreshold * COIN);
 
     //coins are sectioned by address. This combination code only wants to combine inputs that belong to the same address
@@ -123,8 +138,8 @@ void CRewardManager::AutoCombineRewards() {
         std::vector<CRecipient> vecSend;
         int nChangePosRet = -1;
         CScript scriptPubKey = GetScriptForDestination(it->first);
-        // 10% safety margin to avoid "Insufficient funds" errors
-        CRecipient recipient = {scriptPubKey, nTotalRewardsValue - (nTotalRewardsValue / 10), false};
+        // Subtract fee from amount
+        CRecipient recipient = {scriptPubKey, nTotalRewardsValue, true};
         vecSend.push_back(recipient);
 
         //Send change to same address
@@ -169,7 +184,7 @@ void CRewardManager::DoMaintenance(CConnman& connman) {
     }
 
     if (IsAutoCombineEnabled()) {
-        AutoCombineRewards();
+        AutocombineDust();
         int randsleep = GetRandInt(5 * 60 * 1000);
         MilliSleep(randsleep); // Sleep between 3 and 8 minutes
     }

diff --git a/src/reward-manager.h b/src/reward-manager.h
@@ -33,20 +33,16 @@ class CRewardManager
     }
 
     bool fEnableRewardManager;
-
-    bool fEnableAutoCombineRewards;
-    CAmount nAutoCombineAmountThreshold;
     uint32_t nAutoCombineNThreshold;
 
     bool IsReady();
     bool IsCombining();
 
-    bool IsAutoCombineEnabled() { return fEnableAutoCombineRewards; };
-    CAmount GetAutoCombineThresholdAmount() { return nAutoCombineAmountThreshold; };
-    void AutoCombineSettings(bool fEnable, CAmount nAutoCombineAmountThresholdIn = 0);
+    bool IsAutoCombineEnabled();
+    CAmount GetAutoCombineThresholdAmount();
 
     std::map<CTxDestination, std::vector<COutput> > AvailableCoinsByAddress(bool fConfirmed, CAmount maxCoinValue);
-    void AutoCombineRewards();
+    void AutocombineDust();
 
     void DoMaintenance(CConnman& connman);
 };

diff --git a/src/rpc/client.cpp b/src/rpc/client.cpp
@@ -190,8 +190,8 @@ static const CRPCConvertParam vRPCConvertParams[] =
     { "getspecialtxes", 4, "verbosity" },
     { "disconnectnode", 1, "nodeid" },
     { "setstakesplitthreshold", 0, "value" },
-    { "autocombinerewards", 0, "enable" },
-    { "autocombinerewards", 1, "threshold" },
+    { "autocombinedust", 0, "enable" },
+    { "autocombinedust", 1, "threshold" },
     { "createrawtokentransaction", 0, "inputs" },
     { "createrawtokentransaction", 1, "outputs" },
     { "createrawtokentransaction", 2, "token_outputs" },

diff --git a/src/wallet/init.cpp b/src/wallet/init.cpp
@@ -12,7 +12,6 @@
 #include <univalue.h>
 
 #include <pos/staking-manager.h>
-#include <reward-manager.h>
 #include <scheduler.h>
 #include <tokens/rpctokenwallet.h>
 #include <util/check.h>
@@ -46,6 +45,7 @@ class WalletInit : public WalletInitInterface
     void AutoLockMasternodeCollaterals() const override;
     void InitCoinJoinSettings() const override;
     void InitStaking() const override;
+    void InitRewardsManagement() const override;
     bool InitAutoBackup() const override;
 };
 
@@ -236,9 +236,6 @@ void WalletInit::InitStaking() const
         stakingManager = std::shared_ptr<CStakingManager>(new CStakingManager(wallets[0]));
         stakingManager->fEnableStaking = gArgs.GetBoolArg("-staking", true);
         stakingManager->fEnableWAGERRStaking = gArgs.GetBoolArg("-staking", true);
-
-        rewardManager->BindWallet(wallets[0].get());
-        rewardManager->fEnableRewardManager = true;
     }
     if (Params().NetworkIDString() == CBaseChainParams::REGTEST) {
         stakingManager->fEnableStaking = false;
@@ -252,6 +249,16 @@ void WalletInit::InitStaking() const
     stakingManager->nReserveBalance = nReserveBalance;
 }
 
+void WalletInit::InitRewardsManagement() const
+{
+    rewardManager = std::shared_ptr<CRewardManager>(new CRewardManager());
+    std::vector<std::shared_ptr<CWallet>> wallets = GetWallets();
+    if (HasWallets() && wallets.size() >= 1) {
+        rewardManager->BindWallet(wallets[0].get());
+        rewardManager->fEnableRewardManager = true;
+    }
+}
+
 bool WalletInit::InitAutoBackup() const
 {
     return CWallet::InitAutoBackup();

diff --git a/src/wallet/rpcwallet.cpp b/src/wallet/rpcwallet.cpp
@@ -3470,7 +3470,7 @@ UniValue setstakesplitthreshold(const JSONRPCRequest& request)
 
     return result;
 }
-UniValue autocombinerewards(const JSONRPCRequest& request)
+UniValue autocombinedust(const JSONRPCRequest& request)
 {
     std::shared_ptr<CWallet> const wallet = GetWalletForJSONRPCRequest(request);
     CWallet* const pwallet = wallet.get();
@@ -3486,16 +3486,16 @@ UniValue autocombinerewards(const JSONRPCRequest& request)
 
     if (request.fHelp || nParamsSize < 1 || (!fEnable && nParamsSize > 1) || nParamsSize > 2)
         throw std::runtime_error(
-            "autocombinerewards enable ( threshold )\n"
+            "autocombinedust enable ( threshold )\n"
             "\nWallet will automatically monitor for any coins with value below the threshold amount, and combine them if they reside with the same Wagerr address\n"
-            "When autocombinerewards runs it will create a transaction, and therefore will be subject to transaction fees.\n"
+            "When autocombinedust runs it will create a transaction, and therefore will be subject to transaction fees.\n"
 
             "\nArguments:\n"
             "1. enable                  (boolean, required) Enable auto combine (true) or disable (false)\n"
             "2. threshold amount        (numeric, optional) Coins with an aggregated value of this amount will be combined (default: 0)\n"
 
             "\nExamples:\n" +
-            HelpExampleCli("autocombinerewards", "true 500") + HelpExampleRpc("autocombinerewards", "true 500"));
+            HelpExampleCli("autocombinedust", "true 500") + HelpExampleRpc("autocombinedust", "true 500"));
 
     EnsureWalletIsUnlocked(pwallet);
 
@@ -3514,7 +3514,6 @@ UniValue autocombinerewards(const JSONRPCRequest& request)
         throw std::runtime_error("Changed settings in wallet but failed to save to database\n");
     }
 
-    rewardManager->AutoCombineSettings(fEnable, nThresholdAmount);
 
     result.push_back(Pair("threshold", int(rewardManager->GetAutoCombineThresholdAmount())));
     result.push_back(Pair("enabled", rewardManager->IsAutoCombineEnabled()));
@@ -4197,6 +4196,7 @@ static const CRPCCommand commands[] =
     { "wallet",             "importelectrumwallet",             &importelectrumwallet,          {"filename", "index"} },
     { "wallet",             "getstakingstatus",                 &getstakingstatus,              {} },
     { "wallet",             "setstakesplitthreshold",           &setstakesplitthreshold,        {"value"} },
+    { "wallet",             "autocombinedust",                  &autocombinedust,               {"enable", "threshold"} },
     { "wallet",             "importmulti",                      &importmulti,                   {"requests","options"} },
     { "wallet",             "importprivkey",                    &importprivkey,                 {"privkey","label","rescan"} },
     { "wallet",             "importprunedfunds",                &importprunedfunds,             {"rawtransaction","txoutproof"} },

diff --git a/src/wallet/wallet.cpp b/src/wallet/wallet.cpp
@@ -1463,7 +1463,10 @@ CAmount CWallet::GetChange(const CTxOut& txout) const
 bool CWallet::SetStakeSplitThreshold(uint64_t newThreshold)
 {
     WalletBatch batch(*database);
-    return batch.WriteStakeSplitThreshold(newThreshold);
+    if (!batch.WriteStakeSplitThreshold(newThreshold))
+        return false;
+    LoadStakeSplitThreshold(newThreshold);
+    return true;
 }
 
 void CWallet::LoadStakeSplitThreshold(uint64_t newThreshold)
@@ -1479,7 +1482,22 @@ uint64_t CWallet::GetStakeSplitThreshold()
 bool CWallet::SetAutoCombineSettings(bool fEnable, CAmount nCombineThreshold)
 {
     WalletBatch batch(*database);
-    return batch.WriteAutoCombineSettings(fEnable, nCombineThreshold);
+    if (!batch.WriteAutoCombineSettings(fEnable, nCombineThreshold))
+        return false;
+    LoadAutoCombineSettings(fEnable, nCombineThreshold);
+    return true;
+}
+
+void CWallet::LoadAutoCombineSettings(bool fEnableIn, CAmount nCombineThresholdIn)
+{
+    fCombineEnable = fEnableIn;
+    nCombineThreshold = nCombineThresholdIn;
+}
+
+void CWallet::GetAutoCombineSettings(bool& fEnableRet, CAmount& nCombineThresholdRet) const
+{
+    fEnableRet = fCombineEnable;
+    nCombineThresholdRet = nCombineThreshold;
 }
 
 bool CWallet::IsMine(const CTransaction& tx) const

diff --git a/src/wallet/wallet.h b/src/wallet/wallet.h
@@ -818,6 +818,8 @@ class CWallet final : public WalletStorage, public interfaces::Chain::Notificati
 
     // Staking and POS
     uint64_t nStakeSplitThreshold = 2000;
+    CAmount nCombineThreshold = 500;
+    bool fCombineEnable = false;
 
     /* Height of last block processed is used by wallet to know depth of transactions
      * without relying on Chain interface beyond asynchronous updates. For safety, we
@@ -1273,6 +1275,8 @@ class CWallet final : public WalletStorage, public interfaces::Chain::Notificati
     void LoadStakeSplitThreshold(uint64_t newThreshold);
     uint64_t GetStakeSplitThreshold();
     bool SetAutoCombineSettings(bool fEnable, CAmount nCombineThreshold);
+    void LoadAutoCombineSettings(bool fEnableIn, CAmount nCombineThresholdIn);
+    void GetAutoCombineSettings(bool& fEnableRet, CAmount& nCombineThresholdRet) const;
 
     /**
      * HD Wallet Functions

diff --git a/src/wallet/walletdb.cpp b/src/wallet/walletdb.cpp
@@ -506,7 +506,10 @@ ReadKeyValue(CWallet* pwallet, CDataStream& ssKey, CDataStream& ssValue,
         {
             std::pair<bool, CAmount> pSettings;
             ssValue >> pSettings;
-            rewardManager->AutoCombineSettings(pSettings.first, pSettings.second);
+            pwallet->LoadAutoCombineSettings(pSettings.first, pSettings.second);
+        }
+        else if (strType != "bestblock" && strType != "bestblock_nomerkle"){
+            wss.m_unknown_records++;
         }
     } catch (const std::exception& e) {
         if (strErr.empty()) {

diff --git a/src/wallet/walletdb.h b/src/wallet/walletdb.h
@@ -197,7 +197,6 @@ class WalletBatch
     bool WriteMinVersion(int nVersion);
 
     bool WriteStakeSplitThreshold(uint64_t nStakeSplitThreshold);
-    bool LoadStakeSplitThreshold(uint64_t nStakeSplitThreshold);
     bool WriteAutoCombineSettings(bool fEnable, CAmount nCombineThreshold);
 
     bool ReadCoinJoinSalt(uint256& salt, bool fLegacy = false);

diff --git a/src/walletinitinterface.h b/src/walletinitinterface.h
@@ -23,6 +23,7 @@ class WalletInitInterface {
     // Wagerr Specific WalletInitInterface
     virtual void AutoLockMasternodeCollaterals() const = 0;
     void InitStaking() const override {}
+    virtual void InitRewardsManagement() const = 0;
     virtual void InitCoinJoinSettings() const = 0;
     virtual bool InitAutoBackup() const = 0;