SleepServer.java

/**
 * Copyright 2009 the original author or authors.
 *
 * Licensed 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 net.sf.katta.util;

import java.io.File;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Random;
import java.util.Set;

import net.sf.katta.node.IContentServer;

/**
 * This class implements the back-end side of a dummy server, to be used for
 * testing. It just sleeps for a while and then returns nothing.
 */
public class SleepServer implements IContentServer, ISleepServer {

  private Random rand = new Random();
  protected final Set<String> _shards = Collections.synchronizedSet(new HashSet<String>());
  protected String _nodeName;

  public long getProtocolVersion(final String protocol, final long clientVersion) {
    return 0L;
  }

  @Override
  public void init(String nodeName, NodeConfiguration nodeConfiguration) {
    _nodeName = nodeName;
  }

  public void addShard(final String shardName, final File shardDir) {
    _shards.add(shardName);
  }

  public void removeShard(final String shardName) {
    _shards.remove(shardName);
  }

  @Override
  public Collection<String> getShards() {
    return _shards;
  }

  public Map<String, String> getShardMetaData(final String shardName) {
    return new HashMap<String, String>(0);
  }

  public void shutdown() {
    _shards.clear();
  }

  public int sleep(long msec, int delta, String[] shards) throws IllegalArgumentException {
    if (shards != null) {
      String err = "";
      String sep = "";
      for (String shard : shards) {
        if (!_shards.contains(shard)) {
          System.err.println("Node " + _nodeName + " does not have shard " + shard + "!!");
          err += sep + shard;
          sep = ", ";
        }
      }
      if (err.length() > 0) {
        throw new IllegalArgumentException("Node " + _nodeName + " invalid shards: " + err);
      }
    }
    if (delta > 0) {
      msec = Math.max(0, msec + Math.round(((2.0 * rand.nextDouble()) - 1.0) * delta));
    }
    if (msec > 0) {
      try {
        Thread.sleep(msec);
      } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
      }
    }
    return shards != null ? shards.length : 0;
  }

}