cps-ri/src/main/java/org/onap/cps/spi/repository/TempTableCreator.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2022-2023 Nordix Foundation.
   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.cps.spi.repository;
  22
  23 import java.util.Arrays;
  24 import java.util.Collection;
  25 import java.util.HashSet;
  26 import java.util.Iterator;
  27 import java.util.List;
  28 import java.util.UUID;
  29 import java.util.stream.Collectors;
  30 import javax.persistence.EntityManager;
  31 import javax.persistence.PersistenceContext;
  32 import lombok.AllArgsConstructor;
  33 import lombok.extern.slf4j.Slf4j;
  34 import org.onap.cps.spi.utils.EscapeUtils;
  35 import org.springframework.stereotype.Component;
  36 import org.springframework.transaction.annotation.Transactional;
  37
  38 @Slf4j
  39 @Transactional
  40 @AllArgsConstructor
  41 @Component
  42 public class TempTableCreator {
  43
  44     @PersistenceContext
  45     private EntityManager entityManager;
  46
  47     /**
  48      * Create a uniquely named temporary table.
  49      *
  50      * @param prefix      prefix for the table name (so you can recognize it)
  51      * @param sqlData     data to insert (strings only) the inner List present a row of data
  52      * @param columnNames column names (in same order as data in rows in sqlData)
  53      * @return a unique temporary table name with given prefix
  54      */
  55     public String createTemporaryTable(final String prefix,
  56                                        final Collection<List<String>> sqlData,
  57                                        final String... columnNames) {
  58         final String tempTableName = prefix + UUID.randomUUID().toString().replace("-", "");
  59         final StringBuilder sqlStringBuilder = new StringBuilder("CREATE TEMPORARY TABLE ");
  60         sqlStringBuilder.append(tempTableName);
  61         defineColumns(sqlStringBuilder, columnNames);
  62         sqlStringBuilder.append(" ON COMMIT DROP;");
  63         insertData(sqlStringBuilder, tempTableName, columnNames, sqlData);
  64         entityManager.createNativeQuery(sqlStringBuilder.toString()).executeUpdate();
  65         return tempTableName;
  66     }
  67
  68     private static void defineColumns(final StringBuilder sqlStringBuilder, final String[] columnNames) {
  69         sqlStringBuilder.append('(');
  70         final Iterator<String> it = Arrays.stream(columnNames).iterator();
  71         while (it.hasNext()) {
  72             final String columnName = it.next();
  73             sqlStringBuilder.append(" ");
  74             sqlStringBuilder.append(columnName);
  75             sqlStringBuilder.append(" varchar NOT NULL");
  76             if (it.hasNext()) {
  77                 sqlStringBuilder.append(",");
  78             }
  79         }
  80         sqlStringBuilder.append(")");
  81     }
  82
  83     private static void insertData(final StringBuilder sqlStringBuilder,
  84                                    final String tempTableName,
  85                                    final String[] columnNames,
  86                                    final Collection<List<String>> sqlData) {
  87         final Collection<String> sqlInserts = new HashSet<>(sqlData.size());
  88         for (final Collection<String> rowValues : sqlData) {
  89             final Collection<String> escapedValues =
  90                 rowValues.stream().map(EscapeUtils::escapeForSqlStringLiteral).collect(Collectors.toList());
  91             sqlInserts.add("('" + String.join("','", escapedValues) + "')");
  92         }
  93         sqlStringBuilder.append("INSERT INTO ");
  94         sqlStringBuilder.append(tempTableName);
  95         sqlStringBuilder.append(" (");
  96         sqlStringBuilder.append(String.join(",", columnNames));
  97         sqlStringBuilder.append(") VALUES ");
  98         sqlStringBuilder.append(String.join(",", sqlInserts));
  99         sqlStringBuilder.append(";");
 100     }
 101
 102 }