LBRYCommunity/lbcd
1
0
Fork 0
Code Issues 17 Pull requests 4 Projects Releases 28 Packages Wiki Activity

Add support for new rescanfinished notification.

Browse source 
ok @davecgh
This commit is contained in:
Josh Rickmar 2014-06-16 09:42:29 -05:00
parent 7448f9555c
commit aceaed82c5
1 changed files with 28 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
notify.go
View file

@ -122,6 +122,13 @@ type NotificationHandlers struct {
// this to invoked indirectly as the result of a NotifyReceived call. // this to invoked indirectly as the result of a NotifyReceived call.
OnRedeemingTx func(transaction *btcutil.Tx, details *btcws.BlockDetails) OnRedeemingTx func(transaction *btcutil.Tx, details *btcws.BlockDetails)
// OnRescanFinished is invoked after a rescan finishes due to a previous
// call to Rescan or RescanEndHeight. Finished rescans should be
// signaled on this notification, rather than relying on the return
// result of a rescan request, due to how btcd may send various rescan
// notifications after the rescan request has already returned.
OnRescanFinished func(lastProcessedHeight int32)
// OnRescanProgress is invoked periodically when a rescan is underway. // OnRescanProgress is invoked periodically when a rescan is underway.
// It will only be invoked if a preceding call to Rescan or // It will only be invoked if a preceding call to Rescan or
// RescanEndHeight has been made and the function is non-nil. // RescanEndHeight has been made and the function is non-nil.
@ -246,6 +253,23 @@ func (c *Client) handleNotification(ntfn *rawNotification) {
c.ntfnHandlers.OnRedeemingTx(tx, block) c.ntfnHandlers.OnRedeemingTx(tx, block)
// OnRescanFinished
case btcws.RescanFinishedNtfnMethod:
// Ignore the notification is the client is not interested in
// it.
if c.ntfnHandlers.OnRescanFinished == nil {
return
}
lastProcessed, err := parseRescanHeightParams(ntfn.Params)
if err != nil {
log.Warnf("Received invalid rescanfinished "+
"notification: %v", err)
return
}
c.ntfnHandlers.OnRescanFinished(lastProcessed)
// OnRescanProgress // OnRescanProgress
case btcws.RescanProgressNtfnMethod: case btcws.RescanProgressNtfnMethod:
// Ignore the notification is the client is not interested in // Ignore the notification is the client is not interested in
@ -254,7 +278,7 @@ func (c *Client) handleNotification(ntfn *rawNotification) {
return return
} }
lastProcessed, err := parseRescanProgressNtfnParams(ntfn.Params) lastProcessed, err := parseRescanHeightParams(ntfn.Params)
if err != nil { if err != nil {
log.Warnf("Received invalid rescanprogress "+ log.Warnf("Received invalid rescanprogress "+
"notification: %v", err) "notification: %v", err)
@ -447,9 +471,9 @@ func parseChainTxNtfnParams(params []json.RawMessage) (*btcutil.Tx,
return btcutil.NewTx(&msgTx), block, nil return btcutil.NewTx(&msgTx), block, nil
} }
// parseRescanProgressNtfnParams parses out the height of the last rescanned // parseRescanHeightParams parses out the height of the last rescanned block
// from the parameters of a rescanprogress notification. // from the parameters of rescanfinished and rescanprogress notifications.
func parseRescanProgressNtfnParams(params []json.RawMessage) (int32, error) { func parseRescanHeightParams(params []json.RawMessage) (int32, error) {
if len(params) != 1 { if len(params) != 1 {
return 0, wrongNumParams(len(params)) return 0, wrongNumParams(len(params))
} }