some refactoring for SERVER-946

1 files changed, 19 insertions, 3 deletions

diff --git a/s/shard.h b/s/shard.h
index 19df0cd2848..a4571f8e90c 100644
--- a/s/shard.h
+++ b/s/shard.h
@@ -24,7 +24,8 @@
 namespace mongo {
 
     class ShardConnection;
-    
+    class ShardStatus;
+
     class Shard {
     public:
         Shard()
@@ -108,8 +109,17 @@ namespace mongo {
             return runCommand( db , BSON( simple << 1 ) );
         }
         BSONObj runCommand( const string& db , const BSONObj& cmd );
-
-        static void getAllShards( list<Shard>& all );
+        
+        ShardStatus getStatus() const;
+        
+        static void getAllShards( vector<Shard>& all );
+        
+        /**
+         * picks a Shard for more load
+         */
+        static Shard pick();
+        
+        static void reloadShardInfo();
 
         static Shard EMPTY;
 
@@ -118,6 +128,12 @@ namespace mongo {
         string _addr;
     };
 
+    class ShardStatus {
+    private:
+        long long _mapped;
+        double _writeLock;
+    };
+
     class ShardConnection {
     public:
         ShardConnection( const Shard * s )