From fa0ceaf2e005912a8f71496ab434f443f58435dd Mon Sep 17 00:00:00 2001
From: mingda <leimingda@pku.edu.cn>
Date: Sat, 26 Sep 2020 10:34:57 +0800
Subject: [PATCH] dodo mine setreward mass update

---
 contracts/token/DODOMine.sol | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/contracts/token/DODOMine.sol b/contracts/token/DODOMine.sol
index c22391d..8734c9f 100644
--- a/contracts/token/DODOMine.sol
+++ b/contracts/token/DODOMine.sol
@@ -132,7 +132,10 @@ contract DODOMine is Ownable {
         poolInfos[pid].allocPoint = _allocPoint;
     }
 
-    function setReward(uint256 _dodoPerBlock) external onlyOwner {
+    function setReward(uint256 _dodoPerBlock, bool _withUpdate) external onlyOwner {
+        if (_withUpdate) {
+            massUpdatePools();
+        }
         dodoPerBlock = _dodoPerBlock;
     }