Fix read model id field

Author: mahenzon

fastapi_jsonapi/data_layers/sqla/base_model.py

@@ -11,6 +11,7 @@
 from fastapi_jsonapi.data_layers.sqla.query_building import RelationshipInfo
 from fastapi_jsonapi.data_typing import TypeModel
 from fastapi_jsonapi.exceptions import BadRequest, InternalServerError, ObjectNotFound
+from fastapi_jsonapi.storages.models_storage import models_storage
 
 log = logging.getLogger(__name__)
 
@@ -97,8 +98,10 @@ async def count(
         cls,
         session: AsyncSession,
         stmt: Select,
+        resource_type: str,
     ) -> int:
-        stmt = select(func.count(distinct(column("id")))).select_from(stmt.subquery())
+        id_col = models_storage.get_model_id_field_name(resource_type)
+        stmt = select(func.count(distinct(column(id_col)))).select_from(stmt.subquery())
         return (await session.execute(stmt)).scalar_one()
 
     @classmethod

fastapi_jsonapi/data_layers/sqla/orm.py

@@ -428,6 +428,7 @@ async def get_collection(
             objects_count = await self._base_sql.count(
                 session=self.session,
                 stmt=query,
+                resource_type=self.resource_type,
             )
 
         collection = await self.after_get_collection(collection, qs, view_kwargs)