diff --git a/mROA/Implementation/Backend/HubRequestExtractor.cs b/mROA/Implementation/Backend/HubRequestExtractor.cs index c4438ab..7d20a08 100644 --- a/mROA/Implementation/Backend/HubRequestExtractor.cs +++ b/mROA/Implementation/Backend/HubRequestExtractor.cs @@ -10,17 +10,18 @@ namespace mROA.Implementation.Backend private readonly IRealStoreInstanceRepository _contextRepository; private readonly IInstanceRepository _remoteContextRepository; private readonly IExecuteModule _executeModule; - private readonly DistributionOptions _mode; + private readonly IDistributionModule _distribution; private readonly Dictionary _producedExtractors = new(); public HubRequestExtractor(IRealStoreInstanceRepository contextRepository, IInstanceRepository remoteContextRepository, IExecuteModule executeModule, - IOptions mode) + IDistributionModule distribution) { _contextRepository = contextRepository; _remoteContextRepository = remoteContextRepository; _executeModule = executeModule; - _mode = mode.Value; + _distribution = distribution; + } public IRequestExtractor this[int id] => _producedExtractors[id]; @@ -28,7 +29,7 @@ namespace mROA.Implementation.Backend public IRequestExtractor HubOnOnConnected(IRepresentationModule representationModule) { var extractor = CreateExtractor(representationModule); - if (_mode.DistributionType == EDistributionType.Channeled) + if (_distribution is ChannelDistributionModule) { extractor.StartExtraction().ContinueWith(_ => OnDisconnected(representationModule)); } diff --git a/mROA/Implementation/DistributionOptions.cs b/mROA/Implementation/DistributionOptions.cs deleted file mode 100644 index 239dc23..0000000 --- a/mROA/Implementation/DistributionOptions.cs +++ /dev/null @@ -1,13 +0,0 @@ -namespace mROA.Implementation -{ - public class DistributionOptions - { - public EDistributionType DistributionType { get; set; } - } - - public enum EDistributionType - { - Channeled, - ExtractorFirst - } -} \ No newline at end of file