Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

export internal status for monitor and debug #373

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,5 @@ Chain/*
/core/*/*_test.go
/core/*/*/*_test.go
/config/*
node
nodectl
cscope*
8 changes: 6 additions & 2 deletions core/store/ChainStore/ChainStore.go
Original file line number Diff line number Diff line change
Expand Up @@ -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) {
Expand Down
53 changes: 53 additions & 0 deletions core/store/ChainStore/debug.go
Original file line number Diff line number Diff line change
@@ -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)))
}
2 changes: 2 additions & 0 deletions net/net.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,5 +25,7 @@ func StartProtocol(pubKey *crypto.PubKey) protocol.Noder {
net := node.InitNode(pubKey)
net.ConnectSeeds()

node.ExportNodeStatus(net)

return net
}
68 changes: 68 additions & 0 deletions net/node/debug.go
Original file line number Diff line number Diff line change
@@ -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 {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should add lock to guard node.nbrNodes.List

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

fixed

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))))
}
6 changes: 6 additions & 0 deletions net/node/transactionPool.go
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down