Experimenting with decorators.

jwasham · jwasham · commit 61de19b05beb · 2016-08-24T19:00:37.000-07:00
diff --git a/experiments/threads2.py b/experiments/threads2.py
@@ -3,17 +3,26 @@
 import time
 
 
-def count_up(num):
+def synchronize(func):
+    def lock_resource(*args, **kwargs):
+        global lock
+        lock.acquire()
 
-    global database, lock
+        try:
+            func(*args, **kwargs)
+        finally:
+            lock.release()
+
+    return lock_resource
+
+
+@synchronize
+def count_up(num):
 
-    lock.acquire()
+    global database
 
-    try:
-        time.sleep(random.randrange(3))
-        database.append(num)
-    finally:
-        lock.release()
+    time.sleep(random.randrange(3))
+    database.append(num)
 
 
 def main():
