mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-06-22 14:02:32 +00:00
8a4ece1101
* [NOD-1223] Move all network stuff into a new network package. * [NOD-1223] Delete the unused package testutil. * [NOD-1223] Move infrastructure stuff into a new instrastructure package. * [NOD-1223] Move domain stuff into a new domain package.
38 lines
907 B
Go
38 lines
907 B
Go
package integration
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/network/domainmessage"
|
|
"github.com/kaspanet/kaspad/util"
|
|
|
|
rpcclient "github.com/kaspanet/kaspad/network/rpc/client"
|
|
)
|
|
|
|
type rpcClient struct {
|
|
*rpcclient.Client
|
|
onBlockAdded func(*domainmessage.BlockHeader)
|
|
}
|
|
|
|
func newRPCClient(rpcAddress string) (*rpcClient, error) {
|
|
client := &rpcClient{}
|
|
notificationHandlers := &rpcclient.NotificationHandlers{
|
|
OnFilteredBlockAdded: func(height uint64, header *domainmessage.BlockHeader, txs []*util.Tx) {
|
|
if client.onBlockAdded != nil {
|
|
client.onBlockAdded(header)
|
|
}
|
|
},
|
|
}
|
|
|
|
connConfig := &rpcclient.ConnConfig{
|
|
Host: rpcAddress,
|
|
Endpoint: "ws",
|
|
User: rpcUser,
|
|
Pass: rpcPass,
|
|
DisableTLS: true,
|
|
RequestTimeout: defaultTimeout,
|
|
}
|
|
|
|
var err error
|
|
client.Client, err = rpcclient.New(connConfig, notificationHandlers)
|
|
return client, err
|
|
}
|