Ticket #198: type_map_cache.diff

pyamf/__init__.py

@@ -537 +537 @@
             raise TypeError, "context must be of type %s.%s" % (
                 self.context_class.__module__, self.context_class.__name__)
 
-    def _writeElementFunc(self, data):
+        self._write_elem_func_cache = {}
+
+    def _getWriteElementFunc(self, data):
         """
         Gets a function used to encode the data.
 
@@ -565 +567 @@
 
         return None
 
+    def _writeElementFunc(self, data):
+        """
+        Gets a function used to encode the data.
+
+        @type   data: C{mixed}
+        @param  data: Python data.
+        @rtype: callable or C{None}.
+        @return: The function used to encode data to the stream.
+        """
+        key = data.__class__
+        if key not in self._write_elem_func_cache:
+            self._write_elem_func_cache[key] =  self._getWriteElementFunc(data)
+        return self._write_elem_func_cache[key]
+
     def writeElement(self, data):
         """
         Writes the data.