Merge pull request drizzle-team#2574 from lbguilherme/patch-1

feat(pg): allow creating indexes on 3+ columns mixing columns and expressions
singlestore-labs · Jul 23, 2024 · 341f17c · 341f17c
2 parents 026b9bb + dcc13b8
commit 341f17c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/drizzle-orm/src/pg-core/indexes.ts b/drizzle-orm/src/pg-core/indexes.ts
@@ -111,7 +111,7 @@ export class IndexBuilderOn {
 
 	constructor(private unique: boolean, private name?: string) {}
 
-	on(...columns: [Partial<ExtraConfigColumn> | SQL, ...Partial<ExtraConfigColumn>[] | SQL[]]): IndexBuilder {
+	on(...columns: [Partial<ExtraConfigColumn> | SQL, ...Partial<ExtraConfigColumn | SQL>[]]): IndexBuilder {
 		return new IndexBuilder(
 			columns.map((it) => {
 				if (is(it, SQL)) {
@@ -128,7 +128,7 @@ export class IndexBuilderOn {
 		);
 	}
 
-	onOnly(...columns: [Partial<ExtraConfigColumn | SQL>, ...Partial<ExtraConfigColumn>[] | SQL[]]): IndexBuilder {
+	onOnly(...columns: [Partial<ExtraConfigColumn | SQL>, ...Partial<ExtraConfigColumn | SQL>[]]): IndexBuilder {
 		return new IndexBuilder(
 			columns.map((it) => {
 				if (is(it, SQL)) {
@@ -158,7 +158,7 @@ export class IndexBuilderOn {
 	 */
 	using(
 		method: PgIndexMethod,
-		...columns: [Partial<ExtraConfigColumn | SQL>, ...Partial<ExtraConfigColumn>[] | SQL[]]
+		...columns: [Partial<ExtraConfigColumn | SQL>, ...Partial<ExtraConfigColumn | SQL>[]]
 	): IndexBuilder {
 		return new IndexBuilder(
 			columns.map((it) => {