export internal status for monitor and debug

currently have exported the status of local node, peers, connections, transaction pool and ChainStore,
which can be accessed from http://nodeip:HttpJsonPort/debug/vars

Signed-off-by: laizy <[email protected]>

.gitignore

@@ -18,6 +18,5 @@ Chain/*
 /core/*/*_test.go
 /core/*/*/*_test.go
 /config/*
-node
 nodectl
 cscope*

core/store/ChainStore/ChainStore.go

@@ -73,15 +73,19 @@ func NewChainStore(file string) (*ChainStore, error) {
 		return nil, err
 	}
 
-	return &ChainStore{
+	chain := &ChainStore{
 		st:                 st,
 		headerIndex:        map[uint32]Uint256{},
 		blockCache:         map[Uint256]*Block{},
 		headerCache:        map[Uint256]*Header{},
 		currentBlockHeight: 0,
 		storedHeaderCount:  0,
 		disposed:           false,
-	}, nil
+	}
+
+	ExportStoreStatus(chain)
+
+	return chain, nil
 }
 
 func (bd *ChainStore) InitLedgerStoreWithGenesisBlock(genesisBlock *Block, defaultBookKeeper []*crypto.PubKey) (uint32, error) {

core/store/ChainStore/debug.go

@@ -0,0 +1,53 @@
+package ChainStore
+
+import (
+	// . "DNA/common"
+	"expvar"
+	"fmt"
+	// "time"
+)
+
+type storeStatus struct {
+	CurrHeaderHeight uint32
+	CurrBlockHeight  uint32
+	HeaderCache      map[uint32]string
+	BlockCache       map[uint32]string
+	HeaderIndex      map[uint32]string
+}
+
+func expvarStore(store *ChainStore) func() interface{} {
+
+	return func() interface{} {
+		headerHeight := store.GetHeaderHeight()
+		blockHeight := store.GetHeight()
+		store.mu.RLock()
+		defer store.mu.RUnlock()
+
+		ss := storeStatus{
+			HeaderIndex:      make(map[uint32]string, len(store.headerIndex)),
+			CurrHeaderHeight: headerHeight,
+			CurrBlockHeight:  blockHeight,
+			HeaderCache:      make(map[uint32]string, len(store.headerCache)),
+			BlockCache:       make(map[uint32]string, len(store.blockCache)),
+		}
+
+		for k, hash := range store.headerIndex {
+			ss.HeaderIndex[k] = fmt.Sprintf("%x", hash)
+		}
+
+		for k, header := range store.headerCache {
+			ss.HeaderCache[header.Blockdata.Height] = fmt.Sprintf("%x", k)
+		}
+		for k, block := range store.blockCache {
+			ss.BlockCache[block.Blockdata.Height] = fmt.Sprintf("%x", k)
+		}
+
+		return ss
+	}
+
+}
+
+func ExportStoreStatus(store *ChainStore) {
+
+	expvar.Publish("dna_store", expvar.Func(expvarStore(store)))
+}

net/net.go

@@ -25,5 +25,7 @@ func StartProtocol(pubKey *crypto.PubKey) protocol.Noder {
 	net := node.InitNode(pubKey)
 	net.ConnectSeeds()
 
+	node.ExportNodeStatus(net)
+
 	return net
 }

net/node/debug.go

@@ -0,0 +1,68 @@
+package node
+
+import (
+	. "DNA/net/protocol"
+	"expvar"
+	"fmt"
+	"time"
+)
+
+type PeerStatus struct {
+	State         uint32
+	FlightHeights []uint32
+	LastContact   string
+	TryTimes      uint32
+	Addr          string // The address of the node
+	Height        uint64 // The node latest block height
+}
+
+type TxPoolStatus struct {
+	TxCount int
+}
+
+type nodeStatus struct {
+	Id          uint64
+	TxnCnt      uint64 // The transactions be transmit by this node
+	RxTxnCnt    uint64 // The transaction received by this node
+	PublicKey   string
+	Peers       []PeerStatus
+	Connectings []string
+	TxPool      TxPoolStatus
+}
+
+func expvarNodeInfo(node *node) func() interface{} {
+
+	return func() interface{} {
+		pbkey, _ := node.publicKey.EncodePoint(true)
+
+		ns := nodeStatus{
+			Id:          node.id,
+			TxnCnt:      node.txnCnt,
+			RxTxnCnt:    node.rxTxnCnt,
+			PublicKey:   fmt.Sprintf("%x", pbkey),
+			TxPool:      TxPoolStatus{TxCount: node.TXNPool.Len()},
+			Connectings: node.ConnectingAddrs,
+		}
+
+		node.nbrNodes.RLock()
+		for _, n := range node.nbrNodes.List {
+			peer := PeerStatus{
+				State:         n.state,
+				Height:        n.height,
+				FlightHeights: n.flightHeights,
+				LastContact:   fmt.Sprintf("%gs", float64(time.Now().Sub(n.link.time))/float64(time.Second)),
+				TryTimes:      n.tryTimes,
+				Addr:          fmt.Sprintf("%s:%d", n.link.addr, n.link.port),
+			}
+
+			ns.Peers = append(ns.Peers, peer)
+		}
+		node.nbrNodes.RUnlock()
+
+		return ns
+	}
+}
+
+func ExportNodeStatus(nd Noder) {
+	expvar.Publish("dna_node", expvar.Func(expvarNodeInfo(nd.(*node))))
+}

net/node/transactionPool.go

@@ -29,6 +29,12 @@ func (this *TXNPool) init() {
 	this.txnList = make(map[common.Uint256]*transaction.Transaction)
 }
 
+func (this *TXNPool) Len() int {
+	this.RLock()
+	defer this.RUnlock()
+	return len(this.txnList)
+}
+
 //append transaction to txnpool when check ok.
 //1.check transaction. 2.check with ledger(db) 3.check with pool
 func (this *TXNPool) AppendTxnPool(txn *transaction.Transaction) bool {