mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2025-01-18 23:37:26 +00:00
1a6bdd4099
* [pubsub] - remove pubsub package * [chain] - Add height to chain * [peer] - remove unnecesary println * [server] - Implement server package * Add main.go to run node
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/peer"
|
|
"github.com/CityOfZion/neo-go/pkg/syncmgr"
|
|
"github.com/CityOfZion/neo-go/pkg/wire/payload"
|
|
"github.com/CityOfZion/neo-go/pkg/wire/util"
|
|
)
|
|
|
|
func setupSyncManager(s *Server) *syncmgr.Syncmgr {
|
|
|
|
cfg := &syncmgr.Config{
|
|
ProcessBlock: s.processBlock,
|
|
ProcessHeaders: s.processHeaders,
|
|
|
|
RequestBlock: s.requestBlock,
|
|
RequestHeaders: s.requestHeaders,
|
|
|
|
GetNextBlockHash: s.getNextBlockHash,
|
|
AskForNewBlocks: s.askForNewBlocks,
|
|
|
|
FetchHeadersAgain: s.fetchHeadersAgain,
|
|
FetchBlockAgain: s.fetchBlockAgain,
|
|
}
|
|
|
|
return syncmgr.New(cfg)
|
|
}
|
|
|
|
func (s *Server) onHeader(peer *peer.Peer, hdrsMessage *payload.HeadersMessage) {
|
|
s.pmg.MsgReceived(peer, hdrsMessage.Command())
|
|
s.smg.OnHeader(peer, hdrsMessage)
|
|
}
|
|
|
|
func (s *Server) onBlock(peer *peer.Peer, blockMsg *payload.BlockMessage) {
|
|
s.pmg.MsgReceived(peer, blockMsg.Command())
|
|
s.smg.OnBlock(peer, blockMsg)
|
|
}
|
|
|
|
func (s *Server) processBlock(block payload.Block) error {
|
|
return s.chain.ProcessBlock(block)
|
|
}
|
|
|
|
func (s *Server) processHeaders(hdrs []*payload.BlockBase) error {
|
|
return s.chain.ProcessHeaders(hdrs)
|
|
}
|
|
|
|
func (s *Server) requestHeaders(hash util.Uint256) error {
|
|
return s.pmg.RequestHeaders(hash)
|
|
}
|
|
|
|
func (s *Server) requestBlock(hash util.Uint256) error {
|
|
return s.pmg.RequestBlock(hash)
|
|
}
|
|
|
|
// getNextBlockHash searches the database for the blockHash
|
|
// that is the height above our best block. The hash will be taken from a header.
|
|
func (s *Server) getNextBlockHash() (util.Uint256, error) {
|
|
bestBlock, err := s.chain.Db.GetLastBlock()
|
|
if err != nil {
|
|
// Panic!
|
|
// XXX: One alternative, is to get the network, erase the database and then start again from scratch.
|
|
// This should never happen. The latest block will always be atleast the genesis block
|
|
panic("could not get best block from database" + err.Error())
|
|
}
|
|
|
|
index := make([]byte, 4)
|
|
binary.BigEndian.PutUint32(index, bestBlock.Index+1)
|
|
|
|
hdr, err := s.chain.Db.GetHeaderFromHeight(index)
|
|
if err != nil {
|
|
return util.Uint256{}, err
|
|
}
|
|
return hdr.Hash, nil
|
|
}
|
|
|
|
func (s *Server) getBestBlockHash() (util.Uint256, error) {
|
|
return util.Uint256{}, nil
|
|
}
|
|
|
|
func (s *Server) askForNewBlocks() {
|
|
// send a getblocks message with the latest block saved
|
|
|
|
// when we receive something then send get data
|
|
}
|
|
|
|
func (s *Server) fetchHeadersAgain(util.Uint256) error {
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) fetchBlockAgain(util.Uint256) error {
|
|
return nil
|
|
}
|