Remove actor and recipe checks from ControlLoopCompiler.java
[policy/models.git] / models-interactions / model-yaml / src / main / java / org / onap / policy / controlloop / compiler / ControlLoopCompiler.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * policy-yaml
4  * ================================================================================
5  * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
6  * Modifications Copyright (C) 2019 Nordix Foundation.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *      http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.controlloop.compiler;
23
24 import com.google.common.collect.ImmutableList;
25 import com.google.common.collect.ImmutableMap;
26
27 import java.io.InputStream;
28 import java.io.Serializable;
29 import java.util.Collections;
30 import java.util.HashMap;
31 import java.util.List;
32 import java.util.Map;
33 import java.util.Map.Entry;
34 import org.apache.commons.lang3.StringUtils;
35 import org.jgrapht.DirectedGraph;
36 import org.jgrapht.graph.ClassBasedEdgeFactory;
37 import org.jgrapht.graph.DefaultEdge;
38 import org.jgrapht.graph.DirectedMultigraph;
39 import org.onap.policy.controlloop.policy.ControlLoop;
40 import org.onap.policy.controlloop.policy.ControlLoopPolicy;
41 import org.onap.policy.controlloop.policy.FinalResult;
42 import org.onap.policy.controlloop.policy.Policy;
43 import org.onap.policy.controlloop.policy.PolicyResult;
44 import org.onap.policy.controlloop.policy.TargetType;
45 import org.slf4j.Logger;
46 import org.slf4j.LoggerFactory;
47 import org.yaml.snakeyaml.Yaml;
48 import org.yaml.snakeyaml.constructor.Constructor;
49
50
51 public class ControlLoopCompiler implements Serializable {
52     private static final String OPERATION_POLICY = "Operation Policy ";
53     private static final long serialVersionUID = 1L;
54     private static final Logger LOGGER = LoggerFactory.getLogger(ControlLoopCompiler.class.getName());
55
56     /**
57      * Compiles the policy from an object.
58      */
59     public static ControlLoopPolicy compile(ControlLoopPolicy policy,
60                     ControlLoopCompilerCallback callback) throws CompilerException {
61         //
62         // Ensure the control loop is sane
63         //
64         validateControlLoop(policy.getControlLoop(), callback);
65         //
66         // Validate the policies
67         //
68         validatePolicies(policy, callback);
69
70         return policy;
71     }
72
73     /**
74      * Compiles the policy from an input stream.
75      *
76      * @param yamlSpecification the yaml input stream
77      * @param callback method to callback during compilation
78      * @return Control Loop object
79      * @throws CompilerException throws any compile exception found
80      */
81     public static ControlLoopPolicy compile(InputStream yamlSpecification,
82                     ControlLoopCompilerCallback callback) throws CompilerException {
83         Yaml yaml = new Yaml(new Constructor(ControlLoopPolicy.class));
84         Object obj = yaml.load(yamlSpecification);
85         if (obj == null) {
86             throw new CompilerException("Could not parse yaml specification.");
87         }
88         if (! (obj instanceof ControlLoopPolicy)) {
89             throw new CompilerException("Yaml could not parse specification into required ControlLoopPolicy object");
90         }
91         return ControlLoopCompiler.compile((ControlLoopPolicy) obj, callback);
92     }
93
94     private static void validateControlLoop(ControlLoop controlLoop,
95                     ControlLoopCompilerCallback callback) throws CompilerException {
96         if (controlLoop == null && callback != null) {
97             callback.onError("controlLoop cannot be null");
98         }
99         if (controlLoop != null) {
100             if (StringUtils.isEmpty(controlLoop.getControlLoopName()) && callback != null) {
101                 callback.onError("Missing controlLoopName");
102             }
103             if ((!controlLoop.getVersion().contentEquals(ControlLoop.getCompilerVersion())) && callback != null) {
104                 callback.onError("Unsupported version for this compiler");
105             }
106             if (StringUtils.isEmpty(controlLoop.getTrigger_policy())) {
107                 throw new CompilerException("trigger_policy is not valid");
108             }
109         }
110     }
111
112     private static void validatePolicies(ControlLoopPolicy policy,
113                     ControlLoopCompilerCallback callback) throws CompilerException {
114         if (policy == null) {
115             throw new CompilerException("policy cannot be null");
116         }
117         if (policy.getPolicies() == null) {
118             callback.onWarning("controlLoop is an open loop.");
119         } else {
120             //
121             // For this version we can use a directed multigraph, in the future we may not be able to
122             //
123             DirectedGraph<NodeWrapper, LabeledEdge> graph =
124                             new DirectedMultigraph<>(new ClassBasedEdgeFactory<NodeWrapper,
125                                             LabeledEdge>(LabeledEdge.class));
126             //
127             // Check to see if the trigger Event is for OpenLoop, we do so by
128             // attempting to create a FinalResult object from it. If its a policy id, this should
129             // return null.
130             //
131             FinalResult triggerResult = FinalResult.toResult(policy.getControlLoop().getTrigger_policy());
132             TriggerNodeWrapper triggerNode;
133             //
134             // Did this turn into a FinalResult object?
135             //
136             if (triggerResult != null) {
137                 validateOpenLoopPolicy(policy, triggerResult, callback);
138                 return;
139                 //
140             } else {
141                 validatePoliciesContainTriggerPolicyAndCombinedTimeoutIsOk(policy, callback);
142                 triggerNode = new TriggerNodeWrapper(policy.getControlLoop().getControlLoopName());
143             }
144             //
145             // Add in the trigger node
146             //
147             graph.addVertex(triggerNode);
148             //
149             // Add in our Final Result nodes. All paths should end to these nodes.
150             //
151             FinalResultNodeWrapper finalSuccess = new FinalResultNodeWrapper(FinalResult.FINAL_SUCCESS);
152             FinalResultNodeWrapper finalFailure = new FinalResultNodeWrapper(FinalResult.FINAL_FAILURE);
153             FinalResultNodeWrapper finalFailureTimeout = new FinalResultNodeWrapper(FinalResult.FINAL_FAILURE_TIMEOUT);
154             FinalResultNodeWrapper finalFailureRetries = new FinalResultNodeWrapper(FinalResult.FINAL_FAILURE_RETRIES);
155             FinalResultNodeWrapper finalFailureException =
156                             new FinalResultNodeWrapper(FinalResult.FINAL_FAILURE_EXCEPTION);
157             FinalResultNodeWrapper finalFailureGuard = new FinalResultNodeWrapper(FinalResult.FINAL_FAILURE_GUARD);
158             graph.addVertex(finalSuccess);
159             graph.addVertex(finalFailure);
160             graph.addVertex(finalFailureTimeout);
161             graph.addVertex(finalFailureRetries);
162             graph.addVertex(finalFailureException);
163             graph.addVertex(finalFailureGuard);
164             //
165             // Work through the policies and add them in as nodes.
166             //
167             Map<Policy, PolicyNodeWrapper> mapNodes = addPoliciesAsNodes(policy, graph, triggerNode, callback);
168             //
169             // last sweep to connect remaining edges for policy results
170             //
171             for (Policy operPolicy : policy.getPolicies()) {
172                 PolicyNodeWrapper node = mapNodes.get(operPolicy);
173                 //
174                 // Just ensure this has something
175                 //
176                 if (node == null) {
177                     continue;
178                 }
179                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getSuccess(), finalSuccess,
180                                 PolicyResult.SUCCESS, node);
181                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getFailure(), finalFailure,
182                                 PolicyResult.FAILURE, node);
183                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getFailure_timeout(), finalFailureTimeout,
184                                 PolicyResult.FAILURE_TIMEOUT, node);
185                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getFailure_retries(), finalFailureRetries,
186                                 PolicyResult.FAILURE_RETRIES, node);
187                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getFailure_exception(), finalFailureException,
188                                 PolicyResult.FAILURE_EXCEPTION, node);
189                 addEdge(graph, mapNodes, operPolicy.getId(), operPolicy.getFailure_guard(), finalFailureGuard,
190                                 PolicyResult.FAILURE_GUARD, node);
191             }
192             validateNodesAndEdges(graph, callback);
193         }
194     }
195
196     private static void validateOpenLoopPolicy(ControlLoopPolicy policy, FinalResult triggerResult,
197                     ControlLoopCompilerCallback callback) throws CompilerException {
198         //
199         // Ensure they didn't use some other FinalResult code
200         //
201         if (triggerResult != FinalResult.FINAL_OPENLOOP) {
202             throw new CompilerException("Unexpected Final Result for trigger_policy, should only be "
203         + FinalResult.FINAL_OPENLOOP.toString() + " or a valid Policy ID");
204         }
205         //
206         // They really shouldn't have any policies attached.
207         //
208         if ((policy.getPolicies() != null || policy.getPolicies().isEmpty()) && callback != null ) {
209             callback.onWarning("Open Loop policy contains policies. The policies will never be invoked.");
210         }
211     }
212
213     private static void validatePoliciesContainTriggerPolicyAndCombinedTimeoutIsOk(ControlLoopPolicy policy,
214                     ControlLoopCompilerCallback callback) throws CompilerException {
215         int sum = 0;
216         boolean triggerPolicyFound = false;
217         for (Policy operPolicy : policy.getPolicies()) {
218             sum += operPolicy.getTimeout().intValue();
219             if (policy.getControlLoop().getTrigger_policy().equals(operPolicy.getId())) {
220                 triggerPolicyFound = true;
221             }
222         }
223         if (policy.getControlLoop().getTimeout().intValue() < sum && callback != null) {
224             callback.onError("controlLoop overall timeout is less than the sum of operational policy timeouts.");
225         }
226
227         if (!triggerPolicyFound) {
228             throw new CompilerException("Unexpected value for trigger_policy, should only be "
229         + FinalResult.FINAL_OPENLOOP.toString() + " or a valid Policy ID");
230         }
231     }
232
233     private static Map<Policy, PolicyNodeWrapper> addPoliciesAsNodes(ControlLoopPolicy policy,
234             DirectedGraph<NodeWrapper, LabeledEdge> graph, TriggerNodeWrapper triggerNode,
235             ControlLoopCompilerCallback callback) {
236         Map<Policy, PolicyNodeWrapper> mapNodes = new HashMap<>();
237         for (Policy operPolicy : policy.getPolicies()) {
238             //
239             // Is it still ok to add?
240             //
241             if (!okToAdd(operPolicy, callback)) {
242                 //
243                 // Do not add it in
244                 //
245                 continue;
246             }
247             //
248             // Create wrapper policy node and save it into our map so we can
249             // easily retrieve it.
250             //
251             PolicyNodeWrapper node = new PolicyNodeWrapper(operPolicy);
252             mapNodes.put(operPolicy, node);
253             graph.addVertex(node);
254             //
255             // Is this the trigger policy?
256             //
257             if (operPolicy.getId().equals(policy.getControlLoop().getTrigger_policy())) {
258                 //
259                 // Yes add an edge from our trigger event node to this policy
260                 //
261                 graph.addEdge(triggerNode, node, new LabeledEdge(triggerNode, node, new TriggerEdgeWrapper("ONSET")));
262             }
263         }
264         return mapNodes;
265     }
266
267     private static void addEdge(DirectedGraph<NodeWrapper, LabeledEdge> graph, Map<Policy, PolicyNodeWrapper> mapNodes,
268                     String policyId, String connectedPolicy,
269                     FinalResultNodeWrapper finalResultNodeWrapper,
270                     PolicyResult policyResult, NodeWrapper node) throws CompilerException {
271         FinalResult finalResult = FinalResult.toResult(finalResultNodeWrapper.getId());
272         if (FinalResult.isResult(connectedPolicy, finalResult)) {
273             graph.addEdge(node, finalResultNodeWrapper, new LabeledEdge(node, finalResultNodeWrapper,
274                             new FinalResultEdgeWrapper(finalResult)));
275         } else {
276             PolicyNodeWrapper toNode = findPolicyNode(mapNodes, connectedPolicy);
277             if (toNode == null) {
278                 throw new CompilerException(OPERATION_POLICY + policyId + " is connected to unknown policy "
279             + connectedPolicy);
280             } else {
281                 graph.addEdge(node, toNode, new LabeledEdge(node, toNode, new PolicyResultEdgeWrapper(policyResult)));
282             }
283         }
284     }
285
286     private static void validateNodesAndEdges(DirectedGraph<NodeWrapper, LabeledEdge> graph,
287                     ControlLoopCompilerCallback callback) throws CompilerException {
288         for (NodeWrapper node : graph.vertexSet()) {
289             if (node instanceof TriggerNodeWrapper) {
290                 validateTriggerNodeWrapper(graph, node);
291             } else if (node instanceof FinalResultNodeWrapper) {
292                 validateFinalResultNodeWrapper(graph, node);
293             } else if (node instanceof PolicyNodeWrapper) {
294                 validatePolicyNodeWrapper(graph, node, callback);
295             }
296             for (LabeledEdge edge : graph.outgoingEdgesOf(node)) {
297                 LOGGER.info("{} invokes {} upon {}", edge.from.getId(), edge.to.getId(), edge.edge.getId());
298             }
299         }
300     }
301
302     private static void validateTriggerNodeWrapper(DirectedGraph<NodeWrapper, LabeledEdge> graph,
303                     NodeWrapper node) throws CompilerException {
304         if (LOGGER.isDebugEnabled()) {
305             LOGGER.info("Trigger Node {}", node);
306         }
307         if (graph.inDegreeOf(node) > 0 ) {
308             //
309             // Really should NEVER get here unless someone messed up the code above.
310             //
311             throw new CompilerException("No inputs to event trigger");
312         }
313         //
314         // Should always be 1, except in the future we may support multiple events
315         //
316         if (graph.outDegreeOf(node) > 1) {
317             throw new CompilerException("The event trigger should only go to ONE node");
318         }
319     }
320
321     private static void validateFinalResultNodeWrapper(DirectedGraph<NodeWrapper, LabeledEdge> graph,
322                     NodeWrapper node) throws CompilerException {
323         if (LOGGER.isDebugEnabled()) {
324             LOGGER.info("FinalResult Node {}", node);
325         }
326         //
327         // FinalResult nodes should NEVER have an out edge
328         //
329         if (graph.outDegreeOf(node) > 0) {
330             throw new CompilerException("FinalResult nodes should never have any out edges.");
331         }
332     }
333
334     private static void validatePolicyNodeWrapper(DirectedGraph<NodeWrapper, LabeledEdge> graph,
335                     NodeWrapper node, ControlLoopCompilerCallback callback) throws CompilerException {
336         if (LOGGER.isDebugEnabled()) {
337             LOGGER.info("Policy Node {}", node);
338         }
339         //
340         // All Policy Nodes should have the 5 out degrees defined.
341         //
342         if (graph.outDegreeOf(node) != 6) {
343             throw new CompilerException("Policy node should ALWAYS have 6 out degrees.");
344         }
345         //
346         // All Policy Nodes should have at least 1 in degrees
347         //
348         if (graph.inDegreeOf(node) == 0 && callback != null) {
349             callback.onWarning("Policy " + node.getId() + " is not reachable.");
350         }
351     }
352
353     private static boolean okToAdd(Policy operPolicy, ControlLoopCompilerCallback callback) {
354         boolean isOk = isPolicyIdOk(operPolicy, callback);
355         if (! isActorOk(operPolicy, callback)) {
356             isOk = false;
357         }
358         if (! isRecipeOk(operPolicy, callback)) {
359             isOk = false;
360         }
361         if (! isTargetOk(operPolicy, callback) ) {
362             isOk = false;
363         }
364         if (! arePolicyResultsOk(operPolicy, callback) ) {
365             isOk = false;
366         }
367         return isOk;
368     }
369
370     private static boolean isPolicyIdOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
371         boolean isOk = true;
372         if (operPolicy.getId() == null || operPolicy.getId().length() < 1) {
373             if (callback != null) {
374                 callback.onError("Operational Policy has an bad ID");
375             }
376             isOk = false;
377         } else {
378             //
379             // Check if they decided to make the ID a result object
380             //
381             if (PolicyResult.toResult(operPolicy.getId()) != null) {
382                 if (callback != null) {
383                     callback.onError("Policy id is set to a PolicyResult " + operPolicy.getId());
384                 }
385                 isOk = false;
386             }
387             if (FinalResult.toResult(operPolicy.getId()) != null) {
388                 if (callback != null) {
389                     callback.onError("Policy id is set to a FinalResult " + operPolicy.getId());
390                 }
391                 isOk = false;
392             }
393         }
394         return isOk;
395     }
396
397     private static boolean isActorOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
398         if (StringUtils.isBlank(operPolicy.getActor())) {
399             if (callback != null) {
400                 callback.onError("Policy actor is null");
401             }
402             return false;
403         }
404         return true;
405     }
406
407     private static boolean isRecipeOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
408         if (StringUtils.isBlank(operPolicy.getRecipe())) {
409             if (callback != null) {
410                 callback.onError("Policy recipe is null");
411             }
412             return false;
413         }
414         return true;
415     }
416
417     private static boolean isTargetOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
418         boolean isOk = true;
419         if (operPolicy.getTarget() == null) {
420             if (callback != null) {
421                 callback.onError("Policy target is null");
422             }
423             isOk = false;
424         }
425         if (operPolicy.getTarget() != null
426                         && operPolicy.getTarget().getType() != TargetType.VM
427                         && operPolicy.getTarget().getType() != TargetType.VFC
428                         && operPolicy.getTarget().getType() != TargetType.PNF) {
429             if (callback != null) {
430                 callback.onError("Policy target is invalid");
431             }
432             isOk = false;
433         }
434         return isOk;
435     }
436
437     private static boolean arePolicyResultsOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
438         //
439         // Check that policy results are connected to either default final * or another policy
440         //
441         boolean isOk = isSuccessPolicyResultOk(operPolicy, callback);
442         if (! isFailurePolicyResultOk(operPolicy, callback) ) {
443             isOk = false;
444         }
445         if (! isFailureRetriesPolicyResultOk(operPolicy, callback) ) {
446             isOk = false;
447         }
448         if (! isFailureTimeoutPolicyResultOk(operPolicy, callback) ) {
449             isOk = false;
450         }
451         if (! isFailureExceptionPolicyResultOk(operPolicy, callback) ) {
452             isOk = false;
453         }
454         if (! isFailureGuardPolicyResultOk(operPolicy, callback) ) {
455             isOk = false;
456         }
457         return isOk;
458     }
459
460     private static boolean isSuccessPolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
461         if (FinalResult.toResult(operPolicy.getSuccess()) != null
462                         && !operPolicy.getSuccess().equals(FinalResult.FINAL_SUCCESS.toString())) {
463             if (callback != null) {
464                 callback.onError("Policy success is neither another policy nor FINAL_SUCCESS");
465             }
466             return false;
467         }
468         return true;
469     }
470
471     private static boolean isFailurePolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
472         if (FinalResult.toResult(operPolicy.getFailure()) != null
473                         && !operPolicy.getFailure().equals(FinalResult.FINAL_FAILURE.toString())) {
474             if (callback != null) {
475                 callback.onError("Policy failure is neither another policy nor FINAL_FAILURE");
476             }
477             return false;
478         }
479         return true;
480     }
481
482     private static boolean isFailureRetriesPolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
483         if (FinalResult.toResult(operPolicy.getFailure_retries()) != null
484                         && !operPolicy.getFailure_retries().equals(FinalResult.FINAL_FAILURE_RETRIES.toString())) {
485             if (callback != null) {
486                 callback.onError("Policy failure retries is neither another policy nor FINAL_FAILURE_RETRIES");
487             }
488             return false;
489         }
490         return true;
491     }
492
493     private static boolean isFailureTimeoutPolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
494         if (FinalResult.toResult(operPolicy.getFailure_timeout()) != null
495                         && !operPolicy.getFailure_timeout().equals(FinalResult.FINAL_FAILURE_TIMEOUT.toString())) {
496             if (callback != null) {
497                 callback.onError("Policy failure timeout is neither another policy nor FINAL_FAILURE_TIMEOUT");
498             }
499             return false;
500         }
501         return true;
502     }
503
504     private static boolean isFailureExceptionPolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
505         if (FinalResult.toResult(operPolicy.getFailure_exception()) != null
506                         && !operPolicy.getFailure_exception().equals(FinalResult.FINAL_FAILURE_EXCEPTION.toString())) {
507             if (callback != null) {
508                 callback.onError("Policy failure exception is neither another policy nor FINAL_FAILURE_EXCEPTION");
509             }
510             return false;
511         }
512         return true;
513     }
514
515     private static boolean isFailureGuardPolicyResultOk(Policy operPolicy, ControlLoopCompilerCallback callback) {
516         if (FinalResult.toResult(operPolicy.getFailure_guard()) != null
517                         && !operPolicy.getFailure_guard().equals(FinalResult.FINAL_FAILURE_GUARD.toString())) {
518             if (callback != null) {
519                 callback.onError("Policy failure guard is neither another policy nor FINAL_FAILURE_GUARD");
520             }
521             return false;
522         }
523         return true;
524     }
525
526     private static PolicyNodeWrapper findPolicyNode(Map<Policy, PolicyNodeWrapper> mapNodes, String id) {
527         for (Entry<Policy, PolicyNodeWrapper> entry : mapNodes.entrySet()) {
528             if (entry.getKey().getId().equals(id)) {
529                 return entry.getValue();
530             }
531         }
532         return null;
533     }
534
535     @FunctionalInterface
536     private interface NodeWrapper extends Serializable {
537         public String   getId();
538     }
539
540     private static class TriggerNodeWrapper implements NodeWrapper {
541         private static final long serialVersionUID = -187644087811478349L;
542         private String closedLoopControlName;
543
544         public TriggerNodeWrapper(String closedLoopControlName) {
545             this.closedLoopControlName = closedLoopControlName;
546         }
547
548         @Override
549         public String toString() {
550             return "TriggerNodeWrapper [closedLoopControlName=" + closedLoopControlName + "]";
551         }
552
553         @Override
554         public String getId() {
555             return closedLoopControlName;
556         }
557
558     }
559
560     private static class FinalResultNodeWrapper implements NodeWrapper {
561         private static final long serialVersionUID = 8540008796302474613L;
562         private FinalResult result;
563
564         public FinalResultNodeWrapper(FinalResult result) {
565             this.result = result;
566         }
567
568         @Override
569         public String toString() {
570             return "FinalResultNodeWrapper [result=" + result + "]";
571         }
572
573         @Override
574         public String getId() {
575             return result.toString();
576         }
577     }
578
579     private static class PolicyNodeWrapper implements NodeWrapper {
580         private static final long serialVersionUID = 8170162175653823082L;
581         private transient Policy policy;
582
583         public PolicyNodeWrapper(Policy operPolicy) {
584             this.policy = operPolicy;
585         }
586
587         @Override
588         public String toString() {
589             return "PolicyNodeWrapper [policy=" + policy + "]";
590         }
591
592         @Override
593         public String getId() {
594             return policy.getId();
595         }
596     }
597
598     @FunctionalInterface
599     private interface EdgeWrapper extends Serializable {
600         public String getId();
601
602     }
603
604     private static class TriggerEdgeWrapper implements EdgeWrapper {
605         private static final long serialVersionUID = 2678151552623278863L;
606         private String trigger;
607
608         public TriggerEdgeWrapper(String trigger) {
609             this.trigger = trigger;
610         }
611
612         @Override
613         public String getId() {
614             return trigger;
615         }
616
617         @Override
618         public String toString() {
619             return "TriggerEdgeWrapper [trigger=" + trigger + "]";
620         }
621
622     }
623
624     private static class PolicyResultEdgeWrapper implements EdgeWrapper {
625         private static final long serialVersionUID = 6078569477021558310L;
626         private PolicyResult policyResult;
627
628         public PolicyResultEdgeWrapper(PolicyResult policyResult) {
629             super();
630             this.policyResult = policyResult;
631         }
632
633         @Override
634         public String toString() {
635             return "PolicyResultEdgeWrapper [policyResult=" + policyResult + "]";
636         }
637
638         @Override
639         public String getId() {
640             return policyResult.toString();
641         }
642
643
644     }
645
646     private static class FinalResultEdgeWrapper implements EdgeWrapper {
647         private static final long serialVersionUID = -1486381946896779840L;
648         private FinalResult finalResult;
649
650         public FinalResultEdgeWrapper(FinalResult result) {
651             this.finalResult = result;
652         }
653
654         @Override
655         public String toString() {
656             return "FinalResultEdgeWrapper [finalResult=" + finalResult + "]";
657         }
658
659         @Override
660         public String getId() {
661             return finalResult.toString();
662         }
663     }
664
665
666     private static class LabeledEdge extends DefaultEdge {
667         private static final long serialVersionUID = 579384429573385524L;
668
669         private NodeWrapper from;
670         private NodeWrapper to;
671         private EdgeWrapper edge;
672
673         public LabeledEdge(NodeWrapper from, NodeWrapper to, EdgeWrapper edge) {
674             this.from = from;
675             this.to = to;
676             this.edge = edge;
677         }
678
679         @SuppressWarnings("unused")
680         public NodeWrapper from() {
681             return from;
682         }
683
684         @SuppressWarnings("unused")
685         public NodeWrapper to() {
686             return to;
687         }
688
689         @SuppressWarnings("unused")
690         public EdgeWrapper edge() {
691             return edge;
692         }
693
694         @Override
695         public String toString() {
696             return "LabeledEdge [from=" + from + ", to=" + to + ", edge=" + edge + "]";
697         }
698     }
699
700 }