qt: Add PeerTableModel::StatsRole

This change allows access to CNodeCombinedStats instance directly from any view object.
PastaPastaPasta · Dec 30, 2020 · 35007ed · 35007ed
1 parent f1f26b8
commit 35007ed
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/qt/peertablemodel.cpp b/src/qt/peertablemodel.cpp
@@ -181,6 +181,11 @@ QVariant PeerTableModel::data(const QModelIndex &index, int role) const
  default:
  return QVariant();
  }
+ } else if (role == StatsRole) {
+ switch (index.column()) {
+ case NetNodeId: return QVariant::fromValue(rec);
+ default: return QVariant();
+ }
  }
 
  return QVariant();

diff --git a/src/qt/peertablemodel.h b/src/qt/peertablemodel.h
@@ -28,6 +28,7 @@ struct CNodeCombinedStats {
  CNodeStateStats nodeStateStats;
  bool fNodeStateStatsAvailable;
 };
+Q_DECLARE_METATYPE(CNodeCombinedStats*)
 
 class NodeLessThan
 {
@@ -67,6 +68,10 @@ class PeerTableModel : public QAbstractTableModel
  Subversion = 6
  };
 
+ enum {
+ StatsRole = Qt::UserRole,
+ };
+
  /** @name Methods overridden from QAbstractTableModel
  @{*/
  int rowCount(const QModelIndex &parent) const override;