(#1919) Migrate to AddWatchedAddresses (#1920)

* Migrate to AddWatchedAddresses
* Use variadic parameters
* Vendored fixes
* Re-watch addresses on socket restart
* Convert Mutex to RWMutex
* Protect with listenLock
* Update to use bulk call
* Eliminate redundant conversion
* Move statements into the socketclient section
* Remove duplicate lock
* Refactored to be closer to it’s usage
* DB Transaction for Putting PubKeys
* Fix linter errors
* Migrate to PutAll for watched scripts
* Fix missing initialization of err
* Interface check for implemented wallets
* Handle errors from transaction; Report Rollback failures
* Append to BlockBookClient.listenAddrs instead of replace
* Fix race bug and remove redundant replay handling in c/b/client.go

Co-authored-by: Mike Greenberg <[email protected]>

Author: hoffmabc

core/images.go

@@ -165,8 +165,7 @@ func (n *OpenBazaarNode) FetchImage(peerID string, imageType string, size string
 func (n *OpenBazaarNode) GetBase64Image(url string) (base64ImageData, filename string, err error) {
 	var client *http.Client
 	if n.TorDialer != nil {
-		dial := n.TorDialer.Dial
-		tbTransport := &http.Transport{Dial: dial}
+		tbTransport := &http.Transport{Dial: n.TorDialer.Dial}
 		client = &http.Client{Transport: tbTransport, Timeout: time.Second * 30}
 	} else {
 		client = &http.Client{Timeout: time.Second * 30}

core/order.go

@@ -271,7 +271,7 @@ func prepareModeratedOrderContract(data *PurchaseData, n *OpenBazaarNode, contra
 	payment.Chaincode = hex.EncodeToString(chaincode)
 	contract.BuyerOrder.RefundFee = wal.GetFeePerByte(wallet.NORMAL)
 
-	err = wal.AddWatchedAddress(addr)
+	err = wal.AddWatchedAddresses(addr)
 	if err != nil {
 		return nil, err
 	}
@@ -308,7 +308,7 @@ func processOnlineDirectOrder(resp *pb.Message, n *OpenBazaarNode, wal wallet.Wa
 	if err != nil {
 		return "", "", 0, false, err
 	}
-	err = wal.AddWatchedAddress(addr)
+	err = wal.AddWatchedAddresses(addr)
 	if err != nil {
 		return "", "", 0, false, err
 	}
@@ -356,7 +356,7 @@ func processOfflineDirectOrder(n *OpenBazaarNode, wal wallet.Wallet, contract *p
 	payment.RedeemScript = hex.EncodeToString(redeemScript)
 	payment.Chaincode = hex.EncodeToString(chaincode)
 
-	err = wal.AddWatchedAddress(addr)
+	err = wal.AddWatchedAddresses(addr)
 	if err != nil {
 		return "", "", 0, false, err
 	}

net/retriever/retriever.go

@@ -69,8 +69,7 @@ type offlineMessage struct {
 func NewMessageRetriever(cfg MRConfig) *MessageRetriever {
 	var client *http.Client
 	if cfg.Dialer != nil {
-		dial := cfg.Dialer.Dial
-		tbTransport := &http.Transport{Dial: dial}
+		tbTransport := &http.Transport{Dial: cfg.Dialer.Dial}
 		client = &http.Client{Transport: tbTransport, Timeout: time.Second * 30}
 	} else {
 		client = &http.Client{Timeout: time.Second * 30}

net/service/handlers.go

@@ -371,7 +371,7 @@ func (service *OpenBazaarService) handleOrder(peer peer.ID, pmes *pb.Message, op
 		if err != nil {
 			return errorResponse(err.Error()), err
 		}
-		wal.AddWatchedAddress(addr)
+		wal.AddWatchedAddresses(addr)
 		service.node.Datastore.Sales().Put(orderId, *contract, pb.OrderState_AWAITING_PAYMENT, false)
 		log.Debugf("Received direct ORDER message from %s", peer.Pretty())
 		return nil, nil
@@ -395,7 +395,7 @@ func (service *OpenBazaarService) handleOrder(peer peer.ID, pmes *pb.Message, op
 		if err != nil {
 			return errorResponse(err.Error()), err
 		}
-		wal.AddWatchedAddress(addr)
+		wal.AddWatchedAddresses(addr)
 		contract, err = service.node.NewOrderConfirmation(contract, false, false)
 		if err != nil {
 			return errorResponse("Error building order confirmation"), errors.New("error building order confirmation")
@@ -427,7 +427,7 @@ func (service *OpenBazaarService) handleOrder(peer peer.ID, pmes *pb.Message, op
 			log.Error(err)
 			return errorResponse(err.Error()), err
 		}
-		wal.AddWatchedAddress(addr)
+		wal.AddWatchedAddresses(addr)
 		log.Debugf("Received offline moderated ORDER message from %s", peer.Pretty())
 		service.node.Datastore.Sales().Put(orderId, *contract, pb.OrderState_AWAITING_PAYMENT, false)
 		return nil, nil

repo/db/watched_scripts.go

@@ -3,6 +3,7 @@ package db
 import (
 	"database/sql"
 	"encoding/hex"
+	"fmt"
 	"sync"
 
 	"github.com/OpenBazaar/openbazaar-go/repo"
@@ -18,6 +19,37 @@ func NewWatchedScriptStore(db *sql.DB, lock *sync.Mutex, coinType wallet.CoinTyp
 	return &WatchedScriptsDB{modelStore{db, lock}, coinType}
 }
 
+func (w *WatchedScriptsDB) PutAll(scriptPubKeys [][]byte) error {
+	w.lock.Lock()
+	defer w.lock.Unlock()
+
+	tx, err := w.db.Begin()
+	if err != nil {
+		return err
+	}
+
+	stmt, err := tx.Prepare("insert or replace into watchedscripts(coin, scriptPubKey) values(?,?)")
+	if err != nil {
+		if rErr := tx.Rollback(); rErr != nil {
+			return fmt.Errorf("put AND rollback failed: %s (rollback error: %s)", err.Error(), rErr.Error())
+		}
+		return err
+	}
+	defer stmt.Close()
+
+	for _, scriptPubKey := range scriptPubKeys {
+		_, err = stmt.Exec(w.coinType.CurrencyCode(), hex.EncodeToString(scriptPubKey))
+		if err != nil {
+			if rErr := tx.Rollback(); rErr != nil {
+				return fmt.Errorf("put AND rollback failed: %s (rollback error: %s)", err.Error(), rErr.Error())
+			}
+			return err
+		}
+	}
+
+	return tx.Commit()
+}
+
 func (w *WatchedScriptsDB) Put(scriptPubKey []byte) error {
 	w.lock.Lock()
 	defer w.lock.Unlock()