2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2019-2021 Nordix Foundation.
4 * Modifications Copyright (C) 2019-2021 AT&T Intellectual Property. All rights reserved.
5 * ================================================================================
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
18 * SPDX-License-Identifier: Apache-2.0
19 * ============LICENSE_END=========================================================
22 package org.onap.policy.models.dao.impl;
24 import java.sql.Timestamp;
25 import java.time.Instant;
26 import java.util.Collection;
27 import java.util.Collections;
28 import java.util.List;
30 import javax.persistence.EntityManager;
31 import javax.persistence.EntityManagerFactory;
32 import javax.persistence.Persistence;
33 import javax.persistence.TypedQuery;
34 import javax.ws.rs.core.Response;
35 import org.onap.policy.models.base.PfConcept;
36 import org.onap.policy.models.base.PfConceptKey;
37 import org.onap.policy.models.base.PfGeneratedIdKey;
38 import org.onap.policy.models.base.PfModelException;
39 import org.onap.policy.models.base.PfModelRuntimeException;
40 import org.onap.policy.models.base.PfReferenceKey;
41 import org.onap.policy.models.base.PfReferenceTimestampKey;
42 import org.onap.policy.models.base.PfTimestampKey;
43 import org.onap.policy.models.base.PfUtils;
44 import org.onap.policy.models.dao.DaoParameters;
45 import org.onap.policy.models.dao.PfDao;
46 import org.onap.policy.models.dao.PfFilter;
47 import org.onap.policy.models.dao.PfFilterFactory;
48 import org.slf4j.Logger;
49 import org.slf4j.LoggerFactory;
52 * The Class DefaultPfDao is an JPA implementation of the {@link PfDao} class for Policy Framework concepts
53 * ({@link PfConcept}). It uses the default JPA implementation in the javax {@link Persistence} class.
55 public class DefaultPfDao implements PfDao {
56 private static final Logger LOGGER = LoggerFactory.getLogger(DefaultPfDao.class);
59 private static final String NAME = "name";
60 private static final String VERSION = "version";
61 private static final String TIMESTAMP = "timeStamp";
62 private static final String GENERATEDID = "Id";
63 private static final String PARENT_NAME = "parentname";
64 private static final String PARENT_VERSION = "parentversion";
65 private static final String LOCAL_NAME = "localname";
67 private static final String TABLE_TOKEN = "__TABLE__";
69 private static final String DELETE_FROM_TABLE = "DELETE FROM __TABLE__ c";
71 private static final String SELECT_FROM_TABLE = "SELECT c FROM __TABLE__ c";
73 private static final String WHERE = " WHERE ";
74 private static final String AND = " AND ";
75 private static final String ORDER = " ORDER BY ";
77 private static final String NAME_FILTER = "c.key.name = :name";
78 private static final String VERSION_FILTER = "c.key.version = :version";
79 private static final String TIMESTAMP_FILTER = "c.key.timeStamp = :timeStamp";
80 private static final String TIMESTAMP_FILTER_NOKEY = "c.timeStamp = :timeStamp";
81 private static final String GENERATED_ID_FILTER = "c.key.generatedId = :Id";
82 private static final String PARENT_NAME_FILTER = "c.key.parentKeyName = :parentname";
83 private static final String PARENT_VERSION_FILTER = "c.key.parentKeyVersion = :parentversion";
84 private static final String LOCAL_NAME_FILTER = "c.key.localName = :localname";
86 private static final String PARENT_NAME_REF_FILTER = "c.key.referenceKey.parentKeyName = :parentKeyName";
88 private static final String CLONE_ERR_MSG = "Could not clone object of class \"{}\"";
90 private static final String DELETE_BY_CONCEPT_KEY =
91 DELETE_FROM_TABLE + WHERE + NAME_FILTER + AND + VERSION_FILTER;
93 private static final String DELETE_BY_TIMESTAMP_KEY =
94 DELETE_FROM_TABLE + WHERE + NAME_FILTER + AND + VERSION_FILTER + AND + TIMESTAMP_FILTER;
96 private static final String DELETE_BY_GENERATED_ID_KEY =
97 DELETE_FROM_TABLE + WHERE + NAME_FILTER + AND + VERSION_FILTER + AND + GENERATED_ID_FILTER;
99 private static final String DELETE_BY_REFERENCE_KEY =
100 DELETE_FROM_TABLE + WHERE + PARENT_NAME_FILTER + AND + PARENT_VERSION_FILTER + AND + LOCAL_NAME_FILTER;
102 private static final String SELECT_ALL_FOR_PARENT =
103 SELECT_FROM_TABLE + WHERE + PARENT_NAME_FILTER + AND + PARENT_VERSION_FILTER;
105 private static final String SELECT_ALL_VERSIONS_FOR_PARENT =
106 SELECT_FROM_TABLE + WHERE + PARENT_NAME_FILTER;
108 private static final String SELECT_ALL_VERSIONS = SELECT_FROM_TABLE + WHERE + NAME_FILTER;
110 private static final String SELECT_BY_CONCEPT_KEY =
111 SELECT_FROM_TABLE + WHERE + NAME_FILTER + AND + VERSION_FILTER;
113 private static final String SELECT_BY_TIMESTAMP_NOKEY =
114 SELECT_FROM_TABLE + WHERE + NAME_FILTER + AND + VERSION_FILTER + AND + TIMESTAMP_FILTER_NOKEY;
116 private static final String SELECT_BY_REFERENCE_KEY =
117 SELECT_FROM_TABLE + WHERE + PARENT_NAME_FILTER + AND + PARENT_VERSION_FILTER + AND + LOCAL_NAME_FILTER;
120 // Entity manager for JPA
121 private EntityManagerFactory emf = null;
124 public void init(final DaoParameters daoParameters) throws PfModelException {
125 if (daoParameters == null || daoParameters.getPersistenceUnit() == null) {
126 LOGGER.error("Policy Framework persistence unit parameter not set");
127 throw new PfModelException(Response.Status.INTERNAL_SERVER_ERROR,
128 "Policy Framework persistence unit parameter not set");
131 LOGGER.debug("Creating Policy Framework persistence unit \"{}\" . . .", daoParameters.getPersistenceUnit());
133 emf = Persistence.createEntityManagerFactory(daoParameters.getPersistenceUnit(),
134 daoParameters.getJdbcProperties());
135 } catch (final Exception ex) {
136 String errorMessage = "Creation of Policy Framework persistence unit \""
137 + daoParameters.getPersistenceUnit() + "\" failed";
138 LOGGER.warn(errorMessage);
139 throw new PfModelException(Response.Status.INTERNAL_SERVER_ERROR, errorMessage, ex);
141 LOGGER.debug("Created Policy Framework persistence unit \"{}\"", daoParameters.getPersistenceUnit());
145 * Gets the entity manager for this DAO.
147 * @return the entity manager
149 protected final synchronized EntityManager getEntityManager() {
151 LOGGER.warn("Policy Framework DAO has not been initialized");
152 throw new PfModelRuntimeException(Response.Status.INTERNAL_SERVER_ERROR,
153 "Policy Framework DAO has not been initialized");
156 return emf.createEntityManager();
160 public final void close() {
167 public <T extends PfConcept> void create(final T obj) {
171 final EntityManager mg = getEntityManager();
173 mg.getTransaction().begin();
175 mg.getTransaction().commit();
182 public <T extends PfConcept> void delete(final T obj) {
186 final EntityManager mg = getEntityManager();
188 mg.getTransaction().begin();
189 mg.remove(mg.contains(obj) ? obj : mg.merge(obj));
190 mg.getTransaction().commit();
197 public <T extends PfConcept> void delete(final Class<T> someClass, final PfConceptKey key) {
201 final EntityManager mg = getEntityManager();
204 mg.getTransaction().begin();
205 mg.createQuery(setQueryTable(DELETE_BY_CONCEPT_KEY, someClass), someClass)
206 .setParameter(NAME, key.getName())
207 .setParameter(VERSION, key.getVersion())
209 mg.getTransaction().commit();
217 public <T extends PfConcept> void delete(final Class<T> someClass, final PfReferenceKey key) {
221 final EntityManager mg = getEntityManager();
224 mg.getTransaction().begin();
225 mg.createQuery(setQueryTable(DELETE_BY_REFERENCE_KEY, someClass), someClass)
226 .setParameter(PARENT_NAME, key.getParentKeyName())
227 .setParameter(PARENT_VERSION, key.getParentKeyVersion())
228 .setParameter(LOCAL_NAME, key.getLocalName())
230 mg.getTransaction().commit();
238 public <T extends PfConcept> void delete(final Class<T> someClass, final PfTimestampKey key) {
242 final EntityManager mg = getEntityManager();
245 mg.getTransaction().begin();
246 mg.createQuery(setQueryTable(DELETE_BY_TIMESTAMP_KEY, someClass), someClass)
247 .setParameter(NAME, key.getName())
248 .setParameter(VERSION, key.getVersion())
249 .setParameter(TIMESTAMP, key.getTimeStamp())
251 mg.getTransaction().commit();
259 public <T extends PfConcept> void delete(final Class<T> someClass, final PfGeneratedIdKey key) {
263 final EntityManager mg = getEntityManager();
266 mg.getTransaction().begin();
267 mg.createQuery(setQueryTable(DELETE_BY_GENERATED_ID_KEY, someClass), someClass)
268 .setParameter(NAME, key.getName())
269 .setParameter(VERSION, key.getVersion())
270 .setParameter(GENERATEDID, key.getGeneratedId())
272 mg.getTransaction().commit();
280 public <T extends PfConcept> void createCollection(final Collection<T> objs) {
281 if (objs == null || objs.isEmpty()) {
284 final EntityManager mg = getEntityManager();
286 mg.getTransaction().begin();
287 for (final T t : objs) {
290 mg.getTransaction().commit();
297 public <T extends PfConcept> void deleteCollection(final Collection<T> objs) {
298 if (objs == null || objs.isEmpty()) {
301 final EntityManager mg = getEntityManager();
303 mg.getTransaction().begin();
304 for (final T t : objs) {
305 mg.remove(mg.contains(t) ? t : mg.merge(t));
307 mg.getTransaction().commit();
314 public <T extends PfConcept> int deleteByConceptKey(final Class<T> someClass, final Collection<PfConceptKey> keys) {
315 if (keys == null || keys.isEmpty()) {
318 int deletedCount = 0;
319 final EntityManager mg = getEntityManager();
322 mg.getTransaction().begin();
323 for (final PfConceptKey key : keys) {
324 deletedCount += mg.createQuery(setQueryTable(DELETE_BY_CONCEPT_KEY, someClass), someClass)
325 .setParameter(NAME, key.getName())
326 .setParameter(VERSION, key.getVersion())
329 mg.getTransaction().commit();
338 public <T extends PfConcept> int deleteByReferenceKey(final Class<T> someClass,
339 final Collection<PfReferenceKey> keys) {
340 if (keys == null || keys.isEmpty()) {
343 int deletedCount = 0;
344 final EntityManager mg = getEntityManager();
347 mg.getTransaction().begin();
348 for (final PfReferenceKey key : keys) {
349 deletedCount += mg.createQuery(setQueryTable(DELETE_BY_REFERENCE_KEY, someClass), someClass)
350 .setParameter(PARENT_NAME, key.getParentKeyName())
351 .setParameter(PARENT_VERSION, key.getParentKeyVersion())
352 .setParameter(LOCAL_NAME, key.getLocalName())
355 mg.getTransaction().commit();
364 public <T extends PfConcept> void deleteAll(final Class<T> someClass) {
365 final EntityManager mg = getEntityManager();
367 mg.getTransaction().begin();
368 mg.createQuery(setQueryTable(DELETE_FROM_TABLE, someClass), someClass).executeUpdate();
369 mg.getTransaction().commit();
376 public <T extends PfConcept> List<T> getFiltered(final Class<T> someClass, final String name,
377 final String version) {
379 return getAll(someClass);
382 if (version == null) {
383 return getAllVersions(someClass, name);
386 T foundConcept = get(someClass, new PfConceptKey(name, version));
388 return (foundConcept == null ? Collections.emptyList() : Collections.singletonList(foundConcept));
392 public <T extends PfConcept> List<T> getFiltered(final Class<T> someClass, final String name, final String version,
393 final Instant startTime, final Instant endTime, final Map<String, Object> filterMap, final String sortOrder,
394 final int getRecordNum) {
395 final EntityManager mg = getEntityManager();
397 String filterQueryString = SELECT_FROM_TABLE + WHERE;
400 PfFilter timeStampFilter = new PfFilterFactory().createFilter(someClass);
401 filterQueryString = timeStampFilter.addFilter(filterQueryString,
402 name, startTime, endTime, filterMap, sortOrder, getRecordNum);
405 TypedQuery<T> query = mg.createQuery(setQueryTable(filterQueryString, someClass), someClass);
407 if (filterMap != null) {
408 for (Map.Entry<String, Object> entry : filterMap.entrySet()) {
409 query.setParameter(entry.getKey(), entry.getValue());
413 query.setParameter(timeStampFilter.getNameParameter(), name);
415 if (startTime != null) {
416 if (endTime != null) {
417 query.setParameter("startTime", Timestamp.from(startTime));
418 query.setParameter("endTime", Timestamp.from(endTime));
420 query.setParameter("startTime", Timestamp.from(startTime));
423 if (endTime != null) {
424 query.setParameter("endTime", Timestamp.from(endTime));
427 if (getRecordNum > 0) {
428 query.setMaxResults(getRecordNum);
431 LOGGER.debug("filterQueryString is \"{}\"", filterQueryString);
432 return query.getResultList();
439 public <T extends PfConcept> T get(final Class<T> someClass, final PfConceptKey key) {
440 return genericGet(someClass, key);
444 public <T extends PfConcept> T get(final Class<T> someClass, final PfReferenceKey key) {
445 return genericGet(someClass, key);
449 public <T extends PfConcept> T get(final Class<T> someClass, final PfGeneratedIdKey key) {
450 return genericGet(someClass, key);
454 public <T extends PfConcept> T get(final Class<T> someClass, final PfTimestampKey key) {
455 return genericGet(someClass, key);
459 public <T extends PfConcept> T get(final Class<T> someClass, final PfReferenceTimestampKey key) {
460 return genericGet(someClass, key);
463 private <T extends PfConcept> T genericGet(final Class<T> someClass, final Object key) {
464 if (someClass == null) {
467 final EntityManager mg = getEntityManager();
469 final T t = mg.find(someClass, key);
473 return checkAndReturn(someClass, t);
480 public <T extends PfConcept> List<T> getAll(final Class<T> someClass) {
481 if (someClass == null) {
482 return Collections.emptyList();
484 final EntityManager mg = getEntityManager();
486 return mg.createQuery(setQueryTable(SELECT_FROM_TABLE, someClass), someClass).getResultList();
493 public <T extends PfConcept> List<T> getAll(final Class<T> someClass, final PfConceptKey parentKey) {
494 if (someClass == null) {
495 return Collections.emptyList();
497 final EntityManager mg = getEntityManager();
500 return mg.createQuery(setQueryTable(SELECT_ALL_FOR_PARENT, someClass), someClass)
501 .setParameter(PARENT_NAME, parentKey.getName())
502 .setParameter(PARENT_VERSION, parentKey.getVersion())
511 public <T extends PfConcept> List<T> getAllVersionsByParent(final Class<T> someClass, final String parentKeyName) {
512 if (someClass == null || parentKeyName == null) {
513 return Collections.emptyList();
515 final EntityManager mg = getEntityManager();
518 return mg.createQuery(setQueryTable(SELECT_ALL_VERSIONS_FOR_PARENT, someClass), someClass)
519 .setParameter(PARENT_NAME, parentKeyName)
528 public <T extends PfConcept> List<T> getAllVersions(final Class<T> someClass, final String conceptName) {
529 if (someClass == null || conceptName == null) {
530 return Collections.emptyList();
532 final EntityManager mg = getEntityManager();
535 return mg.createQuery(setQueryTable(SELECT_ALL_VERSIONS, someClass), someClass)
536 .setParameter(NAME, conceptName)
545 public <T extends PfConcept> List<T> getByTimestamp(final Class<T> someClass,
546 final PfGeneratedIdKey key,
547 final Instant timeStamp) {
548 if (someClass == null || key == null || timeStamp == null) {
549 return Collections.emptyList();
552 final EntityManager mg = getEntityManager();
555 return mg.createQuery(setQueryTable(SELECT_BY_TIMESTAMP_NOKEY, someClass), someClass)
556 .setParameter(NAME, key.getName())
557 .setParameter(VERSION, key.getVersion())
558 .setParameter(TIMESTAMP, Timestamp.from(timeStamp))
567 public <T extends PfConcept> T getConcept(final Class<T> someClass, final PfConceptKey key) {
568 if (someClass == null || key == null) {
571 final EntityManager mg = getEntityManager();
575 ret = mg.createQuery(setQueryTable(SELECT_BY_CONCEPT_KEY, someClass), someClass)
576 .setParameter(NAME, key.getName())
577 .setParameter(VERSION, key.getVersion())
584 return getSingleResult(someClass, key.getId(), ret);
588 public <T extends PfConcept> T getConcept(final Class<T> someClass, final PfReferenceKey key) {
589 if (someClass == null || key == null) {
592 final EntityManager mg = getEntityManager();
596 ret = mg.createQuery(setQueryTable(SELECT_BY_REFERENCE_KEY, someClass), someClass)
597 .setParameter(PARENT_NAME, key.getParentKeyName())
598 .setParameter(PARENT_VERSION, key.getParentKeyVersion())
599 .setParameter(LOCAL_NAME, key.getLocalName())
606 return getSingleResult(someClass, key.getId(), ret);
610 public <T extends PfConcept> T update(final T obj) {
611 final EntityManager mg = getEntityManager();
614 mg.getTransaction().begin();
617 mg.getTransaction().commit();
625 public <T extends PfConcept> long size(final Class<T> someClass) {
626 if (someClass == null) {
629 final EntityManager mg = getEntityManager();
632 size = mg.createQuery("SELECT COUNT(c) FROM " + someClass.getSimpleName() + " c", Long.class)
641 * Add the table to a query string.
643 * @param queryString the query string
644 * @param tableClass the class name of the table
645 * @return the updated query string
647 private <T extends PfConcept> String setQueryTable(final String queryString, final Class<T> tableClass) {
648 return queryString.replace(TABLE_TOKEN, tableClass.getSimpleName());
652 * Check that a query returned one and only one entry and return that entry.
654 * @param someClass the class being searched for
655 * @param conceptName the concept name being searched for
656 * @param resultList the result list returned by the query
657 * @return the single unique result
659 private <T extends PfConcept> T getSingleResult(final Class<T> someClass, final String searchFilter, List<T> ret) {
660 if (ret == null || ret.isEmpty()) {
663 if (ret.size() > 1) {
664 throw new IllegalArgumentException("More than one result was returned query on " + someClass
665 + " with filter " + searchFilter + ": " + ret);
671 * check the result get from database and return the object.
673 * @param <T> the type of the object to get, a subclass of {@link PfConcept}
674 * @param someClass the class of the object to get, a subclass of {@link PfConcept}
675 * @param t the object that was retrieved from the database
676 * @return the checked object or null
678 private <T extends PfConcept> T checkAndReturn(final Class<T> someClass, final T objToCheck) {
679 if (objToCheck != null) {
681 return PfUtils.makeCopy(objToCheck);
682 } catch (final Exception e) {
683 LOGGER.warn(CLONE_ERR_MSG, someClass.getName(), e);