From 5ba03a1c8640b8a8cef79a19174679a51aefb768 Mon Sep 17 00:00:00 2001
From: spla <salvador.pla@mastodont.cat>
Date: Mon, 14 Mar 2022 12:48:10 +0100
Subject: [PATCH] Save every program execution time to new table
 'execution_time'

---
 fediverse.py    | 46 +++++++++++++++++++++++++++++++++++++++++--
 fetchservers.py | 52 ++++++++++++++++++++++++++++++++++++++++++++-----
 2 files changed, 91 insertions(+), 7 deletions(-)

diff --git a/fediverse.py b/fediverse.py
index 228bf21..e7efbc0 100644
--- a/fediverse.py
+++ b/fediverse.py
@@ -445,6 +445,36 @@ def usage():
     print('usage: python ' + sys.argv[0] + ' --multi' + ' (multiprocessing, fast)')
     print('usage: python ' + sys.argv[0] + ' --mono' + ' (one process, slow)')
 
+def save_time(program, start, finish):
+
+    insert_sql = "INSERT INTO execution_time(program, start, finish) VALUES(%s,%s,%s) ON CONFLICT DO NOTHING"
+
+    conn = None
+
+    try:
+
+        conn = psycopg2.connect(database = fediverse_db, user = fediverse_db_user, password = "", host = "/var/run/postgresql", port = "5432")
+
+        cur = conn.cursor()
+
+        cur.execute(insert_sql, (program, start, finish,))
+
+        cur.execute("UPDATE execution_time SET start=(%s), finish=(%s) where program=(%s)", (start, finish, program))
+
+        conn.commit()
+
+        cur.close()
+
+    except (Exception, psycopg2.DatabaseError) as error:
+
+        print(error)
+
+    finally:
+
+        if conn is not None:
+
+            conn.close()
+
 def get_parameter(parameter, file_path):
 
     # Check if secrets file exists
@@ -474,6 +504,14 @@ if __name__ == '__main__':
 
         if sys.argv[1] == '--multi':
 
+            start = datetime.now()
+
+            program = 'fediverse'
+
+            finish = start
+
+            save_time(program, start, finish)
+
             now = datetime.now()
 
             mastodon, mastodon_hostname = mastodon()
@@ -492,11 +530,11 @@ if __name__ == '__main__':
 
             getservers.now = datetime.now()
 
-            start = time.time()
+            ray_start = time.time()
 
             results = ray.get([getservers.get_alive_servers.remote(server) for server in alive_servers])
 
-            print(f"duration = {time.time() - start}.\nprocessed servers: {len(results)}")
+            print(f"duration = {time.time() - ray_start}.\nprocessed servers: {len(results)}")
 
             ###########################################################################
             # get current total servers and users, get users from every software
@@ -838,6 +876,10 @@ if __name__ == '__main__':
 
             delete_dead_servers()
 
+            finish = datetime.now()
+
+            save_time(program, start, finish)
+
     else:
 
         usage()
diff --git a/fetchservers.py b/fetchservers.py
index 15d347f..a8e3692 100644
--- a/fetchservers.py
+++ b/fetchservers.py
@@ -340,6 +340,36 @@ def get_world_servers():
 
     return world_servers
 
+def save_time(program, start, finish):
+
+    insert_sql = "INSERT INTO execution_time(program, start, finish) VALUES(%s,%s,%s) ON CONFLICT DO NOTHING"
+
+    conn = None
+
+    try:
+
+        conn = psycopg2.connect(database = fediverse_db, user = fediverse_db_user, password = "", host = "/var/run/postgresql", port = "5432")
+
+        cur = conn.cursor()
+
+        cur.execute(insert_sql, (program, start, finish,))
+
+        cur.execute("UPDATE execution_time SET start=(%s), finish=(%s) where program=(%s)", (start, finish, program))
+
+        conn.commit()
+
+        cur.close()
+
+    except (Exception, psycopg2.DatabaseError) as error:
+
+        print(error)
+
+    finally:
+
+        if conn is not None:
+
+            conn.close()
+
 def get_parameter(parameter, file_path):
 
     # Check if secrets file exists
@@ -379,16 +409,28 @@ if __name__ == '__main__':
 
     ## name: fetchservers.py
 
-    now = datetime.now()
+    fediverse_db, fediverse_db_user = get_db_config()
+
+    start = datetime.now()
+
+    program = 'fetchservers'
+
+    finish = start
+
+    save_time(program, start, finish)
+
+    now = start
 
     mastodon_hostname = get_config()
 
-    fediverse_db, fediverse_db_user = get_db_config()
-
     world_servers = get_world_servers()
 
-    start = time.time()
+    ray_start = time.time()
 
     results = ray.get([getsoft.remote(server) for server in world_servers])
 
-    print(f"duration = {time.time() - start}.\nprocessed servers: {len(results)}")
+    print(f"duration = {time.time() - ray_start}.\nprocessed servers: {len(results)}")
+
+    finish = datetime.now()
+
+    save_time(program, start, finish)