diff --git a/apps/studio/components/interfaces/TableGridEditor/SidePanelEditor/TableEditor/TableEditor.tsx b/apps/studio/components/interfaces/TableGridEditor/SidePanelEditor/TableEditor/TableEditor.tsx index d4058972d45..1c685fe09b5 100644 --- a/apps/studio/components/interfaces/TableGridEditor/SidePanelEditor/TableEditor/TableEditor.tsx +++ b/apps/studio/components/interfaces/TableGridEditor/SidePanelEditor/TableEditor/TableEditor.tsx @@ -130,11 +130,21 @@ const TableEditor = ({ (constraint) => constraint.type === CONSTRAINT_TYPE.PRIMARY_KEY_CONSTRAINT ) - const { data: foreignKeyMeta } = useForeignKeyConstraintsQuery({ - projectRef: project?.ref, - connectionString: project?.connectionString, - schema: table?.schema, - }) + const { data: foreignKeyMeta } = useForeignKeyConstraintsQuery( + { + projectRef: project?.ref, + connectionString: project?.connectionString, + schema: table?.schema, + }, + { + onSuccess: (data) => { + const relevantForeignKeys = data.filter( + (fk) => fk.source_schema === table?.schema && fk.source_table === table?.name + ) + setFkRelations(formatForeignKeys(relevantForeignKeys)) + }, + } + ) const foreignKeys = (foreignKeyMeta ?? []).filter( (fk) => fk.source_schema === table?.schema && fk.source_table === table?.name )