summaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/aai/sparky/config/oxm/OxmEntityLookup.java
blob: 168a4b14b891d345c1fe2cc9169984eb04c4b5b8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017 Amdocs
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 *
 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 */
package org.onap.aai.sparky.config.oxm;

import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.eclipse.persistence.dynamic.DynamicType;
import org.eclipse.persistence.internal.oxm.mappings.Descriptor;
import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;

public class OxmEntityLookup implements OxmModelProcessor {

  // TODO: kill singleton collaborator pattern
  private static OxmEntityLookup instance;

  private Map<String, HashMap<String, String>> oxmModel;

  private Map<String, DynamicType> entityTypeLookup;

  private Map<String, OxmEntityDescriptor> entityDescriptors;


  private OxmEntityLookup() {
    oxmModel = new LinkedHashMap<String, HashMap<String, String>>();
    entityTypeLookup = new LinkedHashMap<String, DynamicType>();
    entityDescriptors = new HashMap<String, OxmEntityDescriptor>();
  }

  public synchronized static OxmEntityLookup getInstance() {

    /*
     * I hate this method and I want it to go away. The singleton pattern is transitory, I want this
     * class to be wired via a bean reference instead. But from the starting point, it would require
     * fixing all the classes across the code base up front and I don't want this task to expand
     * beyond just refactoring the OxmModelLoader. For now I'll keep the singleton pattern, but I
     * really want to get rid of it once we are properly spring wired.
     */

    if (instance == null) {
      instance = new OxmEntityLookup();
    }

    return instance;
  }


  @Override
  public void processOxmModel(DynamicJAXBContext jaxbContext) {

    @SuppressWarnings("rawtypes")
    List<Descriptor> descriptorsList = jaxbContext.getXMLContext().getDescriptors();

    for (@SuppressWarnings("rawtypes")
    Descriptor desc : descriptorsList) {

      DynamicType entity = jaxbContext.getDynamicType(desc.getAlias());

      LinkedHashMap<String, String> oxmProperties = new LinkedHashMap<String, String>();

      // Not all fields have key attributes
      if (desc.getPrimaryKeyFields() != null) {
        oxmProperties.put("primaryKeyAttributeNames", desc.getPrimaryKeyFields().toString()
            .replaceAll("/text\\(\\)", "").replaceAll("\\[", "").replaceAll("\\]", ""));
      }

      String entityName = desc.getDefaultRootElement();

      entityTypeLookup.put(entityName, entity);

      // add entityName
      oxmProperties.put("entityName", entityName);

      Map<String, String> properties = entity.getDescriptor().getProperties();

      oxmModel.put(entityName, oxmProperties);

    }

    for (Entry<String, HashMap<String, String>> entityModel : oxmModel.entrySet()) {
      HashMap<String, String> attribute = entityModel.getValue();
      OxmEntityDescriptor entity = new OxmEntityDescriptor();

      entity.setEntityName(attribute.get("entityName"));

      if (attribute.containsKey("primaryKeyAttributeNames")) {

        entity.setPrimaryKeyAttributeNames(
            Arrays.asList(attribute.get("primaryKeyAttributeNames").replace(" ", "").split(",")));

        entityDescriptors.put(attribute.get("entityName"), entity);
      }
    }

  }

  public Map<String, HashMap<String, String>> getOxmModel() {
    return oxmModel;
  }

  public void setOxmModel(Map<String, HashMap<String, String>> oxmModel) {
    this.oxmModel = oxmModel;
  }

  public Map<String, DynamicType> getEntityTypeLookup() {
    return entityTypeLookup;
  }

  public void setEntityTypeLookup(Map<String, DynamicType> entityTypeLookup) {
    this.entityTypeLookup = entityTypeLookup;
  }

  public Map<String, OxmEntityDescriptor> getEntityDescriptors() {
    return entityDescriptors;
  }

  public void setEntityDescriptors(Map<String, OxmEntityDescriptor> entityDescriptors) {
    this.entityDescriptors = entityDescriptors;
  }

  public void addEntityDescriptor(String type, OxmEntityDescriptor descriptor) {
    if (this.entityDescriptors != null) {
      this.entityDescriptors.put(type, descriptor);
    }
  }

}