2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2016-2018 Ericsson. All rights reserved.
4 * ================================================================================
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.policy.apex.testsuites.integration.common.model;
23 import java.util.ArrayList;
24 import java.util.Arrays;
25 import java.util.List;
28 import java.util.TreeMap;
29 import java.util.TreeSet;
30 import org.onap.policy.apex.model.basicmodel.concepts.ApexRuntimeException;
31 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
32 import org.onap.policy.apex.model.basicmodel.concepts.AxKeyInformation;
33 import org.onap.policy.apex.model.basicmodel.concepts.AxReferenceKey;
34 import org.onap.policy.apex.model.basicmodel.concepts.AxValidationResult;
35 import org.onap.policy.apex.model.contextmodel.concepts.AxContextAlbums;
36 import org.onap.policy.apex.model.contextmodel.concepts.AxContextSchema;
37 import org.onap.policy.apex.model.contextmodel.concepts.AxContextSchemas;
38 import org.onap.policy.apex.model.eventmodel.concepts.AxEvent;
39 import org.onap.policy.apex.model.eventmodel.concepts.AxEvents;
40 import org.onap.policy.apex.model.eventmodel.concepts.AxField;
41 import org.onap.policy.apex.model.policymodel.concepts.AxLogicReader;
42 import org.onap.policy.apex.model.policymodel.concepts.AxPolicies;
43 import org.onap.policy.apex.model.policymodel.concepts.AxPolicy;
44 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;
45 import org.onap.policy.apex.model.policymodel.concepts.AxState;
46 import org.onap.policy.apex.model.policymodel.concepts.AxStateOutput;
47 import org.onap.policy.apex.model.policymodel.concepts.AxStateTaskOutputType;
48 import org.onap.policy.apex.model.policymodel.concepts.AxStateTaskReference;
49 import org.onap.policy.apex.model.policymodel.concepts.AxTask;
50 import org.onap.policy.apex.model.policymodel.concepts.AxTaskLogic;
51 import org.onap.policy.apex.model.policymodel.concepts.AxTaskSelectionLogic;
52 import org.onap.policy.apex.model.policymodel.concepts.AxTasks;
53 import org.onap.policy.apex.model.policymodel.handling.PolicyLogicReader;
56 * This class creates sample evaluation Policy Models.
58 * @author John Keeney (john.keeney@ericsson.com)
60 public class EvalDomainModelFactory {
61 // Recurring string constants
62 private static final String ACTION = "Action";
63 private static final String CONDITION = "Condition";
64 private static final String EVENT = "Event";
65 private static final String TASK_SELECTION_LOGIC = "TaskSelectionLogic";
66 private static final String JYTHON = "JYTHON";
67 private static final String MVEL = "MVEL";
68 private static final String ORIENT = "Orient";
69 private static final String STATE_NAME = "<STATE_NAME>";
70 private static final String OBSERVE = "Observe";
71 private static final String JAVASCRIPT = "JAVASCRIPT";
72 private static final String TASK_NAME = "<TASK_NAME>";
73 private static final String TASK_LOGIC = "TaskLogic";
74 private static final String STATE_NUMBER = "<STATE_NUMBER>";
75 private static final String STATE3_TIMESTAMP = "State3Timestamp";
76 private static final String EVENT_NUMBER = "EventNumber";
77 private static final String FIRST_EVENT_TIMESTAMP = "FirstEventTimestamp";
78 private static final String STATE2_TIMESTAMP = "State2Timestamp";
79 private static final String STATE1_TIMESTAMP = "State1Timestamp";
80 private static final String TEST_TEMPERATURE = "TestTemperature";
81 private static final String SENT_TIMESTAMP = "SentTimestamp";
82 private static final String EVENT_TAG = ".events";
83 private static final String DEFAULT_VERSION = "0.0.1";
85 private static final int THIRD_MEMBER = 3;
87 private static final String PACKAGE = EvalDomainModelFactory.class.getPackage().getName();
89 private String justOneLang = null;
91 private final AxContextSchema typeTimestamp;
92 private final AxContextSchema typeDouble;
93 private final AxContextSchema typeCounter;
94 private final AxContextSchemas schemas;
95 private final AxEvent event0000;
96 private final AxEvent event0001;
97 private final AxEvent event0002;
98 private final AxEvent event0003;
99 private final AxEvent event0004;
100 private final AxEvents events;
103 * The Constructor for the factory.
105 public EvalDomainModelFactory() {
110 * The Constructor for the factory that creates models for a single executor language flavour.
112 * @param justOneLang the just one lang
114 public EvalDomainModelFactory(final String justOneLang) {
115 this.justOneLang = justOneLang;
117 typeTimestamp = new AxContextSchema(new AxArtifactKey("TestTimestamp", DEFAULT_VERSION), "Java",
119 typeDouble = new AxContextSchema(new AxArtifactKey(TEST_TEMPERATURE, DEFAULT_VERSION), "Java",
121 typeCounter = new AxContextSchema(new AxArtifactKey("TestTimestamp", DEFAULT_VERSION), "Java",
124 schemas = new AxContextSchemas(new AxArtifactKey("TestDatatypes", DEFAULT_VERSION));
125 schemas.getSchemasMap().put(typeTimestamp.getKey(), typeTimestamp);
126 schemas.getSchemasMap().put(typeDouble.getKey(), typeDouble);
127 schemas.getSchemasMap().put(typeCounter.getKey(), typeCounter);
129 event0000 = new AxEvent(new AxArtifactKey("Event0000", DEFAULT_VERSION), PACKAGE + EVENT_TAG);
130 event0000.getParameterMap().put(SENT_TIMESTAMP,
131 new AxField(new AxReferenceKey(event0000.getKey(), SENT_TIMESTAMP), typeTimestamp.getKey()));
132 event0000.getParameterMap().put(TEST_TEMPERATURE,
133 new AxField(new AxReferenceKey(event0000.getKey(), TEST_TEMPERATURE), typeDouble.getKey()));
134 event0000.getParameterMap().put(FIRST_EVENT_TIMESTAMP, new AxField(
135 new AxReferenceKey(event0000.getKey(), FIRST_EVENT_TIMESTAMP), typeTimestamp.getKey()));
136 event0000.getParameterMap().put(EVENT_NUMBER,
137 new AxField(new AxReferenceKey(event0000.getKey(), EVENT_NUMBER), typeCounter.getKey()));
139 event0001 = new AxEvent(new AxArtifactKey("Event0001", DEFAULT_VERSION), PACKAGE + EVENT_TAG);
140 event0001.getParameterMap().put(SENT_TIMESTAMP,
141 new AxField(new AxReferenceKey(event0001.getKey(), SENT_TIMESTAMP), typeTimestamp.getKey()));
142 event0001.getParameterMap().put(STATE1_TIMESTAMP,
143 new AxField(new AxReferenceKey(event0001.getKey(), STATE1_TIMESTAMP), typeTimestamp.getKey()));
144 event0001.getParameterMap().put(TEST_TEMPERATURE,
145 new AxField(new AxReferenceKey(event0001.getKey(), TEST_TEMPERATURE), typeDouble.getKey()));
146 event0001.getParameterMap().put(FIRST_EVENT_TIMESTAMP, new AxField(
147 new AxReferenceKey(event0001.getKey(), FIRST_EVENT_TIMESTAMP), typeTimestamp.getKey()));
148 event0001.getParameterMap().put(EVENT_NUMBER,
149 new AxField(new AxReferenceKey(event0001.getKey(), EVENT_NUMBER), typeCounter.getKey()));
151 event0002 = new AxEvent(new AxArtifactKey("Event0002", DEFAULT_VERSION), PACKAGE + EVENT_TAG);
152 event0002.getParameterMap().put(SENT_TIMESTAMP,
153 new AxField(new AxReferenceKey(event0002.getKey(), SENT_TIMESTAMP), typeTimestamp.getKey()));
154 event0002.getParameterMap().put(STATE1_TIMESTAMP,
155 new AxField(new AxReferenceKey(event0002.getKey(), STATE1_TIMESTAMP), typeTimestamp.getKey()));
156 event0002.getParameterMap().put(STATE2_TIMESTAMP,
157 new AxField(new AxReferenceKey(event0002.getKey(), STATE2_TIMESTAMP), typeTimestamp.getKey()));
158 event0002.getParameterMap().put(TEST_TEMPERATURE,
159 new AxField(new AxReferenceKey(event0002.getKey(), TEST_TEMPERATURE), typeDouble.getKey()));
160 event0002.getParameterMap().put(FIRST_EVENT_TIMESTAMP, new AxField(
161 new AxReferenceKey(event0002.getKey(), FIRST_EVENT_TIMESTAMP), typeTimestamp.getKey()));
162 event0002.getParameterMap().put(EVENT_NUMBER,
163 new AxField(new AxReferenceKey(event0002.getKey(), EVENT_NUMBER), typeCounter.getKey()));
165 event0003 = new AxEvent(new AxArtifactKey("Event0003", DEFAULT_VERSION), PACKAGE + EVENT_TAG);
166 event0003.getParameterMap().put(SENT_TIMESTAMP,
167 new AxField(new AxReferenceKey(event0003.getKey(), SENT_TIMESTAMP), typeTimestamp.getKey()));
168 event0003.getParameterMap().put(STATE1_TIMESTAMP,
169 new AxField(new AxReferenceKey(event0003.getKey(), STATE1_TIMESTAMP), typeTimestamp.getKey()));
170 event0003.getParameterMap().put(STATE2_TIMESTAMP,
171 new AxField(new AxReferenceKey(event0003.getKey(), STATE2_TIMESTAMP), typeTimestamp.getKey()));
172 event0003.getParameterMap().put(STATE3_TIMESTAMP,
173 new AxField(new AxReferenceKey(event0003.getKey(), STATE3_TIMESTAMP), typeTimestamp.getKey()));
174 event0003.getParameterMap().put(TEST_TEMPERATURE,
175 new AxField(new AxReferenceKey(event0003.getKey(), TEST_TEMPERATURE), typeDouble.getKey()));
176 event0003.getParameterMap().put(FIRST_EVENT_TIMESTAMP, new AxField(
177 new AxReferenceKey(event0003.getKey(), FIRST_EVENT_TIMESTAMP), typeTimestamp.getKey()));
178 event0003.getParameterMap().put(EVENT_NUMBER,
179 new AxField(new AxReferenceKey(event0003.getKey(), EVENT_NUMBER), typeCounter.getKey()));
181 event0004 = new AxEvent(new AxArtifactKey("Event0004", DEFAULT_VERSION), PACKAGE + EVENT_TAG);
182 event0004.getParameterMap().put(SENT_TIMESTAMP,
183 new AxField(new AxReferenceKey(event0004.getKey(), SENT_TIMESTAMP), typeTimestamp.getKey()));
184 event0004.getParameterMap().put(STATE1_TIMESTAMP,
185 new AxField(new AxReferenceKey(event0004.getKey(), STATE1_TIMESTAMP), typeTimestamp.getKey()));
186 event0004.getParameterMap().put(STATE2_TIMESTAMP,
187 new AxField(new AxReferenceKey(event0004.getKey(), STATE2_TIMESTAMP), typeTimestamp.getKey()));
188 event0004.getParameterMap().put(STATE3_TIMESTAMP,
189 new AxField(new AxReferenceKey(event0004.getKey(), STATE3_TIMESTAMP), typeTimestamp.getKey()));
190 event0004.getParameterMap().put("State4Timestamp",
191 new AxField(new AxReferenceKey(event0004.getKey(), "State4Timestamp"), typeTimestamp.getKey()));
192 event0004.getParameterMap().put(TEST_TEMPERATURE,
193 new AxField(new AxReferenceKey(event0004.getKey(), TEST_TEMPERATURE), typeDouble.getKey()));
194 event0004.getParameterMap().put(FIRST_EVENT_TIMESTAMP, new AxField(
195 new AxReferenceKey(event0004.getKey(), FIRST_EVENT_TIMESTAMP), typeTimestamp.getKey()));
196 event0004.getParameterMap().put(EVENT_NUMBER,
197 new AxField(new AxReferenceKey(event0004.getKey(), EVENT_NUMBER), typeCounter.getKey()));
199 events = new AxEvents(new AxArtifactKey("Events", DEFAULT_VERSION));
200 events.getEventMap().put(event0000.getKey(), event0000);
201 events.getEventMap().put(event0001.getKey(), event0001);
202 events.getEventMap().put(event0002.getKey(), event0002);
203 events.getEventMap().put(event0003.getKey(), event0003);
204 events.getEventMap().put(event0004.getKey(), event0004);
208 * Get a sample OODA policy model.
210 * @return the sample policy model
212 public AxPolicyModel getOodaPolicyModel() {
213 final AxTasks tasks = new AxTasks(new AxArtifactKey("Tasks", DEFAULT_VERSION));
215 final AxLogicReader logicReader = new PolicyLogicReader().setLogicPackage(PACKAGE)
216 .setDefaultLogic("EvalTaskLogic");
218 final AxTask obTask = new AxTask(new AxArtifactKey("Task_Observe_0", DEFAULT_VERSION));
219 obTask.duplicateInputFields(event0000.getParameterMap());
220 obTask.duplicateOutputFields(event0001.getParameterMap());
221 final AxTaskLogic obAxLogic = new AxTaskLogic(obTask.getKey(), TASK_LOGIC,
222 (justOneLang == null ? JAVASCRIPT : justOneLang), logicReader);
223 obAxLogic.setLogic(obAxLogic.getLogic().replaceAll(STATE_NAME, OBSERVE)
224 .replaceAll(TASK_NAME, obTask.getKey().getName()).replaceAll(STATE_NUMBER, "1"));
225 obTask.setTaskLogic(obAxLogic);
227 final AxTask orTask = new AxTask(new AxArtifactKey("Task_Orient_0", DEFAULT_VERSION));
228 orTask.duplicateInputFields(event0001.getParameterMap());
229 orTask.duplicateOutputFields(event0002.getParameterMap());
230 final AxTaskLogic orAxLogic = new AxTaskLogic(orTask.getKey(), TASK_LOGIC,
231 (justOneLang == null ? JAVASCRIPT : justOneLang), logicReader);
232 orAxLogic.setLogic(orAxLogic.getLogic().replaceAll(STATE_NAME, ORIENT)
233 .replaceAll(TASK_NAME, orTask.getKey().getName()).replaceAll(STATE_NUMBER, "2"));
234 orTask.setTaskLogic(orAxLogic);
236 final AxTask dTask = new AxTask(new AxArtifactKey("Task_Decide_0", DEFAULT_VERSION));
237 dTask.duplicateInputFields(event0002.getParameterMap());
238 dTask.duplicateOutputFields(event0003.getParameterMap());
239 final AxTaskLogic dAxLogic = new AxTaskLogic(dTask.getKey(), TASK_LOGIC,
240 (justOneLang == null ? MVEL : justOneLang), logicReader);
241 dAxLogic.setLogic(dAxLogic.getLogic().replaceAll(STATE_NAME, ORIENT)
242 .replaceAll(TASK_NAME, dTask.getKey().getName()).replaceAll(STATE_NUMBER, "3"));
243 dTask.setTaskLogic(dAxLogic);
245 final AxTask aTask = new AxTask(new AxArtifactKey("Task_Act_0", DEFAULT_VERSION));
246 aTask.duplicateInputFields(event0003.getParameterMap());
247 aTask.duplicateOutputFields(event0004.getParameterMap());
248 final AxTaskLogic aAxLogic = new AxTaskLogic(aTask.getKey(), TASK_LOGIC,
249 (justOneLang == null ? "JAVA" : justOneLang), logicReader);
250 aAxLogic.setLogic(aAxLogic.getLogic().replaceAll(STATE_NAME, "Act")
251 .replaceAll(TASK_NAME, aTask.getKey().getName()).replaceAll(STATE_NUMBER, "4"));
252 aTask.setTaskLogic(aAxLogic);
254 tasks.getTaskMap().put(obTask.getKey(), obTask);
255 tasks.getTaskMap().put(orTask.getKey(), orTask);
256 tasks.getTaskMap().put(dTask.getKey(), dTask);
257 tasks.getTaskMap().put(aTask.getKey(), aTask);
259 final Set<AxArtifactKey> obTasks = new TreeSet<>();
260 final Set<AxArtifactKey> orTasks = new TreeSet<>();
261 final Set<AxArtifactKey> decTasks = new TreeSet<>();
262 final Set<AxArtifactKey> actTasks = new TreeSet<>();
264 addOodaTasks(tasks, obTasks, orTasks, decTasks, actTasks);
266 final List<Set<AxArtifactKey>> taskReferenceList = new ArrayList<>();
267 taskReferenceList.add(obTasks);
268 taskReferenceList.add(orTasks);
269 taskReferenceList.add(decTasks);
270 taskReferenceList.add(actTasks);
272 final List<AxArtifactKey> p0InEventList = new ArrayList<>();
273 p0InEventList.add(event0000.getKey());
274 p0InEventList.add(event0001.getKey());
275 p0InEventList.add(event0002.getKey());
276 p0InEventList.add(event0003.getKey());
278 final List<AxArtifactKey> p0OutEventList = new ArrayList<>();
279 p0OutEventList.add(event0001.getKey());
280 p0OutEventList.add(event0002.getKey());
281 p0OutEventList.add(event0003.getKey());
282 p0OutEventList.add(event0004.getKey());
284 final List<AxArtifactKey> p0defaultTaskList = new ArrayList<>();
285 p0defaultTaskList.add(tasks.get("Task_Observe_0").getKey());
286 p0defaultTaskList.add(tasks.get("Task_Orient_0").getKey());
287 p0defaultTaskList.add(tasks.get("Task_Decide_0").getKey());
288 p0defaultTaskList.add(tasks.get("Task_Act_0").getKey());
290 final AxPolicy policy0 = new AxPolicy(new AxArtifactKey("OODAPolicy_0", DEFAULT_VERSION));
291 final List<String> axLogicExecutorTypeList = initAxLogicExecutorTypeList(justOneLang);
293 policy0.setStateMap(getOodaStateMap(policy0.getKey(), p0InEventList, p0OutEventList, axLogicExecutorTypeList,
294 p0defaultTaskList, taskReferenceList));
295 policy0.setFirstState(policy0.getStateMap().get(OBSERVE).getKey().getLocalName());
297 final AxPolicies policies = new AxPolicies(new AxArtifactKey("OODAPolicies", DEFAULT_VERSION));
298 policies.getPolicyMap().put(policy0.getKey(), policy0);
300 final AxKeyInformation keyInformation = new AxKeyInformation(
301 new AxArtifactKey("KeyInformation", DEFAULT_VERSION));
302 final AxPolicyModel policyModel = new AxPolicyModel(
303 new AxArtifactKey("EvaluationPolicyModel_OODA", DEFAULT_VERSION));
304 policyModel.setPolicies(policies);
305 policyModel.setEvents(events);
306 policyModel.setTasks(tasks);
307 policyModel.setAlbums(new AxContextAlbums(new AxArtifactKey("Albums", DEFAULT_VERSION)));
308 policyModel.setSchemas(schemas);
309 policyModel.setKeyInformation(keyInformation);
310 policyModel.getKeyInformation().generateKeyInfo(policyModel);
312 final AxValidationResult result = policyModel.validate(new AxValidationResult());
313 if (!result.isOk()) {
314 throw new ApexRuntimeException("model " + policyModel.getId() + " is not valid" + result);
320 * Add OODA tasks to the policy.
322 * @param tasks the policy tasks
323 * @param obTasks observe tasks
324 * @param orTasks orient tasks
325 * @param decTasks decide tasks
326 * @param actTasks act tasks
328 private void addOodaTasks(final AxTasks tasks, final Set<AxArtifactKey> obTasks, final Set<AxArtifactKey> orTasks,
329 final Set<AxArtifactKey> decTasks, final Set<AxArtifactKey> actTasks) {
330 for (final AxTask task : tasks.getTaskMap().values()) {
331 if (task.getKey().getName().contains(OBSERVE)) {
332 obTasks.add(task.getKey());
334 if (task.getKey().getName().contains(ORIENT)) {
335 orTasks.add(task.getKey());
337 if (task.getKey().getName().contains("Decide")) {
338 decTasks.add(task.getKey());
340 if (task.getKey().getName().contains("Act")) {
341 actTasks.add(task.getKey());
347 * Initialize the logic executor list.
349 * @param justOneLang the language to use
350 * @return the list of languages
352 private List<String> initAxLogicExecutorTypeList(final String justOneLang) {
353 List<String> axLogicExecutorTypeList = new ArrayList<>();
355 if (justOneLang != null) {
356 axLogicExecutorTypeList.add(justOneLang);
358 axLogicExecutorTypeList.add(JAVASCRIPT);
359 axLogicExecutorTypeList.add(JYTHON);
360 axLogicExecutorTypeList.add(JYTHON);
361 axLogicExecutorTypeList.add(MVEL);
363 return axLogicExecutorTypeList;
367 * Gets the OODA state map.
369 * @param policyKey the policy key
370 * @param inEventKeyList the in event key list
371 * @param outEventKeyList the out event key list
372 * @param axLogicExecutorTypeList the ax logic executor type list
373 * @param defaultTaskList the default task list
374 * @param taskKeySetList the task key set list
375 * @return the OODA state map
377 private Map<String, AxState> getOodaStateMap(final AxArtifactKey policyKey,
378 final List<AxArtifactKey> inEventKeyList, final List<AxArtifactKey> outEventKeyList,
379 final List<String> axLogicExecutorTypeList, final List<AxArtifactKey> defaultTaskList,
380 final List<Set<AxArtifactKey>> taskKeySetList) {
381 final AxLogicReader logicReader = new PolicyLogicReader().setLogicPackage(PACKAGE)
382 .setDefaultLogic("EvalStateLogic");
384 final AxState actState = new AxState(new AxReferenceKey(policyKey, "Act"));
385 actState.setTrigger(inEventKeyList.get(THIRD_MEMBER));
386 final AxStateOutput act2Out = new AxStateOutput(new AxReferenceKey(actState.getKey(), "Act2Out"),
387 outEventKeyList.get(THIRD_MEMBER), AxReferenceKey.getNullKey());
388 actState.getStateOutputs().put(act2Out.getKey().getLocalName(), act2Out);
389 actState.setTaskSelectionLogic(new AxTaskSelectionLogic(actState.getKey(), TASK_SELECTION_LOGIC,
390 axLogicExecutorTypeList.get(THIRD_MEMBER), logicReader));
391 actState.setDefaultTask(defaultTaskList.get(THIRD_MEMBER));
392 for (final AxArtifactKey taskKey : taskKeySetList.get(THIRD_MEMBER)) {
393 actState.getTaskReferences().put(taskKey,
394 new AxStateTaskReference(new AxReferenceKey(actState.getKey(), taskKey.getName()),
395 AxStateTaskOutputType.DIRECT, act2Out.getKey()));
398 final AxState decState = new AxState(new AxReferenceKey(policyKey, "Decide"));
399 decState.setTrigger(inEventKeyList.get(2));
400 final AxStateOutput dec2Act = new AxStateOutput(new AxReferenceKey(decState.getKey(), "Dec2Act"),
401 outEventKeyList.get(2), actState.getKey());
402 decState.getStateOutputs().put(dec2Act.getKey().getLocalName(), dec2Act);
403 decState.setTaskSelectionLogic(new AxTaskSelectionLogic(decState.getKey(), TASK_SELECTION_LOGIC,
404 axLogicExecutorTypeList.get(2), logicReader));
405 decState.setDefaultTask(defaultTaskList.get(2));
406 for (final AxArtifactKey taskKey : taskKeySetList.get(2)) {
407 decState.getTaskReferences().put(taskKey,
408 new AxStateTaskReference(new AxReferenceKey(decState.getKey(), taskKey.getName()),
409 AxStateTaskOutputType.DIRECT, dec2Act.getKey()));
412 final AxState orState = new AxState(new AxReferenceKey(policyKey, ORIENT));
413 orState.setTrigger(inEventKeyList.get(1));
414 final AxStateOutput or2Dec = new AxStateOutput(new AxReferenceKey(orState.getKey(), "Or2Dec"),
415 outEventKeyList.get(1), decState.getKey());
416 orState.getStateOutputs().put(or2Dec.getKey().getLocalName(), or2Dec);
417 orState.setTaskSelectionLogic(new AxTaskSelectionLogic(orState.getKey(), TASK_SELECTION_LOGIC,
418 axLogicExecutorTypeList.get(1), logicReader));
419 orState.setDefaultTask(defaultTaskList.get(1));
420 for (final AxArtifactKey taskKey : taskKeySetList.get(1)) {
421 orState.getTaskReferences().put(taskKey,
422 new AxStateTaskReference(new AxReferenceKey(orState.getKey(), taskKey.getName()),
423 AxStateTaskOutputType.DIRECT, or2Dec.getKey()));
426 final AxState obState = new AxState(new AxReferenceKey(policyKey, OBSERVE));
427 obState.setTrigger(inEventKeyList.get(0));
428 final AxStateOutput ob2Or = new AxStateOutput(new AxReferenceKey(obState.getKey(), "Ob2Or"),
429 outEventKeyList.get(0), orState.getKey());
430 obState.getStateOutputs().put(ob2Or.getKey().getLocalName(), ob2Or);
431 obState.setTaskSelectionLogic(new AxTaskSelectionLogic(obState.getKey(), TASK_SELECTION_LOGIC,
432 axLogicExecutorTypeList.get(0), logicReader));
433 obState.setDefaultTask(defaultTaskList.get(0));
434 for (final AxArtifactKey taskKey : taskKeySetList.get(0)) {
435 obState.getTaskReferences().put(taskKey,
436 new AxStateTaskReference(new AxReferenceKey(obState.getKey(), taskKey.getName()),
437 AxStateTaskOutputType.DIRECT, ob2Or.getKey()));
440 final Map<String, AxState> stateMap = new TreeMap<>();
441 stateMap.put(obState.getKey().getLocalName(), obState);
442 stateMap.put(orState.getKey().getLocalName(), orState);
443 stateMap.put(decState.getKey().getLocalName(), decState);
444 stateMap.put(actState.getKey().getLocalName(), actState);
450 * Get a sample ECA policy model.
452 * @return the sample policy model
454 public AxPolicyModel getEcaPolicyModel() {
456 final AxTasks tasks = new AxTasks(new AxArtifactKey("Tasks", DEFAULT_VERSION));
458 final AxLogicReader logicReader = new PolicyLogicReader().setLogicPackage(PACKAGE)
459 .setDefaultLogic("EvalTaskLogic");
461 final AxTask eTask = new AxTask(new AxArtifactKey("Task_Event_0", DEFAULT_VERSION));
462 eTask.duplicateInputFields(event0000.getParameterMap());
463 eTask.duplicateOutputFields(event0001.getParameterMap());
464 final AxTaskLogic eAxLogic = new AxTaskLogic(eTask.getKey(), TASK_LOGIC,
465 (justOneLang == null ? JYTHON : justOneLang), logicReader);
466 eAxLogic.setLogic(eAxLogic.getLogic().replaceAll(STATE_NAME, EVENT)
467 .replaceAll(TASK_NAME, eTask.getKey().getName()).replaceAll(STATE_NUMBER, "1"));
468 eTask.setTaskLogic(eAxLogic);
470 final AxTask cTask = new AxTask(new AxArtifactKey("Task_Condition_0", DEFAULT_VERSION));
471 cTask.duplicateInputFields(event0001.getParameterMap());
472 cTask.duplicateOutputFields(event0002.getParameterMap());
473 final AxTaskLogic cAxLogic = new AxTaskLogic(cTask.getKey(), TASK_LOGIC,
474 (justOneLang == null ? JAVASCRIPT : justOneLang), logicReader);
475 cAxLogic.setLogic(cAxLogic.getLogic().replaceAll(STATE_NAME, CONDITION)
476 .replaceAll(TASK_NAME, cTask.getKey().getName()).replaceAll(STATE_NUMBER, "2"));
477 cTask.setTaskLogic(cAxLogic);
479 final AxTask aTask = new AxTask(new AxArtifactKey("Task_Action_0", DEFAULT_VERSION));
480 aTask.duplicateInputFields(event0002.getParameterMap());
481 aTask.duplicateOutputFields(event0003.getParameterMap());
482 final AxTaskLogic aAxLogic = new AxTaskLogic(aTask.getKey(), TASK_LOGIC,
483 (justOneLang == null ? "JAVA" : justOneLang), logicReader);
484 aAxLogic.setLogic(aAxLogic.getLogic().replaceAll(STATE_NAME, ACTION)
485 .replaceAll(TASK_NAME, aTask.getKey().getName()).replaceAll(STATE_NUMBER, "3"));
486 aTask.setTaskLogic(aAxLogic);
488 tasks.getTaskMap().put(eTask.getKey(), eTask);
489 tasks.getTaskMap().put(cTask.getKey(), cTask);
490 tasks.getTaskMap().put(aTask.getKey(), aTask);
492 final Set<AxArtifactKey> eventTasks = new TreeSet<>();
493 final Set<AxArtifactKey> conditionTasks = new TreeSet<>();
494 final Set<AxArtifactKey> actionTasks = new TreeSet<>();
496 addEcaTasks(tasks, eventTasks, conditionTasks, actionTasks);
497 final List<Set<AxArtifactKey>> taskReferenceList = new ArrayList<>();
498 taskReferenceList.add(eventTasks);
499 taskReferenceList.add(conditionTasks);
500 taskReferenceList.add(actionTasks);
502 final List<AxArtifactKey> p0InEventList = new ArrayList<>();
503 p0InEventList.add(event0000.getKey());
504 p0InEventList.add(event0001.getKey());
505 p0InEventList.add(event0002.getKey());
507 final List<AxArtifactKey> p0OutEventList = new ArrayList<>();
508 p0OutEventList.add(event0001.getKey());
509 p0OutEventList.add(event0002.getKey());
510 p0OutEventList.add(event0003.getKey());
512 final List<AxArtifactKey> p0defaultTaskList = new ArrayList<>();
513 p0defaultTaskList.add(tasks.get("Task_Event_0").getKey());
514 p0defaultTaskList.add(tasks.get("Task_Condition_0").getKey());
515 p0defaultTaskList.add(tasks.get("Task_Action_0").getKey());
517 final AxPolicy policy0 = new AxPolicy(new AxArtifactKey("ECAPolicy_0", DEFAULT_VERSION));
518 final List<String> axLogicExecutorTypeList = Arrays.asList((justOneLang == null ? JAVASCRIPT : justOneLang),
519 (justOneLang == null ? MVEL : justOneLang), (justOneLang == null ? JYTHON : justOneLang));
520 policy0.setStateMap(getEcaStateMap(policy0.getKey(), p0InEventList, p0OutEventList, axLogicExecutorTypeList,
521 p0defaultTaskList, taskReferenceList));
522 policy0.setFirstState(policy0.getStateMap().get(EVENT).getKey().getLocalName());
524 final AxPolicies policies = new AxPolicies(new AxArtifactKey("ECAPolicies", DEFAULT_VERSION));
525 policies.getPolicyMap().put(policy0.getKey(), policy0);
527 final AxKeyInformation keyInformation = new AxKeyInformation(
528 new AxArtifactKey("KeyInformation", DEFAULT_VERSION));
529 final AxPolicyModel policyModel = new AxPolicyModel(
530 new AxArtifactKey("EvaluationPolicyModel_ECA", DEFAULT_VERSION));
531 policyModel.setPolicies(policies);
532 policyModel.setEvents(events);
533 policyModel.setTasks(tasks);
534 policyModel.setAlbums(new AxContextAlbums(new AxArtifactKey("Albums", DEFAULT_VERSION)));
535 policyModel.setSchemas(schemas);
536 policyModel.setKeyInformation(keyInformation);
537 policyModel.getKeyInformation().generateKeyInfo(policyModel);
539 final AxValidationResult result = policyModel.validate(new AxValidationResult());
540 if (!result.isOk()) {
541 throw new ApexRuntimeException("model " + policyModel.getId() + " is not valid" + result);
549 * @param tasks the tasks
550 * @param eventTasks event tasks
551 * @param conditionTasks condition tasks
552 * @param actionTasks action tasks
554 private void addEcaTasks(final AxTasks tasks, final Set<AxArtifactKey> eventTasks,
555 final Set<AxArtifactKey> conditionTasks, final Set<AxArtifactKey> actionTasks) {
556 for (final AxTask task : tasks.getTaskMap().values()) {
557 if (task.getKey().getName().contains(EVENT)) {
558 eventTasks.add(task.getKey());
560 if (task.getKey().getName().contains(CONDITION)) {
561 conditionTasks.add(task.getKey());
563 if (task.getKey().getName().contains(ACTION)) {
564 actionTasks.add(task.getKey());
570 * Gets the ECA state map.
572 * @param policyKey the policy key
573 * @param inEventKeyList the in event key list
574 * @param outEventKeyList the out event key list
575 * @param axLogicExecutorTypeList the ax logic executor type list
576 * @param defaultTaskList the default task list
577 * @param taskKeySetList the task key set list
578 * @return the ECA state map
580 private Map<String, AxState> getEcaStateMap(final AxArtifactKey policyKey, final List<AxArtifactKey> inEventKeyList,
581 final List<AxArtifactKey> outEventKeyList, final List<String> axLogicExecutorTypeList,
582 final List<AxArtifactKey> defaultTaskList, final List<Set<AxArtifactKey>> taskKeySetList) {
583 final AxLogicReader logicReader = new PolicyLogicReader().setLogicPackage(PACKAGE)
584 .setDefaultLogic("EvalStateLogic");
586 final AxState actionState = new AxState(new AxReferenceKey(policyKey, ACTION));
587 actionState.setTrigger(inEventKeyList.get(2));
588 final AxStateOutput action2Out = new AxStateOutput(actionState.getKey(), AxReferenceKey.getNullKey(),
589 outEventKeyList.get(2));
590 actionState.getStateOutputs().put(action2Out.getKey().getLocalName(), action2Out);
591 actionState.setTaskSelectionLogic(new AxTaskSelectionLogic(actionState.getKey(), TASK_SELECTION_LOGIC,
592 axLogicExecutorTypeList.get(2), logicReader));
593 actionState.setDefaultTask(defaultTaskList.get(2));
594 for (final AxArtifactKey taskKey : taskKeySetList.get(2)) {
595 actionState.getTaskReferences().put(taskKey,
596 new AxStateTaskReference(new AxReferenceKey(actionState.getKey(), taskKey.getName()),
597 AxStateTaskOutputType.DIRECT, action2Out.getKey()));
600 final AxState conditionState = new AxState(new AxReferenceKey(policyKey, CONDITION));
601 conditionState.setTrigger(inEventKeyList.get(1));
602 final AxStateOutput condition2Action = new AxStateOutput(conditionState.getKey(), actionState.getKey(),
603 outEventKeyList.get(1));
604 conditionState.getStateOutputs().put(condition2Action.getKey().getLocalName(), condition2Action);
605 conditionState.setTaskSelectionLogic(new AxTaskSelectionLogic(conditionState.getKey(), TASK_SELECTION_LOGIC,
606 axLogicExecutorTypeList.get(1), logicReader));
607 conditionState.setDefaultTask(defaultTaskList.get(1));
608 for (final AxArtifactKey taskKey : taskKeySetList.get(1)) {
609 conditionState.getTaskReferences().put(taskKey,
610 new AxStateTaskReference(new AxReferenceKey(conditionState.getKey(), taskKey.getName()),
611 AxStateTaskOutputType.DIRECT, condition2Action.getKey()));
614 final AxState eventState = new AxState(new AxReferenceKey(policyKey, EVENT));
615 eventState.setTrigger(inEventKeyList.get(0));
616 final AxStateOutput event2Condition = new AxStateOutput(eventState.getKey(), conditionState.getKey(),
617 outEventKeyList.get(0));
618 eventState.getStateOutputs().put(event2Condition.getKey().getLocalName(), event2Condition);
619 eventState.setTaskSelectionLogic(new AxTaskSelectionLogic(eventState.getKey(), TASK_SELECTION_LOGIC,
620 axLogicExecutorTypeList.get(0), logicReader));
621 eventState.setDefaultTask(defaultTaskList.get(0));
622 for (final AxArtifactKey taskKey : taskKeySetList.get(0)) {
623 eventState.getTaskReferences().put(taskKey,
624 new AxStateTaskReference(new AxReferenceKey(eventState.getKey(), taskKey.getName()),
625 AxStateTaskOutputType.DIRECT, event2Condition.getKey()));
628 final Map<String, AxState> stateMap = new TreeMap<>();
629 stateMap.put(eventState.getKey().getLocalName(), eventState);
630 stateMap.put(conditionState.getKey().getLocalName(), conditionState);
631 stateMap.put(actionState.getKey().getLocalName(), actionState);