如何在Drizzle动态查询中实现表列别名？

你的问题出在直接展开两个表的全部列时，重复字段（如id）导致别名设置未生效，且Drizzle对键名与列名一致的字段会省略AS关键字。以下是几种可行的解决方法：

方法一：排除重复字段后添加别名 #

使用Drizzle的omit工具函数去掉重复的id字段，再单独给车辆表的id设置别名：

import { omit } from 'drizzle-orm';

let deviceQuery = qb
  .select({
    // 获取设备表所有列
    ...getTableColumns(deviceTable),
    // 获取车辆表所有列，但排除原生id字段
    ...omit(getTableColumns(vehicleTable), ['id']),
    // 给车辆表id设置别名vehicleId
    vehicleId: vehicleTable.id,
  })
  .from(deviceTable)
  .innerJoin(vehicleTable, eq(deviceTable.vehicleId, vehicleTable.id))
  .$dynamic();

此时生成的SQL会包含"vehicles"."id" AS "vehicleId"，同时保留设备表的id字段（无冗余）。

方法二：手动指定字段（更精准） #

如果不需要两个表的全部字段，可以手动列出需要的列并直接设置别名：

let deviceQuery = qb
  .select({
    // 设备表字段
    id: deviceTable.id,
    connectionStatus: deviceTable.connection_status,
    batteryLevel: deviceTable.battery_level_in_volts,
    // 车辆表字段（带别名）
    vehicleId: vehicleTable.id,
    vehicleTag: vehicleTable.vehicle_tag,
    odometer: vehicleTable.odometer_in_kilometers,
    // 其他需要的字段...
  })
  .from(deviceTable)
  .innerJoin(vehicleTable, eq(deviceTable.vehicleId, vehicleTable.id))
  .$dynamic();

这种方式完全避免了字段重复问题，且别名设置清晰。

方法三：使用sql()强制指定别名 #

如果必须保留两个表的全部列，同时添加别名，可以用sql()函数强制生成别名（注意会产生重复列，需根据需求取舍）：

import { sql } from 'drizzle-orm';

let deviceQuery = qb
  .select({
    ...getTableColumns(deviceTable),
    ...getTableColumns(vehicleTable),
    // 强制生成别名
    vehicleId: sql`${vehicleTable.id} AS vehicleId`,
  })
  .from(deviceTable)
  .innerJoin(vehicleTable, eq(deviceTable.vehicleId, vehicleTable.id))
  .$dynamic();

此时SQL会同时包含原生的"vehicles"."id"和别名后的"vehicles"."id" AS "vehicleId"，适合需要保留原始字段又要别名的场景。

内容的提问来源于stack exchange，提问作者PlayMa256