gerrit.onap Code Review - policy/apex-pdp.git/blob

   1 /*-
   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
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  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.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.apex.tools.model.generator.model2event;
  22
  23 import java.util.HashSet;
  24 import java.util.Properties;
  25 import java.util.Set;
  26
  27 import org.apache.avro.Schema;
  28 import org.apache.avro.Schema.Field;
  29 import org.apache.commons.lang3.Validate;
  30 import org.onap.policy.apex.context.parameters.SchemaParameters;
  31 import org.onap.policy.apex.model.basicmodel.concepts.ApexException;
  32 import org.onap.policy.apex.model.basicmodel.concepts.AxArtifactKey;
  33 import org.onap.policy.apex.model.eventmodel.concepts.AxEvent;
  34 import org.onap.policy.apex.model.modelapi.ApexAPIResult;
  35 import org.onap.policy.apex.model.modelapi.ApexModel;
  36 import org.onap.policy.apex.model.modelapi.ApexModelFactory;
  37 import org.onap.policy.apex.model.policymodel.concepts.AxPolicies;
  38 import org.onap.policy.apex.model.policymodel.concepts.AxPolicy;
  39 import org.onap.policy.apex.model.policymodel.concepts.AxPolicyModel;
  40 import org.onap.policy.apex.model.policymodel.concepts.AxState;
  41 import org.onap.policy.apex.model.policymodel.concepts.AxStateOutput;
  42 import org.onap.policy.apex.plugins.context.schema.avro.AvroSchemaHelperParameters;
  43 import org.onap.policy.apex.tools.model.generator.SchemaUtils;
  44 import org.stringtemplate.v4.ST;
  45 import org.stringtemplate.v4.STGroup;
  46 import org.stringtemplate.v4.STGroupFile;
  47
  48 /**
  49  * Takes a model and generates the JSON event schemas.
  50  *
  51  * @author Sven van der Meer (sven.van.der.meer@ericsson.com)
  52  */
  53 public class Model2JsonEventSchema {
  54
  55     /** Application name, used as prompt. */
  56     private final String appName;
  57
  58     /** The file name of the policy model. */
  59     private final String modelFile;
  60
  61     /** The type of events to generate: stimuli, response, internal. */
  62     private final String type;
  63
  64     /**
  65      * Creates a new model to event schema generator.
  66      *
  67      * @param modelFile the model file to be used
  68      * @param type the type of events to generate, one of: stimuli, response, internal
  69      * @param appName application name for printouts
  70      */
  71     public Model2JsonEventSchema(final String modelFile, final String type, final String appName) {
  72         Validate.notNull(modelFile, "Model2JsonEvent: given model file name was blank");
  73         Validate.notNull(type, "Model2JsonEvent: given type was blank");
  74         Validate.notNull(appName, "Model2JsonEvent: given application name was blank");
  75         this.modelFile = modelFile;
  76         this.type = type;
  77         this.appName = appName;
  78     }
  79
  80     /**
  81      * Adds a type to a field for a given schema.
  82      *
  83      * @param schema the schema to add a type for
  84      * @param stg the STG
  85      * @return a template with the type
  86      */
  87     protected ST addFieldType(final Schema schema, final STGroup stg) {
  88         ST ret = null;
  89         switch (schema.getType()) {
  90             case BOOLEAN:
  91             case BYTES:
  92             case DOUBLE:
  93             case FIXED:
  94             case FLOAT:
  95             case INT:
  96             case LONG:
  97             case STRING:
  98                 ret = stg.getInstanceOf("fieldTypeAtomic");
  99                 ret.add("type", schema.getType());
 100                 break;
 101
 102             case ARRAY:
 103                 ret = stg.getInstanceOf("fieldTypeArray");
 104                 ret.add("array", this.addFieldType(schema.getElementType(), stg));
 105                 break;
 106             case ENUM:
 107                 ret = stg.getInstanceOf("fieldTypeEnum");
 108                 ret.add("symbols", schema.getEnumSymbols());
 109                 break;
 110
 111             case MAP:
 112                 ret = stg.getInstanceOf("fieldTypeMap");
 113                 ret.add("map", this.addFieldType(schema.getValueType(), stg));
 114                 break;
 115
 116             case RECORD:
 117                 ret = stg.getInstanceOf("fieldTypeRecord");
 118                 for (final Field field : schema.getFields()) {
 119                     final ST st = stg.getInstanceOf("field");
 120                     st.add("name", field.name());
 121                     st.add("type", this.addFieldType(field.schema(), stg));
 122                     ret.add("fields", st);
 123                 }
 124                 break;
 125
 126             case NULL:
 127                 break;
 128             case UNION:
 129                 break;
 130             default:
 131                 break;
 132         }
 133         return ret;
 134     }
 135
 136     /**
 137      * Runs the application.
 138      *
 139      *
 140      * @return status of the application execution, 0 for success, positive integer for exit
 141      *         condition (such as help or version), negative integer for errors
 142      * @throws ApexException if any problem occurred in the model
 143      */
 144     public int runApp() throws ApexException {
 145         final STGroupFile stg = new STGroupFile("org/onap/policy/apex/tools/model/generator/event-json.stg");
 146         final ST stEvents = stg.getInstanceOf("events");
 147
 148         final ApexModelFactory factory = new ApexModelFactory();
 149         final ApexModel model = factory.createApexModel(new Properties(), true);
 150
 151         final ApexAPIResult result = model.loadFromFile(modelFile);
 152         if (result.isNOK()) {
 153             System.err.println(appName + ": " + result.getMessage());
 154             return -1;
 155         }
 156
 157         final AxPolicyModel policyModel = model.getPolicyModel();
 158         policyModel.register();
 159         new SchemaParameters().getSchemaHelperParameterMap().put("Avro", new AvroSchemaHelperParameters());
 160
 161         final Set<AxEvent> events = new HashSet<>();
 162         final Set<AxArtifactKey> eventKeys = new HashSet<>();
 163         final AxPolicies policies = policyModel.getPolicies();
 164         switch (type) {
 165             case "stimuli":
 166                 for (final AxPolicy policy : policies.getPolicyMap().values()) {
 167                     final String firsState = policy.getFirstState();
 168                     for (final AxState state : policy.getStateMap().values()) {
 169                         if (state.getKey().getLocalName().equals(firsState)) {
 170                             eventKeys.add(state.getTrigger());
 171                         }
 172                     }
 173                 }
 174                 break;
 175             case "response":
 176                 for (final AxPolicy policy : policies.getPolicyMap().values()) {
 177                     for (final AxState state : policy.getStateMap().values()) {
 178                         if (state.getNextStateSet().iterator().next().equals("NULL")) {
 179                             for (final AxStateOutput output : state.getStateOutputs().values()) {
 180                                 eventKeys.add(output.getOutgingEvent());
 181                             }
 182                         }
 183                     }
 184                 }
 185                 break;
 186             case "internal":
 187                 for (final AxPolicy policy : policies.getPolicyMap().values()) {
 188                     final String firsState = policy.getFirstState();
 189                     for (final AxState state : policy.getStateMap().values()) {
 190                         if (state.getKey().getLocalName().equals(firsState)) {
 191                             continue;
 192                         }
 193                         if (state.getNextStateSet().iterator().next().equals("NULL")) {
 194                             continue;
 195                         }
 196                         for (final AxStateOutput output : state.getStateOutputs().values()) {
 197                             eventKeys.add(output.getOutgingEvent());
 198                         }
 199                     }
 200                 }
 201                 break;
 202             default:
 203                 System.err.println(appName + ": unknown type <" + type + ">, cannot proceed");
 204                 return -1;
 205         }
 206
 207         for (final AxEvent event : policyModel.getEvents().getEventMap().values()) {
 208             for (final AxArtifactKey key : eventKeys) {
 209                 if (event.getKey().equals(key)) {
 210                     events.add(event);
 211                 }
 212             }
 213         }
 214
 215         for (final AxEvent event : events) {
 216             final ST stEvent = stg.getInstanceOf("event");
 217             stEvent.add("name", event.getKey().getName());
 218             stEvent.add("nameSpace", event.getNameSpace());
 219             stEvent.add("version", event.getKey().getVersion());
 220             stEvent.add("source", event.getSource());
 221             stEvent.add("target", event.getTarget());
 222
 223             final Schema avro = SchemaUtils.getEventSchema(event);
 224             for (final Field field : avro.getFields()) {
 225                 // filter magic names
 226                 switch (field.name()) {
 227                     case "name":
 228                     case "nameSpace":
 229                     case "version":
 230                     case "source":
 231                     case "target":
 232                         break;
 233                     default:
 234                         stEvent.add("fields", this.setField(field, stg));
 235                 }
 236             }
 237             stEvents.add("event", stEvent);
 238         }
 239         System.err.println(stEvents.render());
 240         return 0;
 241     }
 242
 243     /**
 244      * Adds a field to the output.
 245      *
 246      * @param field the field from the event
 247      * @param stg the STG
 248      * @return a template for the field
 249      */
 250     protected ST setField(final Field field, final STGroup stg) {
 251         final ST st = stg.getInstanceOf("field");
 252         switch (field.name()) {
 253             case "name":
 254             case "nameSpace":
 255             case "version":
 256             case "source":
 257             case "target":
 258                 break;
 259             default:
 260                 st.add("name", field.name());
 261                 st.add("type", this.addFieldType(field.schema(), stg));
 262         }
 263         return st;
 264     }
 265
 266 }