bug: graphql node query with only count field is slow when you have a large number of this kind of node in the database

[wvandeun]

Component

API Server / GraphQL

Infrahub version

0.16.0

Current Behavior

Executing the following GraphQL query is really slow (several minutes) , when you have 3k instances of InfraInterfaceL2 in the database

query {
  InfraInterfaceL2 {
    count
  }
}

Executing the same query, adding one additional field (attribute of the node) resolves instantly:

query {
  InfraInterfaceL2 {
    count
    edges {
      node {
        id
      }
    }
  }
}

Expected Behavior

A GraphQL query using just the count field for a particular node, should resolve much faster.

Steps to Reproduce

• load infrahub instance with demo schema inv dev.start demo.load-infra-schema
• load data into the system with this script infrahubctl run <script.py>

import logging
from infrahub_sdk import InfrahubClient


async def run(client: InfrahubClient, log: logging.Logger, branch: str, num_devices: int=30) -> None:
    site = await client.create("LocationSite", name="atl1")
    await site.save(allow_upsert=True)

    num_devices = int(num_devices)

    device_batch = await client.create_batch()
    interface_batch = await client.create_batch()

    for i in range(num_devices):
        device = await client.create("InfraDevice", name=f"atl1-test{i}", site=site, type="testing")
        device_batch.add(task=device.save, node=device, allow_upsert=True)
        log.info(f"Added device {device.name.value}")
        
    async for node, result in device_batch.execute():
        print(f"device {node.name.value} was created in Infrahub succesfully")
        client.store.set(key=node.name.value, node=node)

    for i in range(num_devices):
        for j in range(100):
            interface = await client.create("InfraInterfaceL2", name=f"Ethernet{j}", l2_mode="Access", speed=10000, device=client.store.get(key=f"atl1-test{i}"))
            interface_batch.add(task=interface.save, node=interface, allow_upsert=True)
            log.info(f"  Added interface {interface.name.value} for device {interface.device.peer.name.value}")

    async for node, result in interface_batch.execute():
        print(f"interface {node.name.value} {node.device.peer.name.value} was created in Infrahub succesfully")

• execute the following GraphQL query

query {
  InfraInterfaceL2 {
    count
  }
}

Additional Information

No response

[ogenstad]

Fixed in #4459.