Java tutorial
/* * 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.ambari.funtest.server.utils; import com.google.gson.JsonElement; import com.google.gson.JsonObject; import com.google.inject.Inject; import com.google.inject.Injector; import org.apache.ambari.funtest.server.AmbariUserRole; import org.apache.ambari.funtest.server.ClusterConfigParams; import org.apache.ambari.funtest.server.ConnectionParams; import org.apache.ambari.funtest.server.WebRequest; import org.apache.ambari.funtest.server.WebResponse; import org.apache.ambari.funtest.server.api.cluster.AddDesiredConfigurationWebRequest; import org.apache.ambari.funtest.server.api.cluster.CreateClusterWebRequest; import org.apache.ambari.funtest.server.api.cluster.CreateConfigurationWebRequest; import org.apache.ambari.funtest.server.api.cluster.GetRequestStatusWebRequest; import org.apache.ambari.funtest.server.api.cluster.SetUserPrivilegeWebRequest; import org.apache.ambari.funtest.server.api.host.AddHostWebRequest; import org.apache.ambari.funtest.server.api.host.RegisterHostWebRequest; import org.apache.ambari.funtest.server.api.service.AddServiceWebRequest; import org.apache.ambari.funtest.server.api.service.InstallServiceWebRequest; import org.apache.ambari.funtest.server.api.servicecomponent.AddServiceComponentWebRequest; import org.apache.ambari.funtest.server.api.servicecomponenthost.BulkAddServiceComponentHostsWebRequest; import org.apache.ambari.funtest.server.api.servicecomponenthost.BulkSetServiceComponentHostStateWebRequest; import org.apache.ambari.funtest.server.api.user.CreateUserWebRequest; import org.apache.ambari.server.actionmanager.HostRoleStatus; import org.apache.ambari.server.controller.AmbariServer; import org.apache.ambari.server.orm.entities.PrincipalTypeEntity; import org.apache.ambari.server.state.Clusters; import org.apache.ambari.server.state.Host; import org.apache.ambari.server.state.State; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import java.util.Arrays; import java.util.HashMap; import java.util.Map; public class ClusterUtils { private static Log LOG = LogFactory.getLog(ClusterUtils.class); @Inject private Injector injector; public void createSampleCluster(ConnectionParams serverParams) throws Exception { WebResponse response = null; JsonElement jsonResponse; String clusterName = "c1"; String hostName = "host1"; String clusterVersion = "HDP-2.2.0"; /** * Create a cluster */ jsonResponse = RestApiUtils .executeRequest(new CreateClusterWebRequest(serverParams, clusterName, clusterVersion)); /** * Register a host */ if (injector == null) { jsonResponse = RestApiUtils.executeRequest(new RegisterHostWebRequest(serverParams, hostName)); } else { /** * Hack: Until we figure out how to get the agent servlet going, * register a host directly using the Clusters class. */ Clusters clusters = injector.getInstance(Clusters.class); clusters.addHost(hostName); Host host1 = clusters.getHost(hostName); Map<String, String> hostAttributes = new HashMap<String, String>(); hostAttributes.put("os_family", "redhat"); hostAttributes.put("os_release_version", "6.3"); host1.setHostAttributes(hostAttributes); host1.persist(); } /** * Add the registered host to the new cluster */ jsonResponse = RestApiUtils.executeRequest(new AddHostWebRequest(serverParams, clusterName, hostName)); /** * Create and add a configuration to our cluster */ String configType = "test-hadoop-env"; String configTag = "version1"; ClusterConfigParams configParams = new ClusterConfigParams(); configParams.setClusterName(clusterName); configParams.setConfigType(configType); configParams.setConfigTag(configTag); configParams.setProperties(new HashMap<String, String>() { { put("fs.default.name", "localhost:9995"); } }); jsonResponse = RestApiUtils.executeRequest(new CreateConfigurationWebRequest(serverParams, configParams)); /** * Apply the desired configuration to our cluster */ jsonResponse = RestApiUtils .executeRequest(new AddDesiredConfigurationWebRequest(serverParams, configParams)); /** * Add a service to the cluster */ String serviceName = "HDFS"; jsonResponse = RestApiUtils .executeRequest(new AddServiceWebRequest(serverParams, clusterName, serviceName)); String[] componentNames = new String[] { "NAMENODE", "DATANODE", "SECONDARY_NAMENODE" }; /** * Add components to the service */ for (String componentName : componentNames) { jsonResponse = RestApiUtils.executeRequest( new AddServiceComponentWebRequest(serverParams, clusterName, serviceName, componentName)); } /** * Install the service */ jsonResponse = RestApiUtils .executeRequest(new InstallServiceWebRequest(serverParams, clusterName, serviceName)); /** * Add components to the host */ jsonResponse = RestApiUtils.executeRequest(new BulkAddServiceComponentHostsWebRequest(serverParams, clusterName, Arrays.asList(hostName), Arrays.asList(componentNames))); /** * Install the service component hosts */ jsonResponse = RestApiUtils.executeRequest(new BulkSetServiceComponentHostStateWebRequest(serverParams, clusterName, State.INIT, State.INSTALLED)); if (!jsonResponse.isJsonNull()) { int requestId = parseRequestId(jsonResponse); RequestStatusPoller.poll(serverParams, clusterName, requestId); } /** * Start the service component hosts */ jsonResponse = RestApiUtils.executeRequest(new BulkSetServiceComponentHostStateWebRequest(serverParams, clusterName, State.INSTALLED, State.STARTED)); if (!jsonResponse.isJsonNull()) { int requestId = parseRequestId(jsonResponse); RequestStatusPoller.poll(serverParams, clusterName, requestId); } /** * Start the service */ //jsonResponse = RestApiUtils.executeRequest(new StartServiceWebRequest(serverParams, clusterName, serviceName)); } /** * Creates a user with the specified role. * * @param connectionParams * @param clusterName * @param userName * @param password * @param userRole * @throws Exception */ public static void createUser(ConnectionParams connectionParams, String clusterName, String userName, String password, AmbariUserRole userRole) throws Exception { JsonElement jsonResponse; jsonResponse = RestApiUtils.executeRequest(new CreateUserWebRequest(connectionParams, userName, password, CreateUserWebRequest.ActiveUser.TRUE, CreateUserWebRequest.AdminUser.FALSE)); LOG.info(jsonResponse); if (userRole != AmbariUserRole.NONE) { jsonResponse = RestApiUtils.executeRequest(new SetUserPrivilegeWebRequest(connectionParams, clusterName, userName, userRole, PrincipalTypeEntity.USER_PRINCIPAL_TYPE_NAME)); LOG.info(jsonResponse); } } /** * Creates a user with CLUSTER.USER privilege. * * @param connectionParams * @param clusterName * @param userName * @param password * @throws Exception */ public static void createUserClusterUser(ConnectionParams connectionParams, String clusterName, String userName, String password) throws Exception { createUser(connectionParams, clusterName, userName, password, AmbariUserRole.CLUSTER_USER); } /** * Creates a user with SERVICE.OPERATOR privilege * * @param connectionParams * @param clusterName * @param userName * @param password * @throws Exception */ public static void createUserServiceOperator(ConnectionParams connectionParams, String clusterName, String userName, String password) throws Exception { createUser(connectionParams, clusterName, userName, password, AmbariUserRole.SERVICE_OPERATOR); } /** * Creates a user with SERVICE.ADMINISTRATOR privilege * * @param connectionParams * @param clusterName * @param userName * @param password * @throws Exception */ public static void createUserServiceAdministrator(ConnectionParams connectionParams, String clusterName, String userName, String password) throws Exception { createUser(connectionParams, clusterName, userName, password, AmbariUserRole.SERVICE_ADMINISTRATOR); } /** * Creates a user with CLUSTER.OPERATOR privilege * * @param connectionParams * @param clusterName * @param userName * @param password * @throws Exception */ public static void createUserClusterOperator(ConnectionParams connectionParams, String clusterName, String userName, String password) throws Exception { createUser(connectionParams, clusterName, userName, password, AmbariUserRole.CLUSTER_OPERATOR); } /** * Creates a user with CLUSTER.ADMINISTRATOR privilege. * * @param connectionParams * @param clusterName * @param userName * @param password * @throws Exception */ public static void createUserClusterAdministrator(ConnectionParams connectionParams, String clusterName, String userName, String password) throws Exception { createUser(connectionParams, clusterName, userName, password, AmbariUserRole.CLUSTER_ADMINISTRATOR); } /** * Parses a JSON response string for { "Requests" : { "id" : "2" } } * * @param jsonResponse * @return - request id * @throws IllegalArgumentException */ private static int parseRequestId(JsonElement jsonResponse) throws IllegalArgumentException { if (jsonResponse.isJsonNull()) { throw new IllegalArgumentException("jsonResponse with request id expected."); } JsonObject jsonObject = jsonResponse.getAsJsonObject(); int requestId = jsonObject.get("Requests").getAsJsonObject().get("id").getAsInt(); return requestId; } }