feat: add server-side pagination to operation overrides with limit and offset parameters

connect/src/wg/cosmo/platform/v1/platform_pb.ts

@@ -13892,6 +13892,16 @@ export class GetAllOverridesRequest extends Message<GetAllOverridesRequest> {
    */
   namespace = "";
 
+  /**
+   * @generated from field: int32 limit = 3;
+   */
+  limit = 0;
+
+  /**
+   * @generated from field: int32 offset = 4;
+   */
+  offset = 0;
+
   constructor(data?: PartialMessage<GetAllOverridesRequest>) {
     super();
     proto3.util.initPartial(data, this);
@@ -13902,6 +13912,8 @@ export class GetAllOverridesRequest extends Message<GetAllOverridesRequest> {
   static readonly fields: FieldList = proto3.util.newFieldList(() => [
     { no: 1, name: "graph_name", kind: "scalar", T: 9 /* ScalarType.STRING */ },
     { no: 2, name: "namespace", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 3, name: "limit", kind: "scalar", T: 5 /* ScalarType.INT32 */ },
+    { no: 4, name: "offset", kind: "scalar", T: 5 /* ScalarType.INT32 */ },
   ]);
 
   static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): GetAllOverridesRequest {
@@ -13935,6 +13947,11 @@ export class GetAllOverridesResponse extends Message<GetAllOverridesResponse> {
    */
   overrides: GetAllOverridesResponse_Override[] = [];
 
+  /**
+   * @generated from field: int32 total_count = 3;
+   */
+  totalCount = 0;
+
   constructor(data?: PartialMessage<GetAllOverridesResponse>) {
     super();
     proto3.util.initPartial(data, this);
@@ -13945,6 +13962,7 @@ export class GetAllOverridesResponse extends Message<GetAllOverridesResponse> {
   static readonly fields: FieldList = proto3.util.newFieldList(() => [
     { no: 1, name: "response", kind: "message", T: Response },
     { no: 2, name: "overrides", kind: "message", T: GetAllOverridesResponse_Override, repeated: true },
+    { no: 3, name: "total_count", kind: "scalar", T: 5 /* ScalarType.INT32 */ },
   ]);
 
   static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): GetAllOverridesResponse {

controlplane/src/core/bufservices/check/getAllOverrides.ts

@@ -8,7 +8,7 @@ import {
 import { FederatedGraphRepository } from '../../repositories/FederatedGraphRepository.js';
 import { OperationsRepository } from '../../repositories/OperationsRepository.js';
 import type { RouterOptions } from '../../routes.js';
-import { enrichLogger, getLogger, handleError } from '../../util.js';
+import { clamp, enrichLogger, getLogger, handleError } from '../../util.js';
 import { UnauthorizedError } from '../../errors/errors.js';
 
 export function getAllOverrides(
@@ -33,6 +33,7 @@ export function getAllOverrides(
           details: 'Requested graph does not exist',
         },
         overrides: [],
+        totalCount: 0,
       };
     }
 
@@ -42,15 +43,22 @@ export function getAllOverrides(
 
     const operationsRepo = new OperationsRepository(opts.db, graph.id);
 
-    const overrides = await operationsRepo.getConsolidatedOverridesView({
+    // default to 10 if no limit is provided
+    const limit = clamp(req.limit || 10, 1, 50);
+    const offset = clamp(req.offset || 0, 0, 500_000);
+
+    const { overrides, totalCount } = await operationsRepo.getConsolidatedOverridesView({
       namespaceId: graph.namespaceId,
+      limit,
+      offset,
     });
 
     return {
       response: {
         code: EnumStatusCode.OK,
       },
       overrides,
+      totalCount,
     };
   });
 }

controlplane/src/core/repositories/OperationsRepository.ts

@@ -1,5 +1,5 @@
 import { OverrideChange } from '@wundergraph/cosmo-connect/dist/platform/v1/platform_pb';
-import { aliasedTable, and, asc, desc, eq, isNull, sql } from 'drizzle-orm';
+import { aliasedTable, and, asc, count, desc, eq, isNull, sql } from 'drizzle-orm';
 import { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
 import { PlainMessage } from '@bufbuild/protobuf';
 import { DBSchemaChangeType } from '../../db/models.js';
@@ -443,7 +443,7 @@ export class OperationsRepository {
     };
   }
 
-  public getConsolidatedOverridesView(data: { namespaceId: string }) {
+  public async getConsolidatedOverridesView(data: { namespaceId: string; limit: number; offset: number }) {
     const change = this.db
       .select({
         hash: schema.operationChangeOverrides.hash,
@@ -484,7 +484,7 @@ export class OperationsRepository {
     // We need to retrieve a consolidated view of overrides from both tables.
     // There is no guarantee that an entry for hash exists in both.
 
-    return this.db
+    const baseQuery = this.db
       .select({
         hash: sql<string>`coalesce(${change.hash}, ${ignore.hash})`,
         name: sql<string>`coalesce(${change.name}, ${ignore.name})`,
@@ -497,7 +497,23 @@ export class OperationsRepository {
       .from(change)
       .fullJoin(ignore, and(eq(change.hash, ignore.hash), eq(change.namespaceId, ignore.namespaceId)))
       .leftJoin(changeCounts, and(eq(change.hash, changeCounts.hash), eq(change.namespaceId, changeCounts.namespaceId)))
-      .orderBy(({ name, hash }) => [asc(name), asc(hash)]);
+      .orderBy(({ name, hash }) => [asc(name), asc(hash)])
+      .limit(data.limit)
+      .offset(data.offset);
+
+    const countQuery = this.db
+      .select({
+        count: count(),
+      })
+      .from(change)
+      .fullJoin(ignore, and(eq(change.hash, ignore.hash), eq(change.namespaceId, ignore.namespaceId)));
+
+    const [overrides, countResult] = await Promise.all([baseQuery, countQuery]);
+
+    return {
+      overrides,
+      totalCount: countResult[0]?.count ?? 0,
+    };
   }
 
   private static createPersistedOperationDTO({

controlplane/test/override.test.ts

@@ -240,4 +240,84 @@ describe('Overrides', (ctx) => {
     expect(overridesRes.overrides[1].changesOverrideCount).toBe(0);
     expect(overridesRes.overrides[1].hasIgnoreAllOverride).toBe(true);
   });
+
+  test('Should paginate overrides with limit and offset', async (testContext) => {
+    const { client, server } = await SetupTest({ dbname });
+    testContext.onTestFinished(() => server.close());
+
+    const fedGraphName = genID('fedGraph');
+    const label = genUniqueLabel();
+
+    const createFedGraphRes = await client.createFederatedGraph({
+      name: fedGraphName,
+      namespace: 'default',
+      routingUrl: 'http://localhost:8081',
+      labelMatchers: [joinLabel(label)],
+    });
+    expect(createFedGraphRes.response?.code).toBe(EnumStatusCode.OK);
+
+    const graphRes = await client.getFederatedGraphByName({
+      name: fedGraphName,
+      namespace: 'default',
+    });
+    expect(graphRes.response?.code).toBe(EnumStatusCode.OK);
+
+    // Create 3 overrides
+    for (let i = 1; i <= 3; i++) {
+      const res = await client.createOperationOverrides({
+        graphName: graphRes.graph?.name,
+        namespace: graphRes.graph?.namespace,
+        operationHash: `hash${i}`,
+        operationName: `op${i}`,
+        changes: [
+          {
+            changeType: 'FIELD_TYPE_CHANGED',
+            path: `A.field${i}`,
+          },
+        ],
+      });
+      expect(res.response?.code).toBe(EnumStatusCode.OK);
+    }
+
+    // Fetch all without pagination params — should return all 3
+    const allRes = await client.getAllOverrides({
+      graphName: graphRes.graph?.name,
+      namespace: graphRes.graph?.namespace,
+    });
+    expect(allRes.response?.code).toBe(EnumStatusCode.OK);
+    expect(allRes.totalCount).toBe(3);
+
+    // Fetch first page with limit 2
+    const page1 = await client.getAllOverrides({
+      graphName: graphRes.graph?.name,
+      namespace: graphRes.graph?.namespace,
+      limit: 2,
+      offset: 0,
+    });
+    expect(page1.response?.code).toBe(EnumStatusCode.OK);
+    expect(page1.overrides.length).toBe(2);
+    expect(page1.totalCount).toBe(3);
+
+    // Fetch second page with limit 2, offset 2
+    const page2 = await client.getAllOverrides({
+      graphName: graphRes.graph?.name,
+      namespace: graphRes.graph?.namespace,
+      limit: 2,
+      offset: 2,
+    });
+    expect(page2.response?.code).toBe(EnumStatusCode.OK);
+    expect(page2.overrides.length).toBe(1);
+    expect(page2.totalCount).toBe(3);
+
+    // Fetch with offset beyond total — should return empty but correct totalCount
+    const page3 = await client.getAllOverrides({
+      graphName: graphRes.graph?.name,
+      namespace: graphRes.graph?.namespace,
+      limit: 10,
+      offset: 10,
+    });
+    expect(page3.response?.code).toBe(EnumStatusCode.OK);
+    expect(page3.overrides.length).toBe(0);
+    expect(page3.totalCount).toBe(3);
+  });
 });

proto/wg/cosmo/platform/v1/platform.proto

@@ -1801,6 +1801,8 @@ message GetOperationOverridesResponse {
 message GetAllOverridesRequest {
   string graph_name = 1;
   string namespace = 2;
+  int32 limit = 3;
+  int32 offset = 4;
 }
 
 message GetAllOverridesResponse {
@@ -1814,6 +1816,7 @@ message GetAllOverridesResponse {
 
   Response response = 1;
   repeated Override overrides = 2;
+  int32 total_count = 3;
 }
 
 message IsGitHubAppInstalledRequest {