Small fixes to resolver

WRadoslaw · Jan 16, 2024 · 1626888 · 1626888
1 parent 91384ef
commit 1626888
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/src/server-extension/resolvers/VideosResolver/index.ts b/src/server-extension/resolvers/VideosResolver/index.ts
@@ -51,7 +51,7 @@ export class VideosResolver {
     @Info() info: GraphQLResolveInfo,
     @Ctx() ctx: Context
   ): Promise<RecommendedVideosQuery> {
-    const { listQuery } = buildRecommendationsVideoQuery(info, ctx)
+    const listQuery = buildRecommendationsVideoQuery(info, ctx)
     // get videosIds from recombee
     let res
 
@@ -97,7 +97,7 @@ export class VideosResolver {
     @Info() info: GraphQLResolveInfo,
     @Ctx() ctx: Context
   ): Promise<RecommendedVideosQuery> {
-    const { listQuery } = buildRecommendationsVideoQuery(info, ctx)
+    const listQuery = buildRecommendationsVideoQuery(info, ctx)
     // get videosIds from recombee
     let res
 

diff --git a/src/server-extension/resolvers/VideosResolver/utils.ts b/src/server-extension/resolvers/VideosResolver/utils.ts
@@ -9,15 +9,9 @@ export function buildRecommendationsVideoQuery(info: GraphQLResolveInfo, ctx: Co
   const tree = getResolveTree(info)
 
   // Extract subsquid-supported Channel fields
-  const videoSubTree = tree.fieldsByTypeName.HomepageVideosConnection.video
+  const videoSubTree = tree.fieldsByTypeName.RecommendedVideosQuery.video
   const videoFields = parseAnyTree(model, 'Video', info.schema, videoSubTree)
 
   // Generate query using subsquid's ListQuery
-  const listQuery = new ListQuery(model, ctx.openreader.dialect, 'Video', videoFields, {})
-
-  return {
-    listQuery,
-    recommId: tree.fieldsByTypeName.HomepageVideosConnection.recommId,
-    numberNextRecommsCalls: tree.fieldsByTypeName.HomepageVideosConnection.numberNextRecommsCalls,
-  }
+  return new ListQuery(model, ctx.openreader.dialect, 'Video', videoFields, {})
 }