diff --git a/apps/enp/serializers.py b/apps/enp/serializers.py
index 787c55ce..d929cedd 100644
--- a/apps/enp/serializers.py
+++ b/apps/enp/serializers.py
@@ -139,7 +139,7 @@ class DrainEquipEnvSerializer(CustomModelSerializer):
         now = timezone.now()
         now_10_before = now - timezone.timedelta(minutes=10)
         obj = EnvData.objects.filter(
-            equipment=obj.equipment, time__gte=now_10_before, time__lte=now).order_by('-time').first()
+            equipment=obj.equipment, time__gte=now_10_before, time__lte=now).order_by('-time').select_related('equipment').first()
         if obj:
             return EnvDataSerializer(instance=obj).data
         else:
diff --git a/apps/enp/views.py b/apps/enp/views.py
index 2b5e0782..593b24d0 100644
--- a/apps/enp/views.py
+++ b/apps/enp/views.py
@@ -32,6 +32,8 @@ class DrainViewSet(CustomModelViewSet):
     @swagger_auto_schema(manual_parameters=[
         openapi.Parameter(name="has_equipdata", in_=openapi.IN_QUERY, description="Include equip data in the response",
                           type=openapi.TYPE_STRING, enum=["yes", "no"], required=False),
+        openapi.Parameter(name="query", in_=openapi.IN_QUERY, description="定制返回数据",
+                          type=openapi.TYPE_STRING, required=False),
     ])
     def list(self, request, *args, **kwargs):
         return super().list(request, *args, **kwargs)