Merge pull request #167 from fjl/feature/split-tx-event

Split TxEvent type for tx pre/post

cmd/mist/gui.go

@@ -408,7 +408,8 @@ func (gui *Gui) update() {
 		eth.ChainSyncEvent{},
 		eth.PeerListEvent{},
 		ethchain.NewBlockEvent{},
-		ethchain.TxEvent{},
+		ethchain.TxPreEvent{},
+		ethchain.TxPostEvent{},
 		ethminer.Event{},
 	)
 
@@ -430,9 +431,8 @@ func (gui *Gui) update() {
 						gui.setWalletValue(gui.eth.StateManager().CurrentState().GetAccount(gui.address()).Balance(), nil)
 					}
 
-				case ethchain.TxEvent:
+				case ethchain.TxPreEvent:
 					tx := ev.Tx
-					if ev.Type == ethchain.TxPre {
 					object := state.GetAccount(gui.address())
 
 					if bytes.Compare(tx.Sender(), gui.address()) == 0 {
@@ -442,11 +442,12 @@ func (gui *Gui) update() {
 					}
 
 					gui.setWalletValue(object.Balance(), unconfirmedFunds)
-
 					gui.insertTransaction("pre", tx)
 
-					} else if ev.Type == ethchain.TxPost {
+				case ethchain.TxPostEvent:
+					tx := ev.Tx
 					object := state.GetAccount(gui.address())
+
 					if bytes.Compare(tx.Sender(), gui.address()) == 0 {
 						object.SubAmount(tx.Value)
 
@@ -460,9 +461,7 @@ func (gui *Gui) update() {
 					}
 
 					gui.setWalletValue(object.Balance(), nil)
-
 					state.UpdateStateObject(object)
-					}
 
 				// case object:
 				// 	gui.loadAddressBook()

ethchain/events.go

@@ -1,10 +1,10 @@
 package ethchain
 
-type TxEvent struct {
-	Type int // TxPre || TxPost
-	Tx   *Transaction
-}
+// TxPreEvent is posted when a transaction enters the transaction pool.
+type TxPreEvent struct{ Tx *Transaction }
 
-type NewBlockEvent struct {
-	Block *Block
-}
+// TxPostEvent is posted when a transaction has been processed.
+type TxPostEvent struct{ Tx *Transaction }
+
+// NewBlockEvent is posted when a block has been imported.
+type NewBlockEvent struct{ Block *Block }

ethchain/state_manager.go

@@ -191,7 +191,7 @@ done:
 		}
 
 		// Notify all subscribers
-		self.eth.EventMux().Post(TxEvent{TxPost, tx})
+		self.eth.EventMux().Post(TxPostEvent{tx})
 
 		receipts = append(receipts, receipt)
 		handled = append(handled, tx)

ethchain/transaction_pool.go

@@ -14,17 +14,12 @@ import (
 
 var txplogger = ethlog.NewLogger("TXP")
 
-const (
-	txPoolQueueSize = 50
-)
+const txPoolQueueSize = 50
 
 type TxPoolHook chan *Transaction
 type TxMsgTy byte
 
 const (
-	TxPre = iota
-	TxPost
-
 	minGasPrice = 1000000
 )
 
@@ -169,7 +164,7 @@ out:
 				txplogger.Debugf("(t) %x => %x (%v) %x\n", tx.Sender()[:4], tmp, tx.Value, tx.Hash())
 
 				// Notify the subscribers
-				pool.Ethereum.EventMux().Post(TxEvent{TxPre, tx})
+				pool.Ethereum.EventMux().Post(TxPreEvent{tx})
 			}
 		case <-pool.quit:
 			break out

ethminer/miner.go

@@ -64,7 +64,7 @@ func (miner *Miner) Start() {
 	miner.block = miner.ethereum.ChainManager().NewBlock(miner.coinbase)
 
 	mux := miner.ethereum.EventMux()
-	miner.events = mux.Subscribe(ethchain.NewBlockEvent{}, ethchain.TxEvent{})
+	miner.events = mux.Subscribe(ethchain.NewBlockEvent{}, ethchain.TxPreEvent{})
 
 	// Prepare inital block
 	//miner.ethereum.StateManager().Prepare(miner.block.State(), miner.block.State())
@@ -118,8 +118,7 @@ func (miner *Miner) listener() {
 				}
 				miner.startMining()
 
-			case ethchain.TxEvent:
-				if event.Type == ethchain.TxPre {
+			case ethchain.TxPreEvent:
 				miner.stopMining()
 
 				found := false
@@ -127,6 +126,8 @@ func (miner *Miner) listener() {
 					if found = bytes.Compare(ctx.Hash(), event.Tx.Hash()) == 0; found {
 						break
 					}
+
+					miner.startMining()
 				}
 				if found == false {
 					// Undo all previous commits
@@ -134,9 +135,6 @@ func (miner *Miner) listener() {
 					// Apply new transactions
 					miner.txs = append(miner.txs, event.Tx)
 				}
-
-					miner.startMining()
-				}
 			}
 
 		case <-miner.powDone: