ndb: fix source view counting

C2Devel · Dec 24, 2023 · bfc37df · bfc37df
1 parent 020b793
commit bfc37df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pyroute2/ndb/source.py b/pyroute2/ndb/source.py
@@ -180,6 +180,12 @@ def __init__(self, ndb, **spec):
         self.ndb.task_manager.db_add_nl_source(self.target, self.kind, spec)
         self.load_sql()
 
+    @classmethod
+    def _count(cls, view):
+        return view.ndb.task_manager.db_fetchone(
+            "SELECT count(*) FROM %s" % view.table,
+        )
+
     @property
     def must_restart(self):
         if self.max_errors < 0 or self.errors_counter <= self.max_errors: