Skip to content

Commit 1bb32ea

Browse files
authored
chore: add missing materialization resolver (#6403)
1 parent 9526dc9 commit 1bb32ea

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

apps/frontend/app/api/v1/osograph/schema/resolvers/data-ingestion.ts

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@ import {
44
validateInput,
55
} from "@/app/api/v1/osograph/utils/validation";
66
import { requireAuthentication } from "@/app/api/v1/osograph/utils/auth";
7-
import { checkMembershipExists } from "@/app/api/v1/osograph/utils/resolver-helpers";
7+
import {
8+
checkMembershipExists,
9+
getMaterializations,
10+
} from "@/app/api/v1/osograph/utils/resolver-helpers";
811
import { createAdminClient } from "@/lib/supabase/admin";
912
import {
1013
AuthenticationErrors,
@@ -15,6 +18,8 @@ import { logger } from "@/lib/logger";
1518
import z from "zod";
1619
import { DataIngestionsRow } from "@/lib/types/schema-types";
1720
import { getModelContext } from "@/app/api/v1/osograph/schema/resolvers/model-context";
21+
import type { FilterableConnectionArgs } from "@/app/api/v1/osograph/utils/pagination";
22+
import { generateTableId } from "@/app/api/v1/osograph/utils/model";
1823

1924
export const dataIngestionResolvers = {
2025
Mutation: {
@@ -118,5 +123,19 @@ export const dataIngestionResolvers = {
118123
) => {
119124
return getModelContext(parent.dataset_id, args.tableName);
120125
},
126+
materializations: async (
127+
parent: DataIngestionsRow,
128+
args: FilterableConnectionArgs & { tableName: string },
129+
context: GraphQLContext,
130+
) => {
131+
const { tableName, ...restArgs } = args;
132+
return getMaterializations(
133+
restArgs,
134+
context,
135+
parent.org_id,
136+
parent.dataset_id,
137+
generateTableId("DATA_INGESTION", tableName),
138+
);
139+
},
121140
},
122141
};

0 commit comments

Comments
 (0)