@rpcbase/server 0.518.0 → 0.519.0

package/dist/index.js

@@ -9,7 +9,7 @@ import { dirname, posix, sep } from "path";
 import fs, { createReadStream, readFileSync } from "node:fs";
 import { createInterface } from "node:readline";
 import { AsyncLocalStorage } from "node:async_hooks";
-import { g as getDerivedKey, r as resolveRtsRequestTenantId, i as isRtsRequestAuthorized, b as buildRtsAbilityFromRequest, n as normalizeRtsQueryOptions, a as runRtsQuery } from "./queryExecutor-B7lb2FR1.js";
+import { g as getDerivedKey, r as resolveRtsRequestTenantId, i as isRtsRequestAuthorized, b as buildRtsAbilityFromRequest, n as normalizeRtsQueryOptions, a as runRtsQuery } from "./queryExecutor-Bg1GGL3j.js";
 import httpProxy from "http-proxy-3";
 import fsPromises from "node:fs/promises";
 import inspector from "node:inspector";
@@ -4570,6 +4570,7 @@ const createRtsSsrCollector = (req, opts) => {
   const registrations = /* @__PURE__ */ new Map();
   const resolved = /* @__PURE__ */ new Map();
   const resolvedPageInfo = /* @__PURE__ */ new Map();
+  const resolvedTotalCount = /* @__PURE__ */ new Map();
   const runtime = {
     registerQuery(query) {
       const modelName = typeof query.modelName === "string" ? query.modelName.trim() : "";
@@ -4590,6 +4591,9 @@ const createRtsSsrCollector = (req, opts) => {
     },
     getQueryPageInfo(modelName, queryKey) {
       return resolvedPageInfo.get(makeRegistrationKey(modelName, queryKey));
+    },
+    getQueryTotalCount(modelName, queryKey) {
+      return resolvedTotalCount.get(makeRegistrationKey(modelName, queryKey));
     }
   };
   const resolve = async () => {
@@ -4621,6 +4625,9 @@ const createRtsSsrCollector = (req, opts) => {
           data: boundedData,
           ...result.pageInfo ? {
             pageInfo: result.pageInfo
+          } : {},
+          ...typeof result.totalCount === "number" ? {
+            totalCount: result.totalCount
           } : {}
         });
       } catch {
@@ -4644,9 +4651,11 @@ const createRtsSsrCollector = (req, opts) => {
     }
     resolved.clear();
     resolvedPageInfo.clear();
+    resolvedTotalCount.clear();
     for (const entry of payload.queries) {
       resolved.set(makeRegistrationKey(entry.modelName, entry.queryKey), entry.data);
       resolvedPageInfo.set(makeRegistrationKey(entry.modelName, entry.queryKey), entry.pageInfo);
+      resolvedTotalCount.set(makeRegistrationKey(entry.modelName, entry.queryKey), entry.totalCount);
     }
     return payload.queries.length ? payload : null;
   };