From d8bcfc55a937249b6782e40d92340586ae710c47 Mon Sep 17 00:00:00 2001
From: Eliot Horowitz <eliot@10gen.com>
Date: Wed, 3 Nov 2010 18:37:31 -0400
Subject: --discover works for mongos now

---
 tools/stat.cpp | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

(limited to 'tools')

diff --git a/tools/stat.cpp b/tools/stat.cpp
index fd5b9d18319..af0a23c4856 100644
--- a/tools/stat.cpp
+++ b/tools/stat.cpp
@@ -569,6 +569,40 @@ namespace mongo {
                         biggest = rows[i].data;
                 }
                 
+                { // check for any headers not in biggest
+
+                    // TODO: we put any new headers at end, 
+                    //       ideally we would interleave
+
+                    set<string> seen;
+                    
+                    BSONObjBuilder b;
+                    
+                    { // iterate biggest
+                        BSONObjIterator i( biggest );
+                        while ( i.more() ){
+                            BSONElement e = i.next();
+                            seen.insert( e.fieldName() );
+                            b.append( e );
+                        }
+                    }
+                    
+                    // now do the rest
+                    for ( unsigned j=0; j<rows.size(); j++ ){
+                        BSONObjIterator i( rows[j].data );
+                        while ( i.more() ){
+                            BSONElement e = i.next();
+                            if ( seen.count( e.fieldName() ) )
+                                continue;
+                            seen.insert( e.fieldName() );
+                            b.append( e );
+                        }
+
+                    }
+
+                    biggest = b.obj();
+                    
+                }
                 
                 // display data
                 
-- 
cgit v1.2.1