mirrors/bitcoinj
1
0
Fork 0
mirror of https://github.com/bitcoinj/bitcoinj.git synced 2025-02-26 07:23:53 +01:00
Code Issues Projects Releases Packages Wiki Activity

WalletAppKit: add a new override-able method that lets you customise objects after they were created, but before chain download starts.

Browse source
This commit is contained in:
Mike Hearn 2013-07-29 17:18:48 +02:00
parent 872967e96e
commit f821207a80
1 changed files with 40 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

59
core/src/main/java/com/google/bitcoin/kits/WalletAppKit.java
View file

@ -26,6 +26,8 @@ import com.google.common.util.concurrent.AbstractIdleService;
import java.io.File; import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.IOException; import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import static com.google.common.base.Preconditions.checkNotNull; import static com.google.common.base.Preconditions.checkNotNull;
@ -43,18 +45,18 @@ import static com.google.common.base.Preconditions.checkState;
* out what went wrong more precisely. Same thing if you use the async start() method.</p> * out what went wrong more precisely. Same thing if you use the async start() method.</p>
*/ */
public class WalletAppKit extends AbstractIdleService { public class WalletAppKit extends AbstractIdleService {
protected final String filePrefix; private final String filePrefix;
protected final NetworkParameters params; private final NetworkParameters params;
protected volatile BlockChain vChain; private volatile BlockChain vChain;
protected volatile SPVBlockStore vStore; private volatile SPVBlockStore vStore;
protected volatile Wallet vWallet; private volatile Wallet vWallet;
protected volatile PeerGroup vPeerGroup; private volatile PeerGroup vPeerGroup;
protected volatile boolean vUseAutoSave = true;
protected final File directory; private final File directory;
protected volatile File vChainFile, vWalletFile; private volatile File vWalletFile;
protected volatile PeerAddress[] vPeerAddresses; private boolean useAutoSave = true;
private PeerAddress[] peerAddresses;
public WalletAppKit(NetworkParameters params, File directory, String filePrefix) { public WalletAppKit(NetworkParameters params, File directory, String filePrefix) {
this.params = checkNotNull(params); this.params = checkNotNull(params);
@ -62,15 +64,27 @@ public class WalletAppKit extends AbstractIdleService {
this.filePrefix = checkNotNull(filePrefix); this.filePrefix = checkNotNull(filePrefix);
} }
/** Will only connect to the given addresses. Cannot be called after startup. */
public WalletAppKit setPeerNodes(PeerAddress... addresses) { public WalletAppKit setPeerNodes(PeerAddress... addresses) {
checkState(state() == State.NEW, "Cannot call after startup"); checkState(state() == State.NEW, "Cannot call after startup");
this.vPeerAddresses = addresses; this.peerAddresses = addresses;
return this; return this;
} }
/** Will only connect to localhost. Cannot be called after startup. */
public WalletAppKit connectToLocalHost() {
try {
final InetAddress localHost = InetAddress.getLocalHost();
return setPeerNodes(new PeerAddress(localHost, params.getPort()));
} catch (UnknownHostException e) {
// Borked machine with no loopback adapter configured properly.
throw new RuntimeException(e);
}
}
public WalletAppKit setAutoSave(boolean value) { public WalletAppKit setAutoSave(boolean value) {
checkState(state() == State.NEW, "Cannot call after startup"); checkState(state() == State.NEW, "Cannot call after startup");
vUseAutoSave = value; useAutoSave = value;
return this; return this;
} }
@ -82,6 +96,12 @@ public class WalletAppKit extends AbstractIdleService {
*/ */
protected void addWalletExtensions() throws Exception { } protected void addWalletExtensions() throws Exception { }
/**
* This method is invoked on a background thread after all objects are initialised, but before the peer group
* or block chain download is started. You can tweak the objects configuration here.
*/
protected void onSetupCompleted() { }
@Override @Override
protected void startUp() throws Exception { protected void startUp() throws Exception {
// Runs in a separate thread. // Runs in a separate thread.
@ -92,17 +112,17 @@ public class WalletAppKit extends AbstractIdleService {
} }
FileInputStream walletStream = null; FileInputStream walletStream = null;
try { try {
vChainFile = new File(directory, filePrefix + ".spvchain"); File chainFile = new File(directory, filePrefix + ".spvchain");
vWalletFile = new File(directory, filePrefix + ".wallet"); vWalletFile = new File(directory, filePrefix + ".wallet");
boolean shouldReplayWallet = vWalletFile.exists() && !vChainFile.exists(); boolean shouldReplayWallet = vWalletFile.exists() && !chainFile.exists();
vStore = new SPVBlockStore(params, vChainFile); vStore = new SPVBlockStore(params, chainFile);
vChain = new BlockChain(params, vStore); vChain = new BlockChain(params, vStore);
vPeerGroup = new PeerGroup(params, vChain); vPeerGroup = new PeerGroup(params, vChain);
// Set up peer addresses or discovery first, so if wallet extensions try to broadcast a transaction // Set up peer addresses or discovery first, so if wallet extensions try to broadcast a transaction
// before we're actually connected the broadcast waits for an appropriate number of connections. // before we're actually connected the broadcast waits for an appropriate number of connections.
if (vPeerAddresses != null) { if (peerAddresses != null) {
for (PeerAddress addr : vPeerAddresses) vPeerGroup.addAddress(addr); for (PeerAddress addr : peerAddresses) vPeerGroup.addAddress(addr);
vPeerAddresses = null; peerAddresses = null;
} else { } else {
vPeerGroup.addPeerDiscovery(new DnsDiscovery(params)); vPeerGroup.addPeerDiscovery(new DnsDiscovery(params));
} }
@ -117,9 +137,10 @@ public class WalletAppKit extends AbstractIdleService {
vWallet = new Wallet(params); vWallet = new Wallet(params);
addWalletExtensions(); addWalletExtensions();
} }
if (vUseAutoSave) vWallet.autosaveToFile(vWalletFile, 1, TimeUnit.SECONDS, null); if (useAutoSave) vWallet.autosaveToFile(vWalletFile, 1, TimeUnit.SECONDS, null);
vChain.addWallet(vWallet); vChain.addWallet(vWallet);
vPeerGroup.addWallet(vWallet); vPeerGroup.addWallet(vWallet);
onSetupCompleted();
vPeerGroup.startAndWait(); vPeerGroup.startAndWait();
vPeerGroup.downloadBlockChain(); vPeerGroup.downloadBlockChain();
// Make sure we shut down cleanly. // Make sure we shut down cleanly.