Function resolvePartitionExpression

packages/sql/src/schema/partitioning.ts:244–265

(
  meta: EntityMetadata,
  expression: NonNullable<EntityPartitionBy['expression']>,
  quoteIdentifier: (id: string) => string,
)

 * user owns identifier quoting inside a raw expression.
 */
const resolvePartitionExpression = (
  meta: EntityMetadata,
  expression: NonNullable<EntityPartitionBy['expression']>,
  quoteIdentifier: (id: string) => string,
): string => {
  if (typeof expression === 'function') {
    return normalizeWhitespace(expression(meta.createSchemaColumnMappingObject()));
  }

  if (Array.isArray(expression)) {
    return (expression as readonly string[]).map(key => resolvePartitionKey(meta, key, quoteIdentifier)).join(', ');
  }

  const trimmed = (expression as string).trim();
  const keys = splitCommaSeparatedIdentifiers(trimmed);

  if (keys) {
    return keys.map(key => resolvePartitionKey(meta, key, quoteIdentifier)).join(', ');
  }

  return trimmed;
};

const createPartitionDefinition = (type: EntityPartitionBy['type'], expression: string): string =>
  `${type.toLowerCase()} (${normalizeWhitespace(expression)})`;