Polymorphic object does not work with Mixin abstract model

[mjperlini]

Trying to use a polymorphic model with a base object that includes an abstract model.

type BaseEntity {
  id         String   @id @default(ulid())
  createdOn  DateTime @default(now())
  updatedOn  DateTime @updatedAt
  isDeleted  Boolean  @default(false)
  isArchived Boolean  @default(false)
}

model RoutineData with BaseEntity {
  dataType  DataType
  sortOrder Float?
  title     String?

  routineId String
  Routine   Routine  @relation(fields: [routineId], references: [id])

  @@delegate(dataType)
  @@allow('all', auth().id == Routine.userId)
}

model DataText extends RoutineData {
  text String
}

Reading DataText object fails with 'dataText.createdOn' does not exist. This pattern worked with V2.

I was able to make it work by copying the BaseEntity rows into the RoutineData table directly.

model RoutineData {
  id         String   @id @default(ulid())
  createdOn  DateTime @default(now())
  updatedOn  DateTime @updatedAt
  isDeleted  Boolean  @default(false)
  isArchived Boolean  @default(false)

  dataType  DataType
  sortOrder Float?
  title     String?

  routineId String
  Routine   Routine  @relation(fields: [routineId], references: [id])

  @@delegate(dataType)
  @@allow('all', auth().id == Routine.userId)
}

model DataText extends RoutineData {
  text String
}

[ymc9]

Hi @mjperlini ,

I couldn't seem to reproduce the issue. My model:

type BaseEntity {
    id         String   @id @default(ulid())
    createdOn  DateTime @default(now())
    updatedOn  DateTime @updatedAt
    isDeleted  Boolean  @default(false)
    isArchived Boolean  @default(false)
}

model RoutineData with BaseEntity {
    dataType  String
    sortOrder Float?
    title     String?

    @@delegate(dataType)
}

model DataText extends RoutineData {
    text String
}

TS code:

await db.dataText.create({ data: { text: 'abc', title: 'Hello' } });
const r = await db.dataText.findFirst();
console.log(r);

The query result is complete (both its value and typing).

Could you share a repro project?

[mjperlini]

It is a large monorepo project so would be hard to break down to a simple example. I will see what I can put together, but in the meantime, this is the error that I get when I try to load a polymorphic entity:

--> GET /api/model/dataText/findUnique?q=%7B%22where%22%3A%7B%22id%22%3A%2201K6F788AA7CVB0BARENXM3F35%22%7D%7D 400 11ms
<-- GET /api/model/dataLink/findUnique?q=%7B%22where%22%3A%7B%22id%22%3A%2201K6F788AYNESACG10G3J1XVBW%22%7D%7D
@zenstackhq/server: [debug] handling "dataLink.findUnique" request with args: {"where":{"id":"01K6F788AYNESACG10G3J1XVBW"}}
@zenstackhq/server: [error] error occurred when handling "dataLink.findUnique" request
Error: Failed to execute query: Error: Failed to execute query: SqliteError: no such column: DataLink.createdOn
@zenstackhq/server: [debug] sending error response: {"status":400,"body":{"error":{"message":"Failed to execute query: Error: Failed to execute query: SqliteError: no such column: DataLink.createdOn","reason":"db-query-error"}}}

All the typing, etc seems fine in the project. It seems to be how the query is being constructed, so it is trying to read the BaseEntity properties from the extended table. As noted before, it does work without the BaseEntity type.

This is where I call it:

export function DataTextPanel({ dataTextId }: { dataTextId: string }) {
const client = useClientQueries(schema);

const { data, isLoading } = client.dataText.useFindUnique({
  where: { id: dataTextId },
});

I'll see if I can break it down to something simpler and send it to you.

[ymc9]

Got it. Could you enable the query log and share the SQL executed right before the failure occurred?

https://zenstack.dev/docs/orm/logging

[mjperlini]

Attached:

@zenstackhq/server: [debug] handling "routineData.findMany" request with args: {"where":{"routineId":"01K6F788EG8D8Y069CXB8HEH3Q"},"select":{"id":true,"dataType":true,"sortOrder":true},"orderBy":{"sortOrder":"asc"}} kysely:query: select "RoutineData"."id" as "id", "RoutineData"."dataType" as "dataType", "RoutineData"."sortOrder" as "sortOrder" from "RoutineData" where "RoutineData"."routineId" = ? and '4C7s8ImcoE3snVtFp0PX2NkwDgva9sth' = (select "userId" as "$t" from "Routine" where "RoutineData"."routineId" = "Routine"."id") order by "RoutineData"."sortOrder" asc kysely:query: duration: 3.8ms @zenstackhq/server: [debug] sending response for "routineData.findMany" request: {"status":200,"body":{"data":[{"id":"01K6F788AA7CVB0BARENXM3F35","dataType":"DataText","sortOrder":null}]}} --> GET /api/model/routineData/findMany?q=%7B%22where%22%3A%7B%22routineId%22%3A%2201K6F788EG8D8Y069CXB8HEH3Q%22%7D%2C%22select%22%3A%7B%22id%22%3Atrue%2C%22dataType%22%3Atrue%2C%22sortOrder%22%3Atrue%7D%2C%22orderBy%22%3A%7B%22sortOrder%22%3A%22asc%22%7D%7D 200 166ms <-- GET /api/model/dataText/findUnique?q=%7B%22where%22%3A%7B%22id%22%3A%2201K6F788AA7CVB0BARENXM3F35%22%7D%7D kysely:query: select "Session"."id" as "id", "Session"."expiresAt" as "expiresAt", "Session"."token" as "token", "Session"."createdAt" as "createdAt", "Session"."updatedAt" as "updatedAt", "Session"."ipAddress" as "ipAddress", "Session"."userAgent" as "userAgent", "Session"."userId" as "userId", "Session"."impersonatedBy" as "impersonatedBy" from "session" as "Session" where "Session"."token" = ? limit ? kysely:query: duration: 0.5ms kysely:query: select "User"."id" as "id", "User"."name" as "name", "User"."email" as "email", "User"."emailVerified" as "emailVerified", "User"."image" as "image", "User"."createdAt" as "createdAt", "User"."updatedAt" as "updatedAt", "User"."role" as "role", "User"."banned" as "banned", "User"."banReason" as "banReason", "User"."banExpires" as "banExpires" from "user" as "User" where "User"."id" = ? limit ? kysely:query: duration: 0.3ms @zenstackhq/server: [debug] handling "dataText.findUnique" request with args: {"where":{"id":"01K6F788AA7CVB0BARENXM3F35"}} kysely:error: SqliteError: no such column: DataText.createdOn at Database.prepare (C:\MyLocalData\mper\dev\praclog\node_modules\.pnpm\better-sqlite3@12.6.2\node_modules\better-sqlite3\lib\methods\wrappers.js:5:21) at SqliteConnection.executeQuery (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/kysely@0.28.10/node_modules/kysely/dist/esm/dialect/sqlite/sqlite-driver.js:62:31) at connection.executeQuery (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:5763:35) at _ZenStackQueryExecutor.internalExecuteQuery (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:6709:31) at _ZenStackQueryExecutor.proceedQuery (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:6529:19) at proceed (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:6477:54) at _p (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:6487:50) at PolicyHandler.handle (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+plugin-policy@3_ee497972f80a806e0ff57bad646177d9/node_modules/@zenstackhq/plugin-policy/dist/index.js:912:14) at PolicyPlugin.onKyselyQuery (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+plugin-policy@3_ee497972f80a806e0ff57bad646177d9/node_modules/@zenstackhq/plugin-policy/dist/index.js:1792:20) at proceed (file:///C:/MyLocalData/mper/dev/praclog/node_modules/.pnpm/@zenstackhq+orm@3.2.1_better-sqlite3@12.6.2_zod@4.3.5/node_modules/@zenstackhq/orm/dist/index.js:6488:34) @zenstackhq/server: [error] error occurred when handling "dataText.findUnique" request Error: Failed to execute query: Error: Failed to execute query: SqliteError: no such column: DataText.createdOn @zenstackhq/server: [debug] sending error response: {"status":400,"body":{"error":{"message":"Failed to execute query: Error: Failed to execute query: SqliteError: no such column: DataText.createdOn","reason":"db-query-error"}}} --> GET /api/model/dataText/findUnique?q=%7B%22where%22%3A%7B%22id%22%3A%2201K6F788AA7CVB0BARENXM3F35%22%7D%7D 400 16ms

[ymc9]

Thanks @mjperlini ! I believe I've got a local repro. Will make a fix soon.

[ymc9]

Fixed in 3.4.5