From fe55aad798ddb559b7201d18eb779acaea54756d Mon Sep 17 00:00:00 2001
From: sdjdd <i@sdjdd.com>
Date: Sun, 7 Apr 2024 13:21:08 +0800
Subject: [PATCH] feat(next/api): has one through pointer relation

---
 next/api/src/model/Ticket.ts    |  6 ++---
 next/api/src/orm/helpers.ts     | 19 +++++++++++++++
 next/api/src/orm/preloader.ts   | 41 +++++++++++++++++++++++++++++++++
 next/api/src/orm/relation.ts    | 13 ++++++++++-
 next/api/src/response/ticket.ts |  3 +--
 next/api/src/router/ticket.ts   |  2 +-
 6 files changed, 77 insertions(+), 7 deletions(-)

diff --git a/next/api/src/model/Ticket.ts b/next/api/src/model/Ticket.ts
index 0a46ee3bb..4e051f3f6 100644
--- a/next/api/src/model/Ticket.ts
+++ b/next/api/src/model/Ticket.ts
@@ -10,8 +10,8 @@ import {
   pointerId,
   pointerIds,
   pointTo,
-  hasManyThroughPointer,
   hasManyThroughPointerArray,
+  hasOneThroughPointer,
   serialize,
 } from '@/orm';
 import { TicketUpdater, UpdateOptions } from '@/ticket/TicketUpdater';
@@ -210,8 +210,8 @@ export class Ticket extends Model {
   @field()
   privateTags?: Tag[];
 
-  @hasManyThroughPointer(() => Notification)
-  notifications?: Notification[];
+  @hasOneThroughPointer(() => Notification)
+  notification?: Notification;
 
   @pointerId(() => Ticket)
   parentId?: string;
diff --git a/next/api/src/orm/helpers.ts b/next/api/src/orm/helpers.ts
index dc8ad6736..da5c631f9 100644
--- a/next/api/src/orm/helpers.ts
+++ b/next/api/src/orm/helpers.ts
@@ -168,6 +168,25 @@ export function hasManyThroughRelation(getRelatedModel: ModelGetter, relatedKey?
   };
 }
 
+export function hasOneThroughPointer(
+  getRelatedModel: ModelGetter,
+  foreignPointerKey?: string,
+  foreignKeyField?: string
+) {
+  return (target: Model, field: string) => {
+    const model = target.constructor as typeof Model;
+    foreignPointerKey ??= lowercaseFirstChar(model.getClassName());
+    model.setRelation({
+      type: RelationType.HasOneThroughPointer,
+      model,
+      field,
+      getRelatedModel,
+      foreignPointerKey,
+      foreignKeyField: foreignKeyField ?? foreignPointerKey + 'Id',
+    });
+  };
+}
+
 function serializeDecoratorFactory(config?: Partial<Omit<SerializedField, 'key'>>) {
   return (target: Model, key: string) => {
     const model = target.constructor as typeof Model;
diff --git a/next/api/src/orm/preloader.ts b/next/api/src/orm/preloader.ts
index 7d77af680..128fa0171 100644
--- a/next/api/src/orm/preloader.ts
+++ b/next/api/src/orm/preloader.ts
@@ -10,6 +10,7 @@ import {
   HasManyThroughPointer,
   HasManyThroughPointerArray,
   HasManyThroughRelation,
+  HasOneThroughPointer,
   RelationName,
   RelationType,
 } from './relation';
@@ -258,6 +259,44 @@ class HasManyThroughRelationPreloader {
   }
 }
 
+class HasOneThroughPointerPreloader implements Preloader {
+  queryModifier?: (query: QueryBuilder<any>) => void;
+
+  constructor(private relation: HasOneThroughPointer) {}
+
+  async load(items: Item[], options?: AuthOptions) {
+    if (items.length === 0) {
+      return;
+    }
+
+    const { model, field, getRelatedModel, foreignPointerKey, foreignKeyField } = this.relation;
+
+    const itemsChunks = _(items)
+      .uniqBy((item) => item.id)
+      .chunk(50)
+      .value();
+
+    const relatedItemsChunks: Item[][] = [];
+
+    for (const items of itemsChunks) {
+      const query = getRelatedModel().queryBuilder();
+      this.queryModifier?.(query);
+      const pointers = items.map((item) => model.ptr(item.id));
+      query.where(foreignPointerKey, 'in', pointers);
+      const relatedItems = await query.find(options);
+      relatedItemsChunks.push(relatedItems);
+    }
+
+    const relatedItemsGroups = _(relatedItemsChunks).flatten().groupBy(foreignKeyField).value();
+    for (const item of items) {
+      const relatedItems = relatedItemsGroups[item.id];
+      if (relatedItems) {
+        item[field] = relatedItems[0];
+      }
+    }
+  }
+}
+
 export function preloaderFactory<M extends typeof Model, N extends RelationName<M>>(
   model: M,
   name: N
@@ -281,5 +320,7 @@ export function preloaderFactory<M extends typeof Model, N extends RelationName<
       return new HasManyThroughPointerArrayPreloader(relation);
     case RelationType.HasManyThroughRelation:
       return new HasManyThroughRelationPreloader(relation);
+    case RelationType.HasOneThroughPointer:
+      return new HasOneThroughPointerPreloader(relation);
   }
 }
diff --git a/next/api/src/orm/relation.ts b/next/api/src/orm/relation.ts
index 4237a6aec..b5a6a18d8 100644
--- a/next/api/src/orm/relation.ts
+++ b/next/api/src/orm/relation.ts
@@ -17,6 +17,7 @@ export enum RelationType {
   HasManyThroughIdArray,
   HasManyThroughPointerArray,
   HasManyThroughRelation,
+  HasOneThroughPointer,
 }
 
 export interface BelongsTo {
@@ -79,6 +80,15 @@ export interface HasManyThroughRelation {
   relatedKey: string;
 }
 
+export interface HasOneThroughPointer {
+  type: RelationType.HasOneThroughPointer;
+  model: typeof Model;
+  field: string;
+  getRelatedModel: ModelGetter;
+  foreignPointerKey: string;
+  foreignKeyField: string;
+}
+
 export type Relation =
   | BelongsTo
   | BelongsToThroughPointer
@@ -86,4 +96,5 @@ export type Relation =
   | HasManyThroughPointer
   | HasManyThroughIdArray
   | HasManyThroughPointerArray
-  | HasManyThroughRelation;
+  | HasManyThroughRelation
+  | HasOneThroughPointer;
diff --git a/next/api/src/response/ticket.ts b/next/api/src/response/ticket.ts
index 75772153b..d1dfe463f 100644
--- a/next/api/src/response/ticket.ts
+++ b/next/api/src/response/ticket.ts
@@ -50,8 +50,7 @@ export class TicketListItemResponse extends BaseTicketResponse {
   toJSON(options?: TicketResponseOptions) {
     return {
       ...super.toJSON(options),
-      // 因为限定 notification.user 为当前用户，所以 notifications 最多只有一个元素
-      unreadCount: this.ticket.notifications?.[0]?.unreadCount,
+      unreadCount: this.ticket.notification?.unreadCount,
     };
   }
 }
diff --git a/next/api/src/router/ticket.ts b/next/api/src/router/ticket.ts
index b2a7be8ea..46e5f8831 100644
--- a/next/api/src/router/ticket.ts
+++ b/next/api/src/router/ticket.ts
@@ -256,7 +256,7 @@ router.get(
       finalQuery.preload('files');
     }
     if (params.includeUnreadCount) {
-      finalQuery.preload('notifications', {
+      finalQuery.preload('notification', {
         onQuery: (query) => {
           return query.where('user', '==', currentUser.toPointer());
         },