在使用 Prisma 进行数据库查询时，有时会遇到 Schema 中定义的关联数组未被返回的问题。本文将深入探讨此问题的原因，并提供详细的解决方案和最佳实践，帮助开发者避免类似错误，确保数据查询的完整性和准确性。核心在于理解 Prisma 的 include 选项，以及如何在查询中正确使用它来获取关联数据。
在使用 Prisma 进行数据库查询时，如果关联数组（例如一对多关系中的子表数据）没有在查询结果中返回，主要原因通常是查询语句中缺少显式的包含声明。Prisma 默认情况下不会自动包含所有关联数据，需要使用 include 或 select 选项来指定需要返回的关联关系。
以下将详细解释如何解决这个问题，并提供一些最佳实践。
解决方案：使用 include 选项
在 Prisma 查询中，include 选项用于指定需要包含的关联数据。对于上述问题，需要在 shoppingList.findUnique 查询中包含 items 字段，才能返回 ShoppingListItem 数组。
const list = await prisma.shoppingList.findUnique({
where: {
id: input.id,
},
include: {
items: true,
},
});这段代码会将 ShoppingList 以及与其关联的所有 ShoppingListItem 一起返回。
代码示例：完整的查询过程
下面是一个完整的代码示例，展示了如何使用 include 选项来获取关联数据：
import { PrismaClient } from '@prisma/client';
import { z } from 'zod';
import { TRPCError } from '@trpc/server';
const prisma = new PrismaClient();
// 假设你使用了 tRPC
const publicProcedure = { // 替换为你实际的 publicProcedure 定义
input: (schema: z.ZodType) => ({
parse: (input: any) => schema.parse(input),
}),
query: (resolver: any) => resolver,
};
const getShoppingListById = publicProcedure
.input(
z.object({
id: z.string(),
})
)
.query(async ({ input }) => {
const list = await prisma.shoppingList.findUnique({
where: {
id: input.id,
},
include: {
items: true, // 包含关联的 ShoppingListItem
},
});
if (!list) {
throw new TRPCError({ code: 'NOT_FOUND' });
}
return list;
});
// 使用示例
async function main() {
try {
const shoppingList = await getShoppingListById.query({ input: { id: 'clhnzfqy80000vv27ahbbk0xz' } }); // 替换为你的 ShoppingList ID
console.log(shoppingList);
} catch (error) {
console.error(error);
} finally {
await prisma.$disconnect();
}
}
main(); 注意事项：select 选项
除了 include，Prisma 还提供了 select 选项，它允许你更精细地控制返回的字段。select 可以选择包含关联数据，也可以选择只包含关联数据的特定字段。
例如，如果只想获取 ShoppingListItem 的 id 和 name 字段，可以使用 select：
const list = await prisma.shoppingList.findUnique({
where: {
id: input.id,
},
select: {
id: true,
name: true,
items: {
select: {
id: true,
name: true,
},
},
},
});最佳实践：性能优化
当关联数据量很大时，使用 include: true 可能会导致性能问题。在这种情况下，可以考虑以下优化策略：
- 使用 select 限制返回字段： 只选择需要的字段，避免返回不必要的数据。
- 分页查询： 对于大型关联集合，可以使用分页查询来分批获取数据。
- 使用 Prisma 的 $queryRaw 或视图： 对于复杂的查询逻辑，可以考虑使用原始 SQL 查询或数据库视图来优化性能。
总结
在使用 Prisma 查询关联数据时，务必使用 include 或 select 选项来显式指定需要返回的关联关系。include 简单易用，适合大多数场景；select 提供了更精细的控制，可以用于优化性能。根据实际需求选择合适的选项，可以确保数据查询的完整性和效率。
