MNPC/RasadDam_Backend
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix - improve response time of org linked ranchers

Browse Source
This commit is contained in:
Mojtaba-z
2026-01-04 10:39:02 +03:30
parent abf1c12e39
commit 627075735c
4 changed files with 62 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
apps/herd/models.py
View File

@@ -16,6 +16,7 @@ class Herd(BaseModel):
'Rancher',
on_delete=models.CASCADE,
related_name='herd',
db_index=True,
null=True
)
cooperative = models.ForeignKey(
@@ -153,12 +154,14 @@ class RancherOrganizationLink(BaseModel):
Organization,
on_delete=models.CASCADE,
related_name='rancher_links',
db_index=True,
null=True
)
rancher = models.ForeignKey(
Rancher,
on_delete=models.CASCADE,
related_name='organization_links',
db_index=True,
null=True
)

85
apps/herd/services/rancher_org_link_services.py
View File

@@ -1,8 +1,11 @@
from collections import defaultdict
from django.db.models import Count
from apps.authentication.models import Organization
from apps.authentication.services.service import get_all_org_child
from apps.herd.models import RancherOrganizationLink
from apps.herd.models import RancherOrganizationLink, Herd
from apps.livestock.models import LiveStock
# class RancherOrganizationService:
@@ -63,15 +66,18 @@ class RancherOrganizationService:
def orgs_linked_rancher(self, org: Organization, org_type_key: str):
# --------------------------------------------------
# 1. Base organizations queryset
# 1. Resolve organization ids
# --------------------------------------------------
if org.type.key != 'ADM':
child_orgs = get_all_org_child(org)
org_ids = [o.id for o in child_orgs]
org_ids = list(
o.id for o in get_all_org_child(org)
)
else:
org_ids = Organization.objects.filter(
type__key=org_type_key
).values_list('id', flat=True)
org_ids = list(
Organization.objects.filter(
type__key=org_type_key
).values_list('id', flat=True)
)
organizations = (
Organization.objects
@@ -84,12 +90,12 @@ class RancherOrganizationService:
# 2. Rancher count per organization
# --------------------------------------------------
rancher_counts = {
row['organization_id']: row['cnt']
for row in (
r['organization_id']: r['cnt']
for r in (
RancherOrganizationLink.objects
.filter(organization_id__in=org_ids)
.values('organization_id')
.annotate(cnt=Count('rancher_id', distinct=True))
.annotate(cnt=Count('rancher_id'))
)
}
@@ -97,39 +103,54 @@ class RancherOrganizationService:
# 3. Herd count per organization
# --------------------------------------------------
herd_counts = {
row['organization_id']: row['cnt']
for row in (
r['organization_id']: r['cnt']
for r in (
RancherOrganizationLink.objects
.filter(organization_id__in=org_ids)
.values('organization_id')
.annotate(cnt=Count('rancher__herd', distinct=True))
.annotate(cnt=Count('rancher__herd'))
)
}
# --------------------------------------------------
# 4. Livestock count per organization
# 4. Livestock count (optimized path)
# LiveStockHerd -> Herd -> Org
# --------------------------------------------------
livestock_counts = {
row['organization_id']: row['cnt']
# 4-1 livestock count per herd (NO JOIN, NO DISTINCT)
livestock_per_herd = {
row['herd_id']: row['cnt']
for row in (
RancherOrganizationLink.objects
.filter(organization_id__in=org_ids)
.values('organization_id')
.annotate(
cnt=Count(
'rancher__herd__live_stock_herd',
distinct=True
)
)
LiveStock.objects
.values('herd_id')
.annotate(cnt=Count('id'))
)
}
# 4-2 sum livestock per organization
livestock_counts = defaultdict(int)
herd_org_map = (
Herd.objects
.filter(
rancher__organization_links__organization_id__in=org_ids
)
.values(
'id',
'rancher__organization_links__organization_id'
)
)
for row in herd_org_map:
livestock_counts[
row['rancher__organization_links__organization_id']
] += livestock_per_herd.get(row['id'], 0)
# --------------------------------------------------
# 5. Final response (merge in memory)
# 5. Final response
# --------------------------------------------------
response = []
for org in organizations:
response.append({
return [
{
"id": org.id,
"name": org.name,
"org_service_area": [
@@ -145,6 +166,6 @@ class RancherOrganizationService:
"rancher_count": rancher_counts.get(org.id, 0),
"herd_count": herd_counts.get(org.id, 0),
"livestock_count": livestock_counts.get(org.id, 0),
})
return response
}
for org in organizations
]