Add more timings to help debug

TulipaEnergy · Nov 4, 2024 · 34ba4a1 · 34ba4a1
1 parent 1984aa7
commit 34ba4a1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/io.jl b/src/io.jl
@@ -273,7 +273,7 @@ function create_internal_structures(connection)
 
     graph = MetaGraphsNext.MetaGraph(_graph, asset_data, flow_data, nothing, nothing, nothing)
 
-    tmp_create_partition_tables(connection)
+    @timeit to "Create partition tables" tmp_create_partition_tables(connection)
     df = TulipaIO.get_table(connection, "asset_time_resolution")
     gdf = DataFrames.groupby(df, [:asset, :year, :rep_period])
     for ((a, year, rp), _df) in pairs(gdf)

diff --git a/src/model-preparation.jl b/src/model-preparation.jl
@@ -60,7 +60,7 @@ function construct_dataframes(
         dataframes[key] = df
     end
 
-    tmp_create_constraints_indices(connection)
+    @timeit to "Create constraints indices" tmp_create_constraints_indices(connection)
 
     # WIP: Can these queries be left undordered by the end of the refactor?
     # WIP: highest_in_out is not included in constraints_partition anymore

diff --git a/src/tmp.jl b/src/tmp.jl
@@ -319,7 +319,7 @@ function tmp_example_of_flow_expression_problem()
 end
 
 function tmp_create_expressions(connection)
-    DuckDB.execute(
+    @timeit to "Create t_incoming_nonzero" DuckDB.execute(
         connection,
         "CREATE OR REPLACE TABLE t_incoming_nonzero AS
             SELECT
@@ -340,7 +340,7 @@ function tmp_create_expressions(connection)
             WHERE duration > 0",
     )
 
-    DuckDB.execute(
+    @timeit to "Create t_outgoing_nonzero" DuckDB.execute(
         connection,
         "CREATE OR REPLACE TABLE t_outgoing_nonzero AS
             SELECT
@@ -361,7 +361,7 @@ function tmp_create_expressions(connection)
             WHERE duration > 0",
     )
 
-    DuckDB.execute(
+    @timeit to "Create highest_in_out_incoming" DuckDB.execute(
         connection,
         "CREATE OR REPLACE TABLE highest_in_out_incoming AS
         SELECT
@@ -387,7 +387,7 @@ function tmp_create_expressions(connection)
         ",
     )
 
-    DuckDB.execute(
+    @timeit to "Create highest_in_out_outgoing" DuckDB.execute(
         connection,
         "CREATE OR REPLACE TABLE highest_in_out_outgoing AS
         SELECT