more nits to OPTIMIZE TABLE

singlestore-labs · Jul 25, 2024 · 52963af · 52963af
1 parent a72e61c
commit 52963af
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/drizzle-orm/src/singlestore-core/dialect.ts b/drizzle-orm/src/singlestore-core/dialect.ts
@@ -161,9 +161,11 @@ export class SingleStoreDialect {
 	buildOptimizeTable({ table, arg, selection }: SingleStoreOptimizeTableConfig): SQL {
 		const argSql = arg ? sql` ${sql.raw(arg)}` : undefined;
 
-		const warmBlobCacheForColumnSql = selection ?
-			sql` warm blob cache for column ${selection}`
-			: undefined;
+		let warmBlobCacheForColumnSql = undefined;
+		if (selection) {
+			const selectionField = selection.map((column) => { return { path: [], field: column } });
+			warmBlobCacheForColumnSql = sql` warm blob cache for column ${this.buildSelection(selectionField, { isSingleTable: true })}`;
+		}
 
 		return sql`optimize table ${table}${argSql}${warmBlobCacheForColumnSql}`;
 	}