65fab01225
This removes all of the global loggers from the project and replaces them with loggers which are instantiated in vspd.go and passed down as params. To support this, the background package was removed. It only contained one file so it was a bit pointless anyway. It only existed so background tasks could have their own named logger.
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package rpc
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/decred/dcrd/wire"
|
|
"github.com/decred/slog"
|
|
)
|
|
|
|
type blockConnectedHandler struct {
|
|
blockConnected chan *wire.BlockHeader
|
|
log slog.Logger
|
|
}
|
|
|
|
// Notify is called every time a notification is received from dcrd client.
|
|
// A wsrpc.Client will never call Notify concurrently. Notify should not return
|
|
// an error because that will cause the client to close and no further
|
|
// notifications will be received until a new connection is established.
|
|
func (n *blockConnectedHandler) Notify(method string, msg json.RawMessage) error {
|
|
if method != "blockconnected" {
|
|
return nil
|
|
}
|
|
|
|
header, err := parseBlockConnected(msg)
|
|
if err != nil {
|
|
n.log.Errorf("Failed to parse dcrd block notification: %v", err)
|
|
return nil
|
|
}
|
|
|
|
n.blockConnected <- header
|
|
|
|
return nil
|
|
}
|
|
|
|
func (n *blockConnectedHandler) Close() error {
|
|
return nil
|
|
}
|
|
|
|
// parseBlockConnected extracts the block header from a
|
|
// blockconnected JSON-RPC notification.
|
|
func parseBlockConnected(msg json.RawMessage) (*wire.BlockHeader, error) {
|
|
var notif []string
|
|
err := json.Unmarshal(msg, ¬if)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("json unmarshal error: %w", err)
|
|
}
|
|
|
|
if len(notif) == 0 {
|
|
return nil, errors.New("notification is empty")
|
|
}
|
|
|
|
var header wire.BlockHeader
|
|
err = header.Deserialize(hex.NewDecoder(bytes.NewReader([]byte(notif[0]))))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating block header from bytes: %w", err)
|
|
}
|
|
|
|
return &header, nil
|
|
}
|