From cc6577f6aa2cf26e2621b13163276d3128ee85bc Mon Sep 17 00:00:00 2001
From: dfkadyr-stakewise <kadyr@stakewise.io>
Date: Fri, 24 Jan 2025 12:34:23 +0300
Subject: [PATCH] [limit] add limit to vault & user stats

---
 src/graphql/subgraph/vault/userRewardsQuery.graphql | 2 ++
 src/graphql/subgraph/vault/vaultStatsQuery.graphql  | 3 ++-
 src/methods/vault/requests/getUserStats/index.ts    | 1 +
 src/methods/vault/requests/getVaultStats/index.ts   | 1 +
 4 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/graphql/subgraph/vault/userRewardsQuery.graphql b/src/graphql/subgraph/vault/userRewardsQuery.graphql
index a7926481..0f5eaa60 100644
--- a/src/graphql/subgraph/vault/userRewardsQuery.graphql
+++ b/src/graphql/subgraph/vault/userRewardsQuery.graphql
@@ -1,8 +1,10 @@
 query UserRewards(
   $where: AllocatorStats_filter
+  $limit: Int
 ) {
   allocator: allocatorStats_collection(
     interval: day
+    first: $limit
     where: $where
   ) {
     apy
diff --git a/src/graphql/subgraph/vault/vaultStatsQuery.graphql b/src/graphql/subgraph/vault/vaultStatsQuery.graphql
index 5ff1e1a0..03116441 100644
--- a/src/graphql/subgraph/vault/vaultStatsQuery.graphql
+++ b/src/graphql/subgraph/vault/vaultStatsQuery.graphql
@@ -1,6 +1,7 @@
-query VaultStats($vaultAddress: String!  $timestamp: Timestamp!) {
+query VaultStats($vaultAddress: String!  $timestamp: Timestamp!, $limit: Int) {
   vaultStats: vaultStats_collection(
     interval: day
+    first: $limit
     where: {
       vault: $vaultAddress
       timestamp_gte: $timestamp,
diff --git a/src/methods/vault/requests/getUserStats/index.ts b/src/methods/vault/requests/getUserStats/index.ts
index 4fce8107..00f8fb79 100644
--- a/src/methods/vault/requests/getUserStats/index.ts
+++ b/src/methods/vault/requests/getUserStats/index.ts
@@ -19,6 +19,7 @@ const getUserStats = (input: GetUserStatsInput) => {
   return graphql.subgraph.vault.fetchUserRewardsQuery({
     url: apiUrls.getSubgraphqlUrl(options),
     variables: {
+      limit: daysCount,
       where: {
         timestamp_gte: String(getTimestamp(daysCount)),
         allocator_: {
diff --git a/src/methods/vault/requests/getVaultStats/index.ts b/src/methods/vault/requests/getVaultStats/index.ts
index 043f0dab..15a6cae6 100644
--- a/src/methods/vault/requests/getVaultStats/index.ts
+++ b/src/methods/vault/requests/getVaultStats/index.ts
@@ -21,6 +21,7 @@ const getVaultStats = (input: GetVaultStatsInput) => {
     url: apiUrls.getSubgraphqlUrl(options),
     variables: {
       timestamp,
+      limit: daysCount,
       vaultAddress: vaultAddress.toLowerCase(),
     },
     modifyResult: modifyVaultStats,