Add support for non-default compare template argument in std::map wrappers

1 files changed, 5 insertions, 5 deletions

diff --git a/Lib/go/std_map.i b/Lib/go/std_map.i
index 84b0c74ff..f514cf743 100644
--- a/Lib/go/std_map.i
+++ b/Lib/go/std_map.i
@@ -20,7 +20,7 @@
 
 namespace std {
 
-    template<class K, class T> class map {
+    template<class K, class T, class C = std::less<K> > class map {
         // add typemaps here
       public:
         typedef size_t size_type;
@@ -28,14 +28,14 @@ namespace std {
         typedef K key_type;
         typedef T mapped_type;
         map();
-        map(const map<K,T> &);
+        map(const map< K, T, C > &);
         
         unsigned int size() const;
         bool empty() const;
         void clear();
         %extend {
             const T& get(const K& key) throw (std::out_of_range) {
-                std::map<K,T >::iterator i = self->find(key);
+                std::map< K, T, C >::iterator i = self->find(key);
                 if (i != self->end())
                     return i->second;
                 else
@@ -45,14 +45,14 @@ namespace std {
                 (*self)[key] = x;
             }
             void del(const K& key) throw (std::out_of_range) {
-                std::map<K,T >::iterator i = self->find(key);
+                std::map< K, T, C >::iterator i = self->find(key);
                 if (i != self->end())
                     self->erase(i);
                 else
                     throw std::out_of_range("key not found");
             }
             bool has_key(const K& key) {
-                std::map<K,T >::iterator i = self->find(key);
+                std::map< K, T, C >::iterator i = self->find(key);
                 return i != self->end();
             }
         }