Merge pull request PortalNetwork#15 from seanchuangportal/develop

Support contract withdraw
wittyCodeX · Jul 24, 2018 · c12b6ce · c12b6ce
2 parents 094594b + 5cecaa6
commit c12b6ce
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/contracts/CryptoHerosGame.sol b/contracts/CryptoHerosGame.sol
@@ -69,4 +69,10 @@ contract CryptoHerosGame is Ownable {
     return uint(sha3(nonce))%(min+max)-min;
   }
 
+  function withdraw(uint amount) public payable onlyOwner returns(bool) {
+    require(amount <= this.balance);
+    owner.transfer(amount);
+    return true;
+  }
+
 }
diff --git a/contracts/CryptoHerosToken.sol b/contracts/CryptoHerosToken.sol
@@ -89,5 +89,11 @@ contract CryptoHerosToken is ERC721Token, Ownable {
   function getHerosLength() public returns (uint) {
     return heros.length;
   }
+
+  function withdraw(uint amount) public payable onlyOwner returns(bool) {
+    require(amount <= this.balance);
+    owner.transfer(amount);
+    return true;
+  }
 
 }