Changeset ef252e9

Timestamp:

Feb 28, 2010 12:24:51 PM (14 years ago)

Author:

Ted Faber <faber@…>

Branches:

axis_example, compt_changes, info-ops, master, version-3.01, version-3.02

Children:

2761484

Parents:

358e0b8

Message:

Add support for loading and saving the synch points (pickle didn't work after
all) and enforce space free keys. That requirement is because of the very
simple file format.

File:

• fedd/federation/synch_store.py (modified) (3 diffs)

fedd/federation/synch_store.py

@@ -16 +16 @@
     # Thrown if a non-existant key is deleted
     class BadDeletionError(Exception): pass
+
+    # Thrown if a key is specified with a white space
+    class BadKeyError(Exception): pass
 
     # A value in the dict.  Each has its own condition (and lock) for threads
@@ -37 +40 @@
     # Set the value in the dict.  If threads are waiting, wake them.
     def set_value(self, key, value):
+        if key.find(' ') != -1:
+            raise self.BadKeyError("Whitespace is not allowed in keys")
+
         self.main_lock.acquire()
         if self.values.has_key(key):
@@ -88 +94 @@
             self.main_lock.release()
             raise synch_store.BadDeletionError("%s does not exist yet" % key)
+
+    def all_keys(self):
+        self.main_lock.acquire()
+        rv = [ k for k in self.values.keys()]
+        self.main_lock.release()
+        return rv
+
+    # Save the store contents to a file in a simple way
+    def save(self, filename):
+        self.main_lock.acquire()
+        f = open(filename, 'w')
+        for k, v in self.values.items():
+            self.values[k].condition.acquire()
+            if self.values[k].value is not None:
+                print >> f, "%s %s" % ( k, str(self.values[k].value))
+            else:
+                print >> f, "%s" % k
+            self.values[k].condition.release()
+        self.main_lock.release()
+        f.close()
+
+    # Load the store contents from a file
+    def load(self, filename):
+        self.main_lock.acquire()
+        self.values = { }
+        f = open(filename, 'r')
+        for l in f:
+            i = l.find(' ')
+            if i != -1: self.values[l[:i]] = synch_store.synch_value(l[i+1:])
+            else: self.values[l] = synch_store.synch_value()
+        f.close()
+        self.main_lock.release()
 
     # A debugging representation