Search in sources :

Example 16 with IdentityEntity

use of org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity in project opennms by OpenNMS.

the class AbstractBusinessServiceRestServiceIT method canCreateBusinessService.

@Test
public void canCreateBusinessService() throws Exception {
    final BusinessServiceEntity bs = new BusinessServiceEntityBuilder().name("some-service").addAttribute("some-key", "some-value").reduceFunction(new HighestSeverityEntity()).addReductionKey("reductionKey-1", new IdentityEntity()).addReductionKey("reductionKey-2", new IdentityEntity()).addReductionKey("reductionKey-3", new IdentityEntity()).toEntity();
    sendData(POST, getMediaType(), "/business-services", marshal(toRequestDto(bs)), 201);
    Assert.assertEquals(1, m_businessServiceDao.countAll());
    for (BusinessServiceEntity eachEntity : m_businessServiceDao.findAll()) {
        BusinessServiceResponseDTO responseDTO = verifyResponse(eachEntity);
        Assert.assertEquals(3, responseDTO.getReductionKeys().size());
    }
}
Also used : BusinessServiceResponseDTO(org.opennms.web.rest.v2.bsm.model.BusinessServiceResponseDTO) BusinessServiceEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity) HighestSeverityEntity(org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity) BusinessServiceEntityBuilder(org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder) IdentityEntity(org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity) Test(org.junit.Test)

Example 17 with IdentityEntity

use of org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity in project opennms by OpenNMS.

the class AbstractBusinessServiceRestServiceIT method canUpdateBusinessService.

@Test
public void canUpdateBusinessService() throws Exception {
    // initialize
    BusinessServiceEntity bs = new BusinessServiceEntityBuilder().name("Dummy Service").addAttribute("some-key", "some-value").addReductionKey("key1", new IdentityEntity()).addReductionKey("key2-deleteMe", new IdentityEntity()).reduceFunction(new HighestSeverityEntity()).toEntity();
    final Long serviceId = m_businessServiceDao.save(bs);
    m_businessServiceDao.flush();
    // update
    BusinessServiceRequestDTO requestDTO = toRequestDto(bs);
    requestDTO.setName("New Name");
    requestDTO.getAttributes().put("key", "value");
    requestDTO.getReductionKeys().clear();
    requestDTO.addReductionKey("key1updated", new FunctionsManager().getMapFunctionDTO(new Ignore()), Edge.DEFAULT_WEIGHT);
    sendData(PUT, getMediaType(), "/business-services/" + serviceId, marshal(requestDTO), 204);
    // Reload from database and verify changes
    bs = m_businessServiceDao.get(serviceId);
    Assert.assertEquals(requestDTO.getName(), bs.getName());
    Assert.assertEquals(requestDTO.getAttributes(), bs.getAttributes());
    Assert.assertEquals(1, bs.getReductionKeyEdges().size());
    Assert.assertEquals(1, bs.getEdges().size());
    Assert.assertEquals(1, m_businessServiceDao.findAll().size());
    Assert.assertEquals(Sets.newHashSet(), bs.getIpServiceEdges());
    BusinessServiceResponseDTO responseDTO = verifyResponse(bs);
    verifyReductionKey("key1updated", responseDTO);
}
Also used : Ignore(org.opennms.netmgt.bsm.service.model.functions.map.Ignore) FunctionsManager(org.opennms.web.rest.v2.bsm.model.meta.FunctionsManager) BusinessServiceResponseDTO(org.opennms.web.rest.v2.bsm.model.BusinessServiceResponseDTO) BusinessServiceRequestDTO(org.opennms.web.rest.v2.bsm.model.BusinessServiceRequestDTO) BusinessServiceEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity) HighestSeverityEntity(org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity) BusinessServiceEntityBuilder(org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder) IdentityEntity(org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity) Test(org.junit.Test)

Example 18 with IdentityEntity

use of org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity in project opennms by OpenNMS.

the class BusinessServiceDaoIT method verifyDeleteOnCascade.

/**
 * If we do not explicitly delete the map or reduce function it should be deleted if not referenced anymore.
 */
@Test
@Transactional
public void verifyDeleteOnCascade() {
    BusinessServiceEntity child2 = new BusinessServiceEntityBuilder().name("Child 2").reduceFunction(new HighestSeverityEntity()).addReductionKey("some-key", new IdentityEntity()).toEntity();
    BusinessServiceEntity child1 = new BusinessServiceEntityBuilder().name("Child 1").reduceFunction(new HighestSeverityEntity()).addChildren(child2, new IdentityEntity()).toEntity();
    BusinessServiceEntity parent = new BusinessServiceEntityBuilder().name("Parent Web Servers").addAttribute("dc", "RDU").addReductionKey("TestReductionKeyA", new IdentityEntity()).addReductionKey("TestReductionKeyB", new IdentityEntity()).addIpService(getMonitoredServiceFromNode1(), new IdentityEntity()).reduceFunction(m_highestSeverity).addChildren(child1, new IdentityEntity()).toEntity();
    m_businessServiceDao.save(child2);
    m_businessServiceDao.save(child1);
    m_businessServiceDao.save(parent);
    m_businessServiceDao.flush();
    assertEquals(3, m_businessServiceDao.countAll());
    assertEquals(3, m_reductionFunctionDao.countAll());
    assertEquals(6, m_edgeDao.countAll());
    // Deletion of child does not delete the edges referencing to that child
    // remove all parent -> child associations manually
    BusinessServiceChildEdgeEntity parentToChild1Edge = parent.getChildEdges().iterator().next();
    parent.removeEdge(parentToChild1Edge);
    m_edgeDao.delete(parentToChild1Edge);
    // edges do not need to be deleted manually, deletes will be cascaded
    m_businessServiceDao.delete(child1);
    m_businessServiceDao.flush();
    assertEquals(2, m_businessServiceDao.countAll());
    assertEquals(2, m_reductionFunctionDao.countAll());
    assertEquals(4, m_edgeDao.countAll());
    // Deletion of parent should delete all references
    m_businessServiceDao.delete(parent);
    assertEquals(1, m_businessServiceDao.countAll());
    assertEquals(1, m_reductionFunctionDao.countAll());
    assertEquals(1, m_edgeDao.countAll());
    // Deletion of Child 2 should also work
    m_businessServiceDao.delete(child2);
    assertEquals(0, m_businessServiceDao.countAll());
    assertEquals(0, m_reductionFunctionDao.countAll());
    assertEquals(0, m_edgeDao.countAll());
}
Also used : BusinessServiceChildEdgeEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceChildEdgeEntity) BusinessServiceEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity) HighestSeverityEntity(org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity) BusinessServiceEntityBuilder(org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder) IdentityEntity(org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity) Test(org.junit.Test) Transactional(org.springframework.transaction.annotation.Transactional)

Example 19 with IdentityEntity

use of org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity in project opennms by OpenNMS.

the class DefaultBusinessServiceStateMachineIT method ensureReductionKeyLookupIsFastEnough.

/**
 * Builds 200 business services.
 * Each business Services has one parent and two children.
 * Each children has 12 reduction key edges.
 * In sum there are:
 *  - 600 business services
 *  - 4800 reduction key edges
 *  - 400 child edges
 *  - 5200 edges
 *
 *  See NMS-8978 for more details.
 */
@Test
public void ensureReductionKeyLookupIsFastEnough() {
    for (int i = 0; i < 200; i++) {
        BusinessServiceEntity parentEntity = new BusinessServiceEntityBuilder().name("Parent " + i).reduceFunction(new HighestSeverityEntity()).toEntity();
        for (int c = 0; c < 2; c++) {
            BusinessServiceEntityBuilder childBuilder = new BusinessServiceEntityBuilder().name("Child " + i + " " + c).reduceFunction(new HighestSeverityEntity());
            for (int a = 0; a < 12; a++) {
                childBuilder.addReductionKey("custom." + i + "." + c + "." + a, new IdentityEntity());
            }
            BusinessServiceEntity childEntity = childBuilder.toEntity();
            parentEntity.addChildServiceEdge(childEntity, new IdentityEntity());
            businessServiceDao.save(childEntity);
        }
        businessServiceDao.save(parentEntity);
    }
    final Set<String> uniqueReductionKeys = businessServiceDao.findMatching(new CriteriaBuilder(BusinessServiceEntity.class).like("name", "Child%").toCriteria()).stream().flatMap(service -> service.getReductionKeyEdges().stream()).flatMap(edge -> edge.getReductionKeys().stream()).collect(Collectors.toSet());
    for (String eachKey : uniqueReductionKeys) {
        final OnmsAlarm alarm = new OnmsAlarm();
        alarm.setUei("custom");
        alarm.setAlarmType(OnmsAlarm.PROBLEM_TYPE);
        alarm.setDescription("dummy");
        alarm.setLogMsg("dummy");
        alarm.setSeverity(OnmsSeverity.WARNING);
        alarm.setReductionKey(eachKey);
        alarm.setDistPoller(distPollerDao.whoami());
        alarm.setCounter(1);
        populator.getAlarmDao().save(alarm);
    }
    populator.getAlarmDao().flush();
    businessServiceDao.flush();
    // Simulate lookup of reduction keys
    final long start = System.currentTimeMillis();
    final DefaultBusinessServiceStateMachine stateMachine = new DefaultBusinessServiceStateMachine();
    stateMachine.setAlarmProvider(alarmProvider);
    stateMachine.setBusinessServices(businessServiceDao.findAll().stream().map(e -> new BusinessServiceImpl(businessServiceManager, e)).collect(Collectors.toList()));
    long diff = System.currentTimeMillis() - start;
    LoggerFactory.getLogger(getClass()).info("Took {} ms to initialize state machine", diff);
    Assert.assertTrue("Reduction Key lookup took much longer than expected. Expected was 1000 ms, but took " + diff + " ms", 1000 >= diff);
}
Also used : CriteriaBuilder(org.opennms.core.criteria.CriteriaBuilder) BusinessServicesShareIpServiceHierarchy(org.opennms.netmgt.bsm.test.hierarchies.BusinessServicesShareIpServiceHierarchy) LoggerFactory(org.slf4j.LoggerFactory) Autowired(org.springframework.beans.factory.annotation.Autowired) AlarmWrapper(org.opennms.netmgt.bsm.service.model.AlarmWrapper) BusinessService(org.opennms.netmgt.bsm.service.model.BusinessService) LoggingStateChangeHandler(org.opennms.netmgt.bsm.test.LoggingStateChangeHandler) IpServiceEdgeImpl(org.opennms.netmgt.bsm.service.internal.edge.IpServiceEdgeImpl) HighestSeverityEntity(org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity) After(org.junit.After) IdentityEntity(org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity) Set(java.util.Set) Collectors(java.util.stream.Collectors) BeanUtils(org.opennms.core.spring.BeanUtils) BusinessServiceEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity) List(java.util.List) OpenNMSJUnit4ClassRunner(org.opennms.core.test.OpenNMSJUnit4ClassRunner) HTTP_8085_BAMBOO_REDUCTION_KEY(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy.HTTP_8085_BAMBOO_REDUCTION_KEY) Edge(org.opennms.netmgt.bsm.service.model.edge.Edge) JUnitTemporaryDatabase(org.opennms.core.test.db.annotations.JUnitTemporaryDatabase) IPServiceEdgeEntity(org.opennms.netmgt.bsm.persistence.api.IPServiceEdgeEntity) Identity(org.opennms.netmgt.bsm.service.model.functions.map.Identity) BambooTestHierarchy(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy) CriteriaBuilder(org.opennms.core.criteria.CriteriaBuilder) IpServiceEdge(org.opennms.netmgt.bsm.service.model.edge.IpServiceEdge) RunWith(org.junit.runner.RunWith) BusinessServiceDao(org.opennms.netmgt.bsm.persistence.api.BusinessServiceDao) BusinessServiceManager(org.opennms.netmgt.bsm.service.BusinessServiceManager) DISK_USAGE_THRESHOLD_BAMBO_REDUCTION_KEY(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy.DISK_USAGE_THRESHOLD_BAMBO_REDUCTION_KEY) EventConstants(org.opennms.netmgt.events.api.EventConstants) OnmsAlarm(org.opennms.netmgt.model.OnmsAlarm) BsmDatabasePopulator(org.opennms.netmgt.bsm.test.BsmDatabasePopulator) Lists(com.google.common.collect.Lists) DistPollerDao(org.opennms.netmgt.dao.api.DistPollerDao) Status(org.opennms.netmgt.bsm.service.model.Status) Qualifier(org.springframework.beans.factory.annotation.Qualifier) AlarmProvider(org.opennms.netmgt.bsm.service.AlarmProvider) SimpleTestHierarchy(org.opennms.netmgt.bsm.test.hierarchies.SimpleTestHierarchy) BAMBOO_AGENT_CAROLINA_REDUCTION_KEY(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy.BAMBOO_AGENT_CAROLINA_REDUCTION_KEY) Before(org.junit.Before) BsmTestUtils.createAlarmWrapper(org.opennms.netmgt.bsm.test.BsmTestUtils.createAlarmWrapper) JUnitConfigurationEnvironment(org.opennms.test.JUnitConfigurationEnvironment) Test(org.junit.Test) BAMBOO_AGENT_DUKE_REDUCTION_KEY(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy.BAMBOO_AGENT_DUKE_REDUCTION_KEY) BusinessServiceEntityBuilder(org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder) ReductionKeyHelper(org.opennms.netmgt.dao.util.ReductionKeyHelper) ContextConfiguration(org.springframework.test.context.ContextConfiguration) BAMBOO_AGENT_NCSTATE_REDUCTION_KEY(org.opennms.netmgt.bsm.test.hierarchies.BambooTestHierarchy.BAMBOO_AGENT_NCSTATE_REDUCTION_KEY) Assert(org.junit.Assert) Assert.assertEquals(org.junit.Assert.assertEquals) OnmsSeverity(org.opennms.netmgt.model.OnmsSeverity) Transactional(org.springframework.transaction.annotation.Transactional) OnmsAlarm(org.opennms.netmgt.model.OnmsAlarm) HighestSeverityEntity(org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity) BusinessServiceEntity(org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity) BusinessServiceEntityBuilder(org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder) IdentityEntity(org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity) Test(org.junit.Test)

Aggregations

IdentityEntity (org.opennms.netmgt.bsm.persistence.api.functions.map.IdentityEntity)19 Test (org.junit.Test)16 BusinessServiceEntity (org.opennms.netmgt.bsm.persistence.api.BusinessServiceEntity)14 HighestSeverityEntity (org.opennms.netmgt.bsm.persistence.api.functions.reduce.HighestSeverityEntity)12 BusinessServiceEntityBuilder (org.opennms.netmgt.bsm.test.BusinessServiceEntityBuilder)12 Transactional (org.springframework.transaction.annotation.Transactional)5 IPServiceEdgeEntity (org.opennms.netmgt.bsm.persistence.api.IPServiceEdgeEntity)4 BusinessServiceResponseDTO (org.opennms.web.rest.v2.bsm.model.BusinessServiceResponseDTO)4 ConstraintViolation (javax.validation.ConstraintViolation)3 Before (org.junit.Before)3 CriteriaBuilder (org.opennms.core.criteria.CriteriaBuilder)2 BusinessServiceChildEdgeEntity (org.opennms.netmgt.bsm.persistence.api.BusinessServiceChildEdgeEntity)2 Lists (com.google.common.collect.Lists)1 List (java.util.List)1 Set (java.util.Set)1 Collectors (java.util.stream.Collectors)1 After (org.junit.After)1 Assert (org.junit.Assert)1 Assert.assertEquals (org.junit.Assert.assertEquals)1 RunWith (org.junit.runner.RunWith)1