public void SetDistributionStrategy(DistributionStrategyConfiguration configuration, IDistributionStrategy strategy)
{
_distributionStrategy = strategy;
// DataDistribution = _distributionStrategy.GetCurrentBucketDistribution();
}
//IDictionary<string, ClusterInfo> _clusterMetaData = new Dictionary<string, ClusterInfo>();
public void AddDistributionStrategy(string cluster, string database, string collection, IDistributionStrategy strategy, DistributionStrategyConfiguration configuration, Boolean needTransfer)
{
ClusterInfo clusterInfo = (ClusterInfo)_clusterMetaData[cluster];
if (clusterInfo.Databases != null)
{
DatabaseInfo databaseInfo = clusterInfo.GetDatabase(database);
CollectionInfo collectionInfo = databaseInfo.GetCollection(collection);
collectionInfo.SetDistributionStrategy(configuration, strategy);
if (collectionInfo.DistributionStrategy.Name.Equals(DistributionType.NonSharded.ToString()))
{
collectionInfo.DistributionStrategy.AddShard(collectionInfo.CollectionShard, null, needTransfer);
//In case of NonShardedDistributionStrategy: The shard told by user must be added first. That is why this check is placed
}
_clusterMetaData[cluster] = clusterInfo;
_configurationStore.InsertOrUpdateDistributionStrategy(clusterInfo, database, collection);
}
}
