Skip to content

Commit dd2142c

Browse files
committed
Returned old RLS syntax in deprecated form
1 parent cc5dd03 commit dd2142c

File tree

2 files changed

+20
-2
lines changed

2 files changed

+20
-2
lines changed

drizzle-orm/src/cockroach-core/table.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ export type CockroachTableWithColumns<T extends TableConfig> =
7373
& T['columns']
7474
& InferTableColumnsModels<T['columns']>
7575
& {
76+
/** @deprecated use `cockroachTable.withRLS()` instead*/
7677
enableRLS: () => Omit<
7778
CockroachTableWithColumns<T>,
7879
'enableRLS'

drizzle-orm/src/pg-core/table.ts

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,14 @@ export type PgTableWithColumns<
6565
> =
6666
& PgTable<T>
6767
& T['columns']
68-
& InferTableColumnsModels<T['columns']>;
68+
& InferTableColumnsModels<T['columns']>
69+
& {
70+
/** @deprecated use `pgTable.withRLS()` instead*/
71+
enableRLS: () => Omit<
72+
PgTableWithColumns<T>,
73+
'enableRLS'
74+
>;
75+
};
6976

7077
/** @internal */
7178
export function pgTableWithSchema<
@@ -123,7 +130,17 @@ export function pgTableWithSchema<
123130
table[PgTable.Symbol.ExtraConfigBuilder] = extraConfig as any;
124131
}
125132

126-
return table as any;
133+
return Object.assign(table, {
134+
enableRLS: () => {
135+
table[PgTable.Symbol.EnableRLS] = true;
136+
return table as PgTableWithColumns<{
137+
name: TTableName;
138+
schema: TSchemaName;
139+
columns: BuildColumns<TTableName, TColumnsMap, 'pg'>;
140+
dialect: 'pg';
141+
}>;
142+
},
143+
}) as any;
127144
}
128145

129146
export interface PgTableFnInternal<TSchema extends string | undefined = undefined> {

0 commit comments

Comments
 (0)