framework/core/src/test/java/org/apache/manifoldcf/core/lockmanager/ZooKeeperInstance.java

/* $Id$ */

/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.manifoldcf.core.lockmanager;

import java.util.*;
import java.io.*;
import org.apache.zookeeper.server.*;
import org.apache.zookeeper.server.quorum.*;

public class ZooKeeperInstance
{
  protected final int zkPort;
  protected final File tempDir;
  
  protected ZooKeeperThread zookeeperThread = null;
  
  public ZooKeeperInstance(int zkPort, File tempDir)
  {
    this.zkPort = zkPort;
    this.tempDir = tempDir;
  }

  public void start()
    throws Exception
  {
    Properties startupProperties = new Properties();
    startupProperties.setProperty("tickTime","2000");
    startupProperties.setProperty("dataDir",tempDir.toString());
    startupProperties.setProperty("clientPort",Integer.toString(zkPort));

    final QuorumPeerConfig quorumConfiguration = new QuorumPeerConfig();
    quorumConfiguration.parseProperties(startupProperties);

    final ServerConfig configuration = new ServerConfig();
    configuration.readFrom(quorumConfiguration);

    zookeeperThread = new ZooKeeperThread(configuration);
    zookeeperThread.start();
    // We have no way of knowing whether zookeeper is alive or not, but the
    // client is supposed to know about that.  But it doesn't, so wait for 5 seconds
    Thread.sleep(5000L);
  }
  
  public void stop()
    throws Exception
  {
    while (true)
    {
      if (zookeeperThread == null)
        break;
      else if (!zookeeperThread.isAlive())
      {
        Throwable e = zookeeperThread.finishUp();
        if (e != null)
        {
          if (e instanceof RuntimeException)
            throw (RuntimeException)e;
          else if (e instanceof Exception)
            throw (Exception)e;
          else if (e instanceof Error)
            throw (Error)e;
        }
        zookeeperThread = null;
      }
      else
      {
        // This isn't the best way to kill zookeeper but it's the only way
        // we've got.
        zookeeperThread.interrupt();
        Thread.sleep(1000L);
      }
    }
  }
  
  protected static class ZooKeeperThread extends Thread
  {
    protected final ServerConfig config;
    
    protected Throwable exception = null;
    
    public ZooKeeperThread(ServerConfig config)
    {
      this.config = config;
    }
    
    public void run()
    {
      try
      {
        ZooKeeperServerMain server = new ZooKeeperServerMain();
        server.runFromConfig(config);
      }
      catch (IOException e)
      {
        // Ignore IOExceptions, since that seems to be normal when shutting
        // down zookeeper via thread.interrupt()
      }
      catch (Throwable e)
      {
        exception = e;
      }
    }
    
    public Throwable finishUp()
      throws InterruptedException
    {
      join();
      return exception;
    }
  }
  
}