TestCluster.java

/*
 * Copyright 2014-2016 CyberVision, Inc.
 *
 * 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 org.kaaproject.kaa.server.operations.service.event;

import org.apache.curator.RetryPolicy;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.apache.curator.test.InstanceSpec;
import org.apache.curator.test.TestingCluster;
import org.kaaproject.kaa.server.common.zk.gen.ConnectionInfo;
import org.kaaproject.kaa.server.common.zk.gen.LoadInfo;
import org.kaaproject.kaa.server.common.zk.gen.OperationsNodeInfo;
import org.kaaproject.kaa.server.common.zk.gen.TransportMetaData;
import org.kaaproject.kaa.server.common.zk.operations.OperationsNode;
import org.kaaproject.kaa.server.common.zk.operations.OperationsNodeListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.ByteBuffer;
import java.util.ArrayList;

/**
 * The Class TestCluster.
 */
public class TestCluster {

  /**
   * The Constant ENDPOINT_NODE_HOST.
   */
  public static final String OPERATIONS_NODE_HOST = "127.0.0.1";
  /**
   * The Constant LOG.
   */
  private static final Logger LOG = LoggerFactory
      .getLogger(TestCluster.class);
  /**
   * The Constant BOOTSTRAP_NODE_HOST.
   */
  private static final String BOOTSTRAP_NODE_HOST = "192.168.0.202";
  /**
   * The zookeeper cluster static instance.
   */
  private static TestingCluster zkCluster;

  /**
   * The endpoint node static instance.
   */
  private static OperationsNode operationsNode;

  /**
   * Start.
   *
   * @throws Exception the exception
   */
  public static void checkStarted() throws Exception {
    zkCluster = new TestingCluster(new InstanceSpec(null, EventServiceThriftTestIT.ZK_PORT, -1, -1, true, -1, -1, -1));
    zkCluster.start();
    LOG.info("ZK Cluster started");
    OperationsNodeInfo endpointNodeInfo = buildOperationsNodeInfo();
    CuratorFramework zkClient = CuratorFrameworkFactory.newClient(zkCluster.getConnectString(), buildDefaultRetryPolicy());
    operationsNode = new OperationsNode(endpointNodeInfo, zkClient);
    operationsNode.start();

  }

  /**
   * Stop.
   *
   * @throws Exception the exception
   */
  public static void stop() throws Exception {
    operationsNode.close();
    zkCluster.close();
  }

  public static void addOperationsListener(OperationsNodeListener listener) {
    operationsNode.addListener(listener);
  }

  public static void removeOperationsListener(OperationsNodeListener listener) {
    operationsNode.removeListener(listener);
  }

  /**
   * Builds the default retry policy.
   *
   * @return the retry policy
   */
  private static RetryPolicy buildDefaultRetryPolicy() {
    return new ExponentialBackoffRetry(100, 1);
  }

  /**
   * Builds the endpoint node info.
   *
   * @return the endpoint node info
   */
  private static OperationsNodeInfo buildOperationsNodeInfo() {
    OperationsNodeInfo nodeInfo = new OperationsNodeInfo();
    ByteBuffer testKeyData = ByteBuffer.wrap(new byte[]{10, 11, 12, 45, 34, 23, 67, 89, 66, 12});
    nodeInfo.setConnectionInfo(new ConnectionInfo(OPERATIONS_NODE_HOST, 1000, testKeyData));
    nodeInfo.setLoadInfo(new LoadInfo(1, 1.0));
    nodeInfo.setTimeStarted(System.currentTimeMillis());
    nodeInfo.setTransports(new ArrayList<TransportMetaData>());
    return nodeInfo;
  }

  public static OperationsNode getOperationsNode() {
    return operationsNode;
  }
}