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