From 9a0bf4785deec06cae6cd37b6384f0cbe0e28dfa Mon Sep 17 00:00:00 2001
From: Victor Feyens <vfe@odoo.com>
Date: Tue, 1 Feb 2022 11:45:26 +0000
Subject: [PATCH] [IMP] base: display query counts on profiling records
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

closes odoo/odoo#83746

Signed-off-by: Xavier Dollé (xdo) <xdo@odoo.com>
---
 odoo/addons/base/models/ir_profile.py       |  1 +
 odoo/addons/base/views/ir_profile_views.xml | 11 +++++++----
 odoo/tools/profiler.py                      |  1 +
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/odoo/addons/base/models/ir_profile.py b/odoo/addons/base/models/ir_profile.py
index b9987897405..bb75bef8bae 100644
--- a/odoo/addons/base/models/ir_profile.py
+++ b/odoo/addons/base/models/ir_profile.py
@@ -32,6 +32,7 @@ class IrProfile(models.Model):
     init_stack_trace = fields.Text('Initial stack trace', prefetch=False)
 
     sql = fields.Text('Sql', prefetch=False)
+    sql_count = fields.Integer('Queries Count')
     traces_async = fields.Text('Traces Async', prefetch=False)
     traces_sync = fields.Text('Traces Sync', prefetch=False)
     qweb = fields.Text('Qweb', prefetch=False)
diff --git a/odoo/addons/base/views/ir_profile_views.xml b/odoo/addons/base/views/ir_profile_views.xml
index 775610a58c9..2171f5a2a94 100644
--- a/odoo/addons/base/views/ir_profile_views.xml
+++ b/odoo/addons/base/views/ir_profile_views.xml
@@ -22,6 +22,7 @@
                 <field name="session"/>
                 <field name="name"/>
                 <field name="entry_count"/>
+                <field name="sql_count" optional="hide"/>
                 <field name="speedscope_url" widget="url"/>
                 <field name="duration"/>
             </tree>
@@ -37,9 +38,11 @@
                     <field name="name"/>
                     <field name="session"/>
                     <field name="entry_count"/>
+                    <!-- Do not rely on sql field for the invisible attrs to avoid fetching whole trace from server. -->
+                    <field name="sql_count" attrs="{'invisible': [('sql_count', '=', 0)]}"/>
                     <field name="speedscope_url" widget="url"/>
                 </group>
-                <group attrs="{'invisible': [('qweb','=', '')]}">
+                <group attrs="{'invisible': [('qweb', '=', '')]}">
                     <field name="qweb" widget="profiling_qweb_view" nolabel="1"/>
                 </group>
             </form>
@@ -78,10 +81,10 @@
         <field name="context">{'search_default_group_session': 1}</field>
     </record>
 
-    <menuitem 
+    <menuitem
         name="Profiling"
-        action="action_menu_ir_profile" 
-        id="menu_ir_profile" 
+        action="action_menu_ir_profile"
+        id="menu_ir_profile"
         parent="base.next_id_9"/>
 
 </odoo>
diff --git a/odoo/tools/profiler.py b/odoo/tools/profiler.py
index 85016b86cba..79673683fe7 100644
--- a/odoo/tools/profiler.py
+++ b/odoo/tools/profiler.py
@@ -586,6 +586,7 @@ class Profiler:
                         "init_stack_trace": json.dumps(_format_stack(self.init_stack_trace)),
                         "duration": self.duration,
                         "entry_count": self.entry_count(),
+                        "sql_count": sum(len(collector.entries) for collector in self.collectors if collector.name == 'sql')
                     }
                     for collector in self.collectors:
                         if collector.entries: