IDEMPIERE-6107 Improve Table Partitioning API (#2315)

* IDEMPIERE-6107 - Improve Table Partitioning API

org.adempiere.base/src/org/compiere/model/MTable.java

@@ -1098,4 +1098,27 @@ public class MTable extends X_AD_Table implements ImmutablePOSupport
 		return indexName.toString();
 	}
 
+	/**
+	 * Get Partition Name of the table of the given level
+	 * @param tableName
+	 * @param primaryLevelOnly - if true, ignore the sub-partition, if exists
+	 * @return table partition name, or empty
+	 */
+	public static String getPartitionName(Properties ctx, String tableName, boolean primaryLevelOnly, String trxName) {
+		if(Util.isEmpty(tableName))
+			return "";
+		
+		String[] partitionColsAll = MTablePartition.getPartitionKeyColumns(ctx, tableName, trxName);
+		
+		if(partitionColsAll.length == 0)
+			return tableName;
+		
+		int level = primaryLevelOnly ? 1 : partitionColsAll.length;
+		StringBuilder partitionName = new StringBuilder();
+		partitionName.append(tableName);
+		for(int i = 0; i < level; i++) {
+			partitionName.append("_").append(partitionColsAll[i]);
+		}
+		return partitionName.toString();
+	}
 }	//	MTable

org.adempiere.base/src/org/compiere/model/MTablePartition.java

@@ -0,0 +1,164 @@
+/**********************************************************************
+* This file is part of iDempiere ERP Open Source                      *
+* http://www.idempiere.org                                            *
+*                                                                     *
+* Copyright (C) Contributors                                          *
+*                                                                     *
+* This program is free software; you can redistribute it and/or       *
+* modify it under the terms of the GNU General Public License         *
+* as published by the Free Software Foundation; either version 2      *
+* of the License, or (at your option) any later version.              *
+*                                                                     *
+* This program is distributed in the hope that it will be useful,     *
+* but WITHOUT ANY WARRANTY; without even the implied warranty of      *
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the        *
+* GNU General Public License for more details.                        *
+*                                                                     *
+* You should have received a copy of the GNU General Public License   *
+* along with this program; if not, write to the Free Software         *
+* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,          *
+* MA 02110-1301, USA.                                                 *
+*                                                                     *
+* Contributors:                                                       *
+* - Peter Takacs, Cloudempiere                                        *
+**********************************************************************/
+package org.compiere.model;
+
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Properties;
+import java.util.stream.Collectors;
+
+import org.compiere.util.DB;
+
+public class MTablePartition extends X_AD_TablePartition {
+
+	/** Serial Version ID */
+	private static final long serialVersionUID = 1L;
+
+	/**
+	 * @param ctx
+	 * @param AD_TablePartition_ID
+	 * @param trxName
+	 */
+	public MTablePartition(Properties ctx, int AD_TablePartition_ID, String trxName) {
+		super(ctx, AD_TablePartition_ID, trxName);
+	}
+
+	/**
+	 * @param ctx
+	 * @param AD_TablePartition_ID
+	 * @param trxName
+	 * @param virtualColumns
+	 */
+	public MTablePartition(Properties ctx, int AD_TablePartition_ID, String trxName, String[] virtualColumns) {
+		super(ctx, AD_TablePartition_ID, trxName, virtualColumns);
+	}
+
+	/**
+	 * @param ctx
+	 * @param rs
+	 * @param trxName
+	 */
+	public MTablePartition(Properties ctx, ResultSet rs, String trxName) {
+		super(ctx, rs, trxName);
+	}
+
+	/**
+	 * UUID based Constructor
+	 * @param ctx
+	 * @param AD_TablePartition_UU
+	 * @param trxName
+	 */
+	public MTablePartition(Properties ctx, String AD_TablePartition_UU, String trxName) {
+		super(ctx, AD_TablePartition_UU, trxName);
+	}
+
+	/**
+	 * UUID based Constructor
+	 * @param ctx
+	 * @param AD_TablePartition_UU
+	 * @param trxName
+	 * @param virtualColumns
+	 */
+	public MTablePartition(Properties ctx, String AD_TablePartition_UU, String trxName, String[] virtualColumns) {
+		super(ctx, AD_TablePartition_UU, trxName, virtualColumns);
+	}
+	
+	/**
+	 * Check if the given table partition exists for the table
+	 * @param tableName
+	 * @param partitionName
+	 * @return true if partition already exists
+	 */
+	public static boolean partitionExists(Properties ctx, String tableName, String partitionName, String trxName) {
+		 return partitionExists(ctx, MTable.getTable_ID(tableName), partitionName, trxName);
+	}
+	
+	/**
+	 * Check if the given table partition exists for the table
+	 * @param tableId
+	 * @param partitionName
+	 * @return true if partition already exists
+	 */
+	public static boolean partitionExists(Properties ctx, int tableId, String partitionName, String trxName) {
+		boolean returnVal = false;
+		StringBuilder sqlSelect = new StringBuilder();
+		sqlSelect.append("SELECT 1 FROM ")
+		    .append(X_AD_TablePartition.Table_Name)
+		    .append(" WHERE AD_Table_ID = ? AND UPPER(Name) LIKE UPPER(?) ");
+		PreparedStatement ps = null;
+		ResultSet rs = null;
+		try {
+		    ps = DB.prepareStatement(sqlSelect.toString(), trxName);
+		    ps.setInt(1,tableId);
+		    ps.setString(2, partitionName);
+		    rs = ps.executeQuery();
+		    if (rs.next()) {
+		        returnVal = true;
+		    }
+		} catch (SQLException e) {
+		    e.printStackTrace();
+		} finally {
+		    DB.close(rs, ps);
+		    rs = null; ps = null;
+		}
+		
+		return returnVal;
+	}
+	
+	/**
+	 * Get list of partition key columns for the given table
+	 * @param ctx
+	 * @param tableName
+	 * @param trxName
+	 * @return array of column names 
+	 */
+	public static String[] getPartitionKeyColumns(Properties ctx, String tableName, String trxName) {
+		return getPartitionKeyColumns(ctx, MTable.getTable_ID(tableName), trxName);
+	}
+	
+	/**
+	 * Get list of partition key columns for the given table
+	 * @param ctx
+	 * @param tableId
+	 * @param trxName
+	 * @return array of column names
+	 */
+	public static String[] getPartitionKeyColumns(Properties ctx, int tableId, String trxName) {
+	    String whereClause = "AD_Table_ID=? AND IsPartitionKey='Y'";
+	    List<MColumn> keyCols = new Query(ctx, MColumn.Table_Name, whereClause, trxName)
+	    		.setParameters(tableId)
+	    		.setOnlyActiveRecords(true)
+	            .setOrderBy("SeqNoPartition ASC")
+	            .list();
+
+	    List<String> keyColNames = keyCols.stream()
+	            .map(MColumn::getColumnName)
+	            .collect(Collectors.toList());
+
+	    return keyColNames.toArray(new String[0]);
+	}
+}

org.compiere.db.postgresql.provider/src/org/adempiere/db/postgresql/partition/TablePartitionService.java

@@ -64,9 +64,23 @@ public class TablePartitionService implements ITablePartitionService {
 		return DB.getSQLValueEx(trxName, sql, table.getTableName()) == 1;
 	}
 	
-	private String getDefaultPartitionName(MTable table)
+	/**
+	 * Get default partition name for table
+	 * @param table
+	 * @return String default partition name for table
+	 */
+	public String getDefaultPartitionName(MTable table)
 	{
-		return table.getTableName() + "_default_partition";
+		return getDefaultPartitionName(table.getTableName());
+	}
+	
+	/**
+	 * Get default partition name for table
+	 * @param tableName
+	 * @return String default partition name for table
+	 */
+	public String getDefaultPartitionName(String tableName) {
+		return tableName + "_default_partition";
 	}
 	
 	/**