package com.hubspot.singularity.data.zkmigrations;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.hubspot.singularity.config.SingularityConfiguration;
import com.hubspot.singularity.data.ShuffleConfigurationManager;
import org.apache.curator.framework.CuratorFramework;

@Singleton
public class ShuffleBlacklistMigration extends ZkDataMigration {

  private final CuratorFramework curator;
  private final SingularityConfiguration configuration;
  private final ShuffleConfigurationManager manager;


  @Inject
  public ShuffleBlacklistMigration(CuratorFramework curator, SingularityConfiguration configuration, ShuffleConfigurationManager manager) {
    super(16);
    this.curator = curator;
    this.configuration = configuration;
    this.manager = manager;
  }

  @Override
  public void applyMigration() {
    for (String requestId : configuration.getDoNotShuffleRequests()) {
      manager.addToShuffleBlacklist(requestId);
    }
  }
}

@@ -246,62 +246,8 @@
 ger;
-%0Aimport org.apache.curator.framework.CuratorFramework;
 %0A%0A@S
@@ -323,51 +323,8 @@
 on %7B
-%0A%0A  private final CuratorFramework curator;
 %0A  p
@@ -480,34 +480,8 @@
 ion(
-CuratorFramework curator, 
 Sing
@@ -574,36 +574,8 @@
 6);%0A
-    this.curator = curator;%0A
     
@@ -819,8 +819,9 @@
 %7D%0A  %7D%0A%7D%0A
+a

remove stray parameter