You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
9ab1c34e02
Add chunk latency param for streaming api in web tests server; Make sure streaming api in web tests server can actually return response data by chunks in django asgi server. #jira UE-197485 [REVIEW] [at]michael.kirzinger [at]michael.atchison [at]rafa.lecina #rb michael.atchison, Michael.Kirzinger, Rafa.Lecina #tests Tried in WebTests project [CL 30669130 by lorry li in ue5-main branch]
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
import asyncio
|
|
from datetime import datetime
|
|
from django.shortcuts import render, redirect
|
|
from django.http.response import JsonResponse
|
|
from django.http import JsonResponse
|
|
from django.http import StreamingHttpResponse
|
|
from rest_framework.decorators import api_view, parser_classes
|
|
from rest_framework.parsers import FileUploadParser
|
|
import time
|
|
import logging
|
|
|
|
logger = logging.getLogger('django')
|
|
|
|
@api_view(['GET', 'POST', 'DELETE', 'PUT'])
|
|
def methods(request):
|
|
return JsonResponse({})
|
|
|
|
@api_view(['GET'])
|
|
def query_with_params(request):
|
|
var_int = int(request.GET.get('var_int'))
|
|
var_str = request.GET.get('var_str')
|
|
return JsonResponse({'var_int' : var_int, 'var_str' : var_str}, content_type="application/json")
|
|
|
|
@api_view(['GET'])
|
|
def get_large_response_without_chunks(request, bytes_number):
|
|
data = "d" * bytes_number
|
|
return JsonResponse({'data' : data})
|
|
|
|
@api_view(['GET'])
|
|
def streaming_download(request, chunks, chunk_size, chunk_latency):
|
|
async def data_chunk_generator(): # using 'async' for StreamingHttpResponse in asgi
|
|
for x in range(chunks):
|
|
if chunk_latency > 0:
|
|
logger.debug("[%s]sleeping %d seconds", datetime.now().strftime("%H:%M:%S:%f"), chunk_latency)
|
|
await asyncio.sleep(chunk_latency) # using 'await asyncio.sleep' instead of 'time.sleep' for StreamingHttpResponse in asgi
|
|
data_chunk = 'd' * chunk_size
|
|
logger.debug("[%s]sending %d bytes of data", datetime.now().strftime("%H:%M:%S:%f"), chunk_size)
|
|
yield data_chunk
|
|
response = StreamingHttpResponse(data_chunk_generator())
|
|
response['Content-Length'] = chunks * chunk_size
|
|
return response
|
|
|
|
def output_file_info(f):
|
|
if f.multiple_chunks():
|
|
# big file
|
|
total_chunks = 0
|
|
total_bytes = 0
|
|
for chunk in f.chunks():
|
|
total_chunks += 1
|
|
total_bytes += len(chunk)
|
|
logger.info("Received %d chunks: %d bytes in total", total_chunks, total_bytes)
|
|
else:
|
|
# small file
|
|
logger.info("Received the whole file: %d bytes", f.size)
|
|
|
|
@api_view(['POST'])
|
|
def streaming_upload_post(request):
|
|
f = request.FILES['file']
|
|
output_file_info(f)
|
|
return JsonResponse({})
|
|
|
|
@api_view(['PUT'])
|
|
@parser_classes((FileUploadParser,))
|
|
def streaming_upload_put(request):
|
|
f = request.data['file']
|
|
output_file_info(f)
|
|
return JsonResponse({})
|
|
|
|
@api_view(['GET'])
|
|
def redirect_from(request):
|
|
return redirect('redirect_to')
|
|
|
|
@api_view(['GET'])
|
|
def redirect_to(request):
|
|
return JsonResponse({})
|
|
|
|
@api_view(['GET'])
|
|
def mock_latency(request, latency):
|
|
time.sleep(latency)
|
|
return JsonResponse({})
|