DT
Drizzle TeamSuji

Infer return type for relational query with pagination

I have created a function that can paginate data for relational queries, problem is i cant figure out how to make to include the relations that are passed into the inferred return type 
async paginatedQuery<
    ReturnType extends drizzleOrm.InferModel<Table>,
    Table extends AnyPgTable,
  >(
    table: Table,
    {
      limit,
      page,
      where,
      takeAll,
      with: queryWith,
    }: {
      limit: number;
      page: number;
      where?: drizzleOrm.SQL;
      takeAll?: boolean;
      // eslint-disable-next-line @typescript-eslint/ban-types
      with?: object;
    },
  ) {
    const tableNameSymbol = Object.getOwnPropertySymbols(table).find(
      (value) => value.description === 'Name',
    );

    const tableName = table[tableNameSymbol!].toString().toLowerCase();

    const count = await (where?.queryChunks.length === 0
      ? this.db
          .select({ count: sql<number>`count(*)`.mapWith(Number) })
          .from(table)
      : this.db
          .select({ count: sql<number>`count(*)`.mapWith(Number) })
          .from(table)
          .where(where));

    const offset = (page - 1) * limit;

    const data: ReturnType[] = await this.db.query[tableName].findMany({
      ...(queryWith && { with: queryWith }),
      ...(where?.queryChunks.length !== 0 && { where }),
      ...(!takeAll && { limit, offset: offset <= 0 ? 0 : offset }),
    });

    const meta: IPageMeta = {
      currentPage: page,
      isLastPage: page * limit >= count[0].count,
      isFirstPage: page === 1,
      nextPage: page * limit >= count[0].count ? null : page + 1,
      previousPage: page - 1 === 0 ? null : page - 1,
      pageCount: Math.ceil(count[0].count / limit),
    };

    return { data, meta };
  }
async paginatedQuery<
    ReturnType extends drizzleOrm.InferModel<Table>,
    Table extends AnyPgTable,
  >(
    table: Table,
    {
      limit,
      page,
      where,
      takeAll,
      with: queryWith,
    }: {
      limit: number;
      page: number;
      where?: drizzleOrm.SQL;
      takeAll?: boolean;
      // eslint-disable-next-line @typescript-eslint/ban-types
      with?: object;
    },
  ) {
    const tableNameSymbol = Object.getOwnPropertySymbols(table).find(
      (value) => value.description === 'Name',
    );

    const tableName = table[tableNameSymbol!].toString().toLowerCase();

    const count = await (where?.queryChunks.length === 0
      ? this.db
          .select({ count: sql<number>`count(*)`.mapWith(Number) })
          .from(table)
      : this.db
          .select({ count: sql<number>`count(*)`.mapWith(Number) })
          .from(table)
          .where(where));

    const offset = (page - 1) * limit;

    const data: ReturnType[] = await this.db.query[tableName].findMany({
      ...(queryWith && { with: queryWith }),
      ...(where?.queryChunks.length !== 0 && { where }),
      ...(!takeAll && { limit, offset: offset <= 0 ? 0 : offset }),
    });

    const meta: IPageMeta = {
      currentPage: page,
      isLastPage: page * limit >= count[0].count,
      isFirstPage: page === 1,
      nextPage: page * limit >= count[0].count ? null : page + 1,
      previousPage: page - 1 === 0 ? null : page - 1,
      pageCount: Math.ceil(count[0].count / limit),
    };

    return { data, meta };
  }
returnType is typed with only the base table fields, this doenst include the fields that get included from relations. Any clues or pointers ??
A
Amur•309d ago
@therealsujje did you manage to solve this?
S
Suji•309d ago
Nope
Want results from more Discord servers?
Add your server
More Posts
How to implement triggers or db hooks ?Is there a way to know if sql statement got executed successfully or not ? I wanted to run a code whnoob help: INSERT with JOIN SQL to DrizzleFrontend developer dabbling with SQL for the first time in years. With the help of ChatGPT I have thOrderBy with dynamic queries dont work``` ....orderBy(sql`${products.xyz} ${order}`) ``` ``` error: syntax error at or near "$2" ``` ordeIs there a way to limit update to just 1 document?I know there will be multiple documents matching this where query, but I only want to update 1 of thHow to get InferModel to work with Relation model?Is there a way to get the infer type for a relation model? eg: type User = InferModel<typeof users,MySQL column type: "Generated"?Does drizzle support defining generated columns? https://planetscale.com/courses/mysql-for-develope