fix projecton generation for export since it uses getFieldDotted SERVER-2369

1 files changed, 18 insertions, 3 deletions

diff --git a/tools/export.cpp b/tools/export.cpp
index b1b0f5f1ac9..0262c4bf174 100644
--- a/tools/export.cpp
+++ b/tools/export.cpp
@@ -82,10 +82,25 @@ public:
 
         if ( hasParam( "fields" ) || csv ) {
             needFields();
-            fieldsToReturn = &_fieldsObj;
+            
+            // we can't use just _fieldsObj since we support everything getFieldDotted does
+            
+            set<string> seen;
+            BSONObjBuilder b;
+            
+            BSONObjIterator i( _fieldsObj );
+            while ( i.more() ){
+                BSONElement e = i.next();
+                string f = str::before( e.fieldName() , '.' );
+                if ( seen.insert( f ).second )
+                    b.append( f , 1 );
+            }
+            
+            realFieldsToReturn = b.obj();
+            fieldsToReturn = &realFieldsToReturn;
         }
-
-
+        
+        
         if ( csv && _fields.size() == 0 ) {
             cerr << "csv mode requires a field list" << endl;
             return -1;