mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-03-30 15:08:33 +00:00
a7299c1b87
* Add stability-tests * Fix requires * Fix golint errors * Update README.md * Remove payloadHash from everywhere * don't run vet on kaspad in stability-tests/install_and_test
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/kaspanet/kaspad/app/protocol/common"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func sendMessages(address string, messagesChan <-chan []byte) error {
|
|
connection, err := dialToNode(address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for message := range messagesChan {
|
|
messageHex := make([]byte, hex.EncodedLen(len(message)))
|
|
hex.Encode(messageHex, message)
|
|
log.Infof("Sending message %s", messageHex)
|
|
|
|
err := sendMessage(connection, message)
|
|
if err != nil {
|
|
// if failed once, we might have been disconnected because of a previous message,
|
|
// so re-connect and retry before reporting error
|
|
connection, err = dialToNode(address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = sendMessage(connection, message)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func sendMessage(connection net.Conn, message []byte) error {
|
|
err := connection.SetDeadline(time.Now().Add(common.DefaultTimeout))
|
|
if err != nil {
|
|
return errors.Wrap(err, "Error setting connection deadline")
|
|
}
|
|
|
|
_, err = connection.Write(message)
|
|
return err
|
|
}
|
|
|
|
func dialToNode(address string) (net.Conn, error) {
|
|
connection, err := net.Dial("tcp", address)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "Error connecting to node")
|
|
}
|
|
return connection, nil
|
|
}
|