com.hubspot.singularity.data.zkmigrations.AgentAndRackMigration Maven / Gradle / Ivy

Go to download

package com.hubspot.singularity.data.zkmigrations;

import com.google.inject.Inject;
import javax.inject.Singleton;
import org.apache.curator.framework.CuratorFramework;
import org.apache.zookeeper.KeeperException.NoNodeException;

@Singleton
public class AgentAndRackMigration extends ZkDataMigration {
  private final CuratorFramework curator;

  @Inject
  public AgentAndRackMigration(CuratorFramework curator) {
    super(3);
    this.curator = curator;
  }

  @Override
  public void applyMigration() {
    try {
      try {
        curator.delete().deletingChildrenIfNeeded().forPath("/slaves");
      } catch (NoNodeException nee) {}
      try {
        curator.delete().deletingChildrenIfNeeded().forPath("/racks");
      } catch (NoNodeException nee) {}
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }
}