use of com.hortonworks.streamline.streams.cluster.catalog.Component in project streamline by hortonworks.
the class StormServiceRegistrarTest method testRegister.
@Test
public void testRegister() throws Exception {
Cluster cluster = getTestCluster(1L);
StormServiceRegistrar registrar = initializeServiceRegistrar();
// explicit convert Object
Config config = new Config();
config.put(StormServiceRegistrar.PARAM_NIMBUS_SEEDS, "storm-1,storm-2");
config.put(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_PORT, (Object) 6627);
config.put(StormServiceRegistrar.PARAM_UI_HOST, "storm-1");
config.put(StormServiceRegistrar.PARAM_UI_PORT, (Object) 8080);
config.put(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE, (Object) 102476800);
config.put(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT, "org.apache.storm.security.auth.SimpleTransportPlugin");
config.put(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL, "org.apache.storm.security.auth.DefaultPrincipalToLocal");
config.put(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME, "nimbus/_HOST@EXAMPLE.COM");
registrar.register(cluster, config, Collections.emptyList());
Service stormService = environmentService.getServiceByName(cluster.getId(), Constants.Storm.SERVICE_NAME);
assertNotNull(stormService);
Component nimbus = environmentService.getComponentByName(stormService.getId(), ComponentPropertyPattern.NIMBUS.name());
assertNotNull(nimbus);
Collection<ComponentProcess> nimbusProcesses = environmentService.listComponentProcesses(nimbus.getId());
List<String> hosts = nimbusProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toList());
assertEquals(Sets.newHashSet("storm-1", "storm-2"), new HashSet<>(hosts));
List<Integer> ports = nimbusProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toList());
assertEquals(Sets.newHashSet(6627, 6627), new HashSet<>(ports));
Component ui = environmentService.getComponentByName(stormService.getId(), ComponentPropertyPattern.STORM_UI_SERVER.name());
assertNotNull(ui);
Collection<ComponentProcess> uiProcesses = environmentService.listComponentProcesses(ui.getId());
assertEquals(Sets.newHashSet("storm-1"), uiProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toSet()));
assertEquals(Sets.newHashSet(8080), uiProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toSet()));
ServiceConfiguration stormYamlConf = environmentService.getServiceConfigurationByName(stormService.getId(), CONFIGURATION_NAME_STORM_YAML);
assertNotNull(stormYamlConf);
Map<String, String> stormYamlConfMap = stormYamlConf.getConfigurationMap();
assertEquals(config.getAny(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE), Integer.valueOf(stormYamlConfMap.get(StormServiceRegistrar.PARAM_NIMBUS_THRIFT_MAX_BUFFER_SIZE)));
assertEquals(config.get(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT), stormYamlConfMap.get(StormServiceRegistrar.PARAM_THRIFT_TRANSPORT));
assertEquals(config.get(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL), stormYamlConfMap.get(StormServiceRegistrar.PARAM_PRINCIPAL_TO_LOCAL));
ServiceConfiguration stormEnvConf = environmentService.getServiceConfigurationByName(stormService.getId(), CONFIGURATION_NAME_STORM_ENV);
assertNotNull(stormEnvConf);
Map<String, String> stormEnvConfMap = stormEnvConf.getConfigurationMap();
assertEquals(config.get(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME), stormEnvConfMap.get(StormServiceRegistrar.PARAM_NIMBUS_PRINCIPAL_NAME));
}
use of com.hortonworks.streamline.streams.cluster.catalog.Component in project streamline by hortonworks.
the class ZookeeperServiceRegistrarTest method testRegister.
@Test
public void testRegister() throws Exception {
Cluster cluster = getTestCluster(1L);
ZookeeperServiceRegistrar registrar = initializeServiceRegistrar();
Config config = new Config();
config.put(ZookeeperServiceRegistrar.PARAM_ZOOKEEPER_SERVER_HOSTNAMES, Lists.newArrayList("zookeeper-1", "zookeeper-2"));
config.put(ZookeeperServiceRegistrar.PARAM_ZOOKEEPER_PORT, (Object) 2181);
registrar.register(cluster, config, Collections.emptyList());
Service zkService = environmentService.getServiceByName(cluster.getId(), Constants.Zookeeper.SERVICE_NAME);
assertNotNull(zkService);
Component zkServer = environmentService.getComponentByName(zkService.getId(), ComponentPropertyPattern.ZOOKEEPER_SERVER.name());
assertNotNull(zkServer);
Collection<ComponentProcess> zkServerProcesses = environmentService.listComponentProcesses(zkServer.getId());
assertEquals(Sets.newHashSet("zookeeper-1", "zookeeper-2"), zkServerProcesses.stream().map(ComponentProcess::getHost).collect(Collectors.toSet()));
assertEquals(Sets.newHashSet(2181, 2181), zkServerProcesses.stream().map(ComponentProcess::getPort).collect(Collectors.toSet()));
ServiceConfiguration zooConf = environmentService.getServiceConfigurationByName(zkService.getId(), CONFIGURATION_NAME_ZOO_CFG);
assertNotNull(zooConf);
}
use of com.hortonworks.streamline.streams.cluster.catalog.Component in project streamline by hortonworks.
the class ComponentCatalogResource method addOrUpdateComponent.
@PUT
@Path("/services/{serviceId}/components")
@Timed
public Response addOrUpdateComponent(@PathParam("serviceId") Long serviceId, Component component, @Context SecurityContext securityContext) {
SecurityUtil.checkPermissions(authorizer, securityContext, Cluster.NAMESPACE, getClusterId(serviceId), WRITE);
// overwrite service id to given path param
component.setServiceId(serviceId);
Service service = environmentService.getService(serviceId);
if (service == null) {
throw EntityNotFoundException.byId("service: " + serviceId.toString());
}
Component createdComponent = environmentService.addOrUpdateComponent(serviceId, component);
return WSUtils.respondEntity(createdComponent, CREATED);
}
use of com.hortonworks.streamline.streams.cluster.catalog.Component in project streamline by hortonworks.
the class StormMetadataService method newSupervisorComponent.
private Component newSupervisorComponent() {
final Component component = new Component();
component.setName("SUPERVISOR");
return component;
}
use of com.hortonworks.streamline.streams.cluster.catalog.Component in project streamline by hortonworks.
the class EnvironmentService method createComponent.
public Component createComponent(Service service, String componentName) {
Component component = new Component();
component.setId(this.dao.nextId(COMPONENT_NAMESPACE));
component.setName(componentName);
component.setServiceId(service.getId());
component.setTimestamp(System.currentTimeMillis());
return component;
}
Aggregations