| package org.apache.helix.mock.spectator; |
| |
| /* |
| * 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. |
| */ |
| |
| import java.util.List; |
| |
| import org.apache.helix.HelixManager; |
| import org.apache.helix.HelixManagerFactory; |
| import org.apache.helix.InstanceType; |
| import org.apache.helix.PropertyPathBuilder; |
| import org.apache.helix.zookeeper.datamodel.ZNRecord; |
| import org.apache.helix.manager.zk.ZNRecordSerializer; |
| import org.apache.helix.model.InstanceConfig; |
| import org.apache.helix.spectator.RoutingTableProvider; |
| import org.apache.helix.tools.ClusterSetup; |
| import org.apache.helix.zookeeper.zkclient.IDefaultNameSpace; |
| import org.apache.helix.zookeeper.zkclient.ZkClient; |
| import org.apache.helix.zookeeper.zkclient.ZkServer; |
| |
| |
| /** |
| * A MockSpectatorProcess to demonstrate the integration with cluster manager. |
| * This uses Zookeeper in local mode and runs at port 2188 |
| */ |
| public class MockSpectatorProcess { |
| private static final int port = 2188; |
| static long runId = System.currentTimeMillis(); |
| private static final String dataDir = "/tmp/zkDataDir-" + runId; |
| |
| private static final String logDir = "/tmp/zkLogDir-" + runId; |
| |
| static String clusterName = "mock-cluster-" + runId; |
| |
| static String zkConnectString = "localhost:2188"; |
| |
| private final RoutingTableProvider _routingTableProvider; |
| private static ZkServer zkServer; |
| |
| public MockSpectatorProcess() { |
| _routingTableProvider = new RoutingTableProvider(); |
| } |
| |
| public static void main(String[] args) throws Exception { |
| setup(); |
| zkServer.getZkClient().setZkSerializer(new ZNRecordSerializer()); |
| ZNRecord record = zkServer.getZkClient().readData(PropertyPathBuilder.idealState(clusterName, "TestDB")); |
| String externalViewPath = PropertyPathBuilder.externalView(clusterName, "TestDB"); |
| |
| MockSpectatorProcess process = new MockSpectatorProcess(); |
| process.start(); |
| // try to route, there is no master or slave available |
| process.routeRequest("TestDB", "TestDB_1"); |
| |
| // update the externalview on zookeeper |
| zkServer.getZkClient().createPersistent(externalViewPath, record); |
| // sleep for sometime so that the ZK Callback is received. |
| Thread.sleep(1000); |
| process.routeRequest("TestDB", "TestDB_1"); |
| System.exit(1); |
| } |
| |
| private static void setup() { |
| |
| IDefaultNameSpace defaultNameSpace = new IDefaultNameSpace() { |
| @Override |
| public void createDefaultNameSpace(ZkClient client) { |
| client.deleteRecursive("/" + clusterName); |
| |
| } |
| }; |
| |
| zkServer = new ZkServer(dataDir, logDir, defaultNameSpace, port); |
| zkServer.start(); |
| ClusterSetup clusterSetup = new ClusterSetup(zkConnectString); |
| clusterSetup.setupTestCluster(clusterName); |
| try { |
| Thread.sleep(1000); |
| } catch (InterruptedException e) { |
| e.printStackTrace(); |
| } |
| } |
| |
| public void routeRequest(String database, String partition) { |
| List<InstanceConfig> masters; |
| List<InstanceConfig> slaves; |
| masters = _routingTableProvider.getInstances(database, partition, "MASTER"); |
| if (masters != null && !masters.isEmpty()) { |
| System.out.println("Available masters to route request"); |
| for (InstanceConfig config : masters) { |
| System.out.println("HostName:" + config.getHostName() + " Port:" + config.getPort()); |
| } |
| } else { |
| System.out.println("No masters available to route request"); |
| } |
| slaves = _routingTableProvider.getInstances(database, partition, "SLAVE"); |
| if (slaves != null && !slaves.isEmpty()) { |
| System.out.println("Available slaves to route request"); |
| for (InstanceConfig config : slaves) { |
| System.out.println("HostName:" + config.getHostName() + " Port:" + config.getPort()); |
| } |
| } else { |
| System.out.println("No slaves available to route request"); |
| } |
| } |
| |
| public void start() { |
| |
| try { |
| HelixManager manager = |
| HelixManagerFactory.getZKHelixManager(clusterName, null, InstanceType.SPECTATOR, |
| zkConnectString); |
| |
| manager.connect(); |
| manager.addExternalViewChangeListener(_routingTableProvider); |
| } catch (Exception e) { |
| e.printStackTrace(); |
| } |
| } |
| } |