stores: add wallet metric

api/metrcis.go

@@ -13,6 +13,7 @@ const (
 	MetricContractSet      = "contractset"
 	MetricContractSetChurn = "churn"
 	MetricContract         = "contract"
+	MetricWallet           = "wallet"
 )
 
 type (
@@ -75,6 +76,20 @@ type (
 		ContractID types.FileContractID
 		HostKey    types.PublicKey
 	}
+
+	WalletMetric struct {
+		Timestamp time.Time `json:"timestamp"`
+
+		Address types.Address `json:"address"`
+
+		ConfirmedBalance   types.Currency `json:"confirmedBalance"`
+		SpendableBalance   types.Currency `json:"spendableBalance"`
+		UnconfirmedBalance types.Currency `json:"unconfirmedBalance"`
+	}
+
+	WalletMetricsQueryOpts struct {
+		Address types.Address
+	}
 )
 
 type (

bus/bus.go

@@ -201,6 +201,8 @@ type (
 
 		ContractSetChurnMetrics(ctx context.Context, start time.Time, n uint64, interval time.Duration, opts api.ContractSetChurnMetricsQueryOpts) ([]api.ContractSetChurnMetric, error)
 		RecordContractSetChurnMetric(ctx context.Context, metrics ...api.ContractSetChurnMetric) error
+
+		WalletMetrics(ctx context.Context, start time.Time, n uint64, interval time.Duration, opts api.WalletMetricsQueryOpts) ([]api.WalletMetric, error)
 	}
 )
 
@@ -2059,6 +2061,16 @@ func (b *bus) metricsHandlerGET(jc jape.Context) {
 			jc.Encode(metrics)
 			return
 		}
+	case api.MetricWallet:
+		var opts api.WalletMetricsQueryOpts
+		if jc.DecodeForm("address", &opts.Address) != nil {
+			return
+		} else if metrics, err := b.metrics(jc.Request.Context(), key, start, n, interval, opts); jc.Check("failed to get wallet metrics", err) != nil {
+			return
+		} else {
+			jc.Encode(metrics)
+			return
+		}
 	default:
 		jc.Error(fmt.Errorf("unknown metric '%s'", key), http.StatusBadRequest)
 		return
@@ -2073,6 +2085,8 @@ func (b *bus) metrics(ctx context.Context, key string, start time.Time, n uint64
 		return b.mtrcs.ContractSetMetrics(ctx, start, n, interval, opts.(api.ContractSetMetricsQueryOpts))
 	case api.MetricContractSetChurn:
 		return b.mtrcs.ContractSetChurnMetrics(ctx, start, n, interval, opts.(api.ContractSetChurnMetricsQueryOpts))
+	case api.MetricWallet:
+		return b.mtrcs.WalletMetrics(ctx, start, n, interval, opts.(api.WalletMetricsQueryOpts))
 	}
 	return nil, nil
 }

internal/node/node.go

@@ -140,6 +140,9 @@ func NewBus(cfg BusConfig, dir string, seed types.PrivateKey, l *zap.Logger) (ht
 
 	w := wallet.NewSingleAddressWallet(seed, sqlStore, cfg.UsedUTXOExpiry, zap.NewNop().Sugar())
 	tp.TransactionPoolSubscribe(w)
+	if err := cs.ConsensusSetSubscribe(w, modules.ConsensusChangeRecent, nil); err != nil {
+		return nil, nil, err
+	}
 
 	if m := cfg.Miner; m != nil {
 		if err := cs.ConsensusSetSubscribe(m, ccid, nil); err != nil {

stores/hostdb_test.go

@@ -1104,7 +1104,7 @@ func (s *SQLStore) addCustomTestHost(hk types.PublicKey, na string) error {
 		announcement: hostdb.Announcement{NetAddress: na},
 	}}...)
 	s.lastSave = time.Now().Add(s.persistInterval * -2)
-	return s.applyUpdates(false)
+	return s.applyUpdates(false, false)
 }
 
 // hosts returns all hosts in the db. Only used in testing since preloading all