@rpcbase/server 0.518.0 → 0.520.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";
@@ -4563,6 +4563,13 @@ const hasSessionUser = (req) => {
 const escapeForInlineScript = (value) => {
   return value.replace(/</g, "\\u003c").replace(/\u2028/g, "\\u2028").replace(/\u2029/g, "\\u2029");
 };
+const toSerializedHydrationPayload = (payload) => {
+  try {
+    return JSON.parse(JSON.stringify(payload));
+  } catch {
+    return null;
+  }
+};
 const createRtsSsrCollector = (req, opts) => {
   const maxQueries = DEFAULT_MAX_QUERIES;
   const maxDocsPerQuery = DEFAULT_MAX_DOCS_PER_QUERY;
@@ -4570,6 +4577,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 +4598,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 +4632,9 @@ const createRtsSsrCollector = (req, opts) => {
           data: boundedData,
           ...result.pageInfo ? {
             pageInfo: result.pageInfo
+          } : {},
+          ...typeof result.totalCount === "number" ? {
+            totalCount: result.totalCount
           } : {}
         });
       } catch {
@@ -4642,13 +4656,17 @@ const createRtsSsrCollector = (req, opts) => {
         break;
       }
     }
+    const serializedPayload = toSerializedHydrationPayload(payload);
+    if (!serializedPayload?.queries.length) return null;
     resolved.clear();
     resolvedPageInfo.clear();
-    for (const entry of payload.queries) {
+    resolvedTotalCount.clear();
+    for (const entry of serializedPayload.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;
+    return serializedPayload;
   };
   const hasRegistrations = () => registrations.size > 0;
   return {