@sanity/client 6.20.1 → 6.20.2-beta.0

package/dist/index.browser.cjs

@@ -872,6 +872,55 @@ function optionsFromFile(opts, file) {
     opts
   );
 }
+const UNSHARDED_URL_RE = /^https:\/\/([a-z0-9]+)\.api\.(sanity\..*)/, SHARDED_URL_RE = /^https:\/\/[a-z0-9]+\.api\.s(\d+)\.sanity\.(.*)/, domainSharder = getDomainSharder();
+function getDomainSharder(initialBuckets) {
+  const buckets = new Array(10).fill(0, 0);
+  function incrementBucketForUrl(url) {
+    const shard = getShardFromUrl(url);
+    shard !== null && buckets[shard]++;
+  }
+  function decrementBucketForUrl(url) {
+    const shard = getShardFromUrl(url);
+    shard !== null && buckets[shard]--;
+  }
+  function getShardedUrl(url) {
+    const [isMatch, projectId2, rest] = url.match(UNSHARDED_URL_RE) || [];
+    if (!isMatch)
+      return url;
+    const bucket = buckets.reduce(
+      (smallest, count, index) => count < buckets[smallest] ? index : smallest,
+      0
+    );
+    return `https://${projectId2}.api.s${bucket}.${rest}`;
+  }
+  function getShardFromUrl(url) {
+    const [isMatch, shard] = url.match(SHARDED_URL_RE) || [];
+    return isMatch ? parseInt(shard, 10) : null;
+  }
+  return {
+    middleware: {
+      processOptions: (options) => {
+        if (!useDomainSharding(options))
+          return options;
+        const url = getShardedUrl(options.url);
+        return options.url = url, options;
+      },
+      onRequest(req) {
+        return useDomainSharding(req.options) && incrementBucketForUrl(req.options.url), req;
+      },
+      onResponse(res, context) {
+        return useDomainSharding(context.options) && decrementBucketForUrl(context.options.url), res;
+      }
+    },
+    incrementBucketForUrl,
+    decrementBucketForUrl,
+    getShardedUrl,
+    getBuckets: () => buckets
+  };
+}
+function useDomainSharding(options) {
+  return "useDomainSharding" in options && options.useDomainSharding === !0;
+}
 var defaults = (obj, defaults2) => Object.keys(defaults2).concat(Object.keys(obj)).reduce((target, prop) => (target[prop] = typeof obj[prop] > "u" ? defaults2[prop] : obj[prop], target), {});
 const pick = (obj, props) => props.reduce((selection, prop) => (typeof obj[prop] > "u" || (selection[prop] = obj[prop]), selection), {}), MAX_URL_LENGTH = 14800, possibleOptions = [
   "includePreviousRevision",
@@ -883,15 +932,16 @@ const pick = (obj, props) => props.reduce((selection, prop) => (typeof obj[prop]
   includeResult: !0
 };
 function _listen(query, params, opts = {}) {
-  const { url, token, withCredentials, requestTagPrefix } = this.config(), tag = opts.tag && requestTagPrefix ? [requestTagPrefix, opts.tag].join(".") : opts.tag, options = { ...defaults(opts, defaultOptions), tag }, listenOpts = pick(options, possibleOptions), qs = encodeQueryString({ query, params, options: { tag, ...listenOpts } }), uri = `${url}${_getDataUrl(this, "listen", qs)}`;
-  if (uri.length > MAX_URL_LENGTH)
+  const { url, token, withCredentials, requestTagPrefix } = this.config(), tag = opts.tag && requestTagPrefix ? [requestTagPrefix, opts.tag].join(".") : opts.tag, options = { ...defaults(opts, defaultOptions), tag }, listenOpts = pick(options, possibleOptions), qs = encodeQueryString({ query, params, options: { tag, ...listenOpts } });
+  let uri = `${url}${_getDataUrl(this, "listen", qs)}`;
+  if (this.config().useDomainSharding && (uri = domainSharder.getShardedUrl(uri)), uri.length > MAX_URL_LENGTH)
     return new rxjs.Observable((observer) => observer.error(new Error("Query too large for listener")));
   const listenFor = options.events ? options.events : ["mutation"], shouldEmitReconnect = listenFor.indexOf("reconnect") !== -1, esOptions = {};
   return (token || withCredentials) && (esOptions.withCredentials = !0), token && (esOptions.headers = {
     Authorization: `Bearer ${token}`
   }), new rxjs.Observable((observer) => {
     let es, reconnectTimer, stopped = !1, unsubscribed = !1;
-    open();
+    domainSharder.incrementBucketForUrl(uri), open();
     function onError() {
       stopped || (emitReconnect(), !stopped && es.readyState === es.CLOSED && (unsubscribe(), clearTimeout(reconnectTimer), reconnectTimer = setTimeout(open, 100)));
     }
@@ -926,7 +976,7 @@ function _listen(query, params, opts = {}) {
       });
     }
     function stop() {
-      stopped = !0, unsubscribe(), unsubscribed = !0;
+      stopped = !0, unsubscribe(), unsubscribed = !0, domainSharder.decrementBucketForUrl(uri);
     }
     return stop;
   });
@@ -1537,6 +1587,7 @@ function defineCreateClientExports(envMiddleware2, ClassConstructor) {
       maxRedirects: 0,
       maxRetries: config.maxRetries,
       retryDelay: config.retryDelay,
+      useDomainSharding: config.useDomainSharding,
       ...options
     }),
     config
@@ -1547,7 +1598,7 @@ function defineDeprecatedCreateClient(createClient2) {
     return printNoDefaultExport(), createClient2(config);
   };
 }
-var envMiddleware = [];
+var envMiddleware = [domainSharder.middleware];
 const exp = defineCreateClientExports(envMiddleware, SanityClient), requester = exp.requester, createClient = exp.createClient, deprecatedCreateClient = defineDeprecatedCreateClient(createClient);
 Object.defineProperty(exports, "unstable__adapter", {
   enumerable: !0,