From: erickson <erickson@6d9bc8c9-1ec2-4278-b937-99fde70a366f>
Date: Wed, 24 Feb 2010 02:41:22 +0000 (+0000)
Subject: only update end time on current task set (not all).  log some basic stats after stori... 
X-Git-Url: https://old-git.evergreen-ils.org/?a=commitdiff_plain;h=bc0d960b5becd0c084c6c6dae47c8dc15a33a3d0;p=working%2Frandom.git

only update end time on current task set (not all).  log some basic stats after storing data

git-svn-id: svn://svn.open-ils.org/ILS-Contrib/constrictor/trunk@794 6d9bc8c9-1ec2-4278-b937-99fde70a366f
---

diff --git a/constrictor/data.py b/constrictor/data.py
index 4c0435ef2..6af142ccd 100644
--- a/constrictor/data.py
+++ b/constrictor/data.py
@@ -67,8 +67,9 @@ class Data(object):
  
 
     def create_task_set(self):
+        self.task_set_start_time = time.time()
         res = self.engine.execute(
-            self.task_set_table.insert().values(start_time = time.time())
+            self.task_set_table.insert().values(start_time = self.task_set_start_time)
         )
         #self.task_set_id = res.inserted_primary_key
         self.task_set_id = res.last_inserted_ids()[0]
@@ -76,11 +77,17 @@ class Data(object):
         
     def store_data(self):
 
+        end_time = time.time()
         log.log_info("Inserting data into data store...")
 
         # close out the task set
         self.engine.execute(
-            self.task_set_table.update().values(end_time = time.time())
+            self.task_set_table.update(
+                ).where(
+                    self.task_set_table.c.id == self.task_set_id
+                ).values(
+                    end_time = end_time
+                )
         ).close()
 
         # insert all of the task data
@@ -97,4 +104,10 @@ class Data(object):
                 )
             ).close()
 
+        # log some basic stats
+        task_set_time = end_time - self.task_set_start_time
+        log.log_info(
+            "Total time %0.3f seconds, %d total tasks, average task duration %0.3f seconds" % (
+                task_set_time, len(self.runtime_data), task_set_time / len(self.runtime_data)))
+