count function calls

zhaoyf3509 · Feb 27, 2023 · 7c5cfe5 · 7c5cfe5
1 parent 7b075cc
commit 7c5cfe5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/db.py b/db.py
@@ -132,6 +132,14 @@ async def _insert_transition(conn: asyncpg.Connection, exe_tx_db_id: int | None,
                             transition_finalize_db_id, str(finalize.record)
                         )
 
+        program_db_id = await conn.fetchval(
+            "SELECT id FROM program WHERE program_id = $1", str(transition.program_id)
+        )
+        await conn.execute(
+            "UPDATE program_function SET called = called + 1 WHERE program_id = $1 AND name = $2",
+            program_db_id, str(transition.function_name)
+        )
+
     async def _save_block(self, block: Block):
         async with self.pool.acquire() as conn:
             async with conn.transaction():