diff options
7 files changed, 110 insertions, 20 deletions
diff --git a/policy-management/src/main/java/org/onap/policy/drools/controller/DroolsController.java b/policy-management/src/main/java/org/onap/policy/drools/controller/DroolsController.java index 8e78b077..36a4d5bd 100644 --- a/policy-management/src/main/java/org/onap/policy/drools/controller/DroolsController.java +++ b/policy-management/src/main/java/org/onap/policy/drools/controller/DroolsController.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2017-2019, 2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -233,6 +233,17 @@ public interface DroolsController extends Startable, Lockable { <T> boolean delete(@NonNull Class<T> fact); /** + * Checks is a fact exists in a given session. + */ + <T> boolean exists(@NonNull String sessionName, @NonNull T fact); + + /** + * Checks if a fact exists in any session. + */ + <T> boolean exists(@NonNull T fact); + + + /** * halts and permanently releases all resources. * */ diff --git a/policy-management/src/main/java/org/onap/policy/drools/controller/internal/MavenDroolsController.java b/policy-management/src/main/java/org/onap/policy/drools/controller/internal/MavenDroolsController.java index b8edb55e..c1eb9f37 100644 --- a/policy-management/src/main/java/org/onap/policy/drools/controller/internal/MavenDroolsController.java +++ b/policy-management/src/main/java/org/onap/policy/drools/controller/internal/MavenDroolsController.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2017-2020 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2017-2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -870,24 +870,31 @@ public class MavenDroolsController implements DroolsController { } @Override - public <T> boolean delete(@NonNull String sessionName, @NonNull T fact) { - String factClassName = fact.getClass().getName(); + public <T> boolean delete(@NonNull String sessionName, @NonNull T objFact) { + KieSession kieSession = getSession(sessionName).getKieSession(); - PolicySession session = getSession(sessionName); - KieSession kieSession = session.getKieSession(); + // try first to get the object to delete first by reference + + FactHandle quickFact = kieSession.getFactHandle(objFact); + if (quickFact != null) { + logger.info("Fast delete of {} from {}", objFact, sessionName); + kieSession.delete(quickFact); + return true; + } + + // otherwise, try to the delete the fact associated with the object by scanning all + // facts from the same type and performing object equality. The set of facts + // is restricted to those of the same type - Collection<FactHandle> factHandles = kieSession.getFactHandles(new ClassObjectFilter(fact.getClass())); + Collection<FactHandle> factHandles = kieSession.getFactHandles(new ClassObjectFilter(objFact.getClass())); for (FactHandle factHandle : factHandles) { - try { - if (Objects.equals(fact, kieSession.getObject(factHandle))) { - logger.info("Deleting {} from {}", factClassName, sessionName); - kieSession.delete(factHandle); - return true; - } - } catch (Exception e) { - logger.warn(FACT_RETRIEVE_ERROR, factHandle, e); + if (Objects.equals(objFact, kieSession.getObject(factHandle))) { + logger.info("Slow delete of {} of type {} from {}", objFact, sessionName); + kieSession.delete(factHandle); + return true; } } + return false; } @@ -919,6 +926,30 @@ public class MavenDroolsController implements DroolsController { return this.getSessionNames().stream().map(ss -> delete(ss, fact)).reduce(false, Boolean::logicalOr); } + @Override + public <T> boolean exists(@NonNull String sessionName, @NonNull T objFact) { + KieSession kieSession = getSession(sessionName).getKieSession(); + if (kieSession.getFactHandle(objFact) != null) { + return true; + } + + // try to find the object by equality comparison instead if it could not be + // found by reference + + Collection<FactHandle> factHandles = kieSession.getFactHandles(new ClassObjectFilter(objFact.getClass())); + for (FactHandle factHandle : factHandles) { + if (Objects.equals(objFact, kieSession.getObject(factHandle))) { + return true; + } + } + + return false; + } + + @Override + public <T> boolean exists(@NonNull T fact) { + return this.getSessionNames().stream().anyMatch(ss -> exists(ss, fact)); + } @Override public Class<?> fetchModelClass(String className) { diff --git a/policy-management/src/main/java/org/onap/policy/drools/controller/internal/NullDroolsController.java b/policy-management/src/main/java/org/onap/policy/drools/controller/internal/NullDroolsController.java index 4373b35e..e20d1777 100644 --- a/policy-management/src/main/java/org/onap/policy/drools/controller/internal/NullDroolsController.java +++ b/policy-management/src/main/java/org/onap/policy/drools/controller/internal/NullDroolsController.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2017-2020 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2017-2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -214,6 +214,16 @@ public class NullDroolsController implements DroolsController { return false; } + @Override + public <T> boolean exists(@NonNull String sessionName, @NonNull T fact) { + return false; + } + + @Override + public <T> boolean exists(@NonNull T fact) { + return false; + } + private String makeInvokeMsg() { return this.getClass().getName() + " invoked"; } diff --git a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsController2Test.java b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsController2Test.java index 26103aa2..8628830c 100644 --- a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsController2Test.java +++ b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsController2Test.java @@ -202,6 +202,9 @@ public class MavenDroolsController2Test { when(kieSess.getKieBase()).thenReturn(kieBase); when(kieSess.getQueryResults(QUERY, PARM1, PARM2)).thenReturn(queryResults); + when(kieSess.getFactHandle(FACT1_OBJECT)).thenReturn(fact1); + when(kieSess.getFactHandle(FACT3_OBJECT)).thenReturn(fact3); + when(kieSess.getObject(fact1)).thenReturn(FACT1_OBJECT); when(kieSess.getObject(fact2)).thenReturn(""); when(kieSess.getObject(fact3)).thenReturn(FACT3_OBJECT); diff --git a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsControllerTest.java b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsControllerTest.java index b39276d9..a2f41bc6 100644 --- a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsControllerTest.java +++ b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/MavenDroolsControllerTest.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2019-2020 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2019-2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -93,6 +93,35 @@ public class MavenDroolsControllerTest { new GsonTestUtils().compareGson(controller, MavenDroolsControllerTest.class); } + @Test + public void testFact() throws InterruptedException { + DroolsController controller = createDroolsController(30000L); + + Integer one = 1; + Integer two = 2; + + Assert.assertTrue(controller.offer(one)); + Assert.assertTrue(controller.exists(one)); + Assert.assertFalse(controller.exists(two)); + + Assert.assertTrue(controller.offer(two)); + Assert.assertTrue(controller.exists(one)); + Assert.assertTrue(controller.exists(two)); + + Integer three = 3; + Assert.assertFalse(controller.delete(three)); + Assert.assertTrue(controller.exists(one)); + Assert.assertTrue(controller.exists(two)); + + Assert.assertTrue(controller.delete(two)); + Assert.assertTrue(controller.exists(one)); + Assert.assertFalse(controller.exists(two)); + + Assert.assertTrue(controller.delete(one)); + Assert.assertFalse(controller.exists(one)); + Assert.assertFalse(controller.exists(two)); + } + private DroolsController createDroolsController(long courtesyStartTimeMs) throws InterruptedException { if (releaseId == null) { throw new IllegalStateException("no prereq artifact installed in maven repository"); diff --git a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/NullDroolsControllerTest.java b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/NullDroolsControllerTest.java index 743de8f9..010056bb 100644 --- a/policy-management/src/test/java/org/onap/policy/drools/controller/internal/NullDroolsControllerTest.java +++ b/policy-management/src/test/java/org/onap/policy/drools/controller/internal/NullDroolsControllerTest.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2018-2020 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2018-2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -157,4 +157,11 @@ public class NullDroolsControllerTest { public void factQuery() { Assert.assertTrue(new NullDroolsController().factQuery(null, null, null, false).isEmpty()); } + + @Test + public void exists() { + Object o1 = new Object(); + Assert.assertFalse(new NullDroolsController().exists("blah", o1)); + Assert.assertFalse(new NullDroolsController().exists(o1)); + } } diff --git a/policy-management/src/test/resources/echo.drl b/policy-management/src/test/resources/echo.drl index bd26f95b..03716d96 100644 --- a/policy-management/src/test/resources/echo.drl +++ b/policy-management/src/test/resources/echo.drl @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2018-2019 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2018-2019, 2021 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -35,5 +35,4 @@ when $o : Object(); then System.out.println("ECHO: " + $o.toString()); - retract($o); end |