Update server.py

server.py

@@ -5,7 +5,105 @@ import threading
 import time
 from collections import deque
 from datetime import datetime, timedelta
+import loggingfrom flask import Flask, request, jsonify, Response
+import random
+import string
+import threading
+import time
+from datetime import datetime, timedelta
 import logging
+from gevent import sleep
+
+app = Flask(__name__)
+
+# Конфигурация
+TRANSFER_LIFETIME = timedelta(hours=6)
+CLEANUP_INTERVAL = 300
+MAX_CHUNK_SIZE = 100 * 1024 * 1024
+KEEP_ALIVE_INTERVAL = 25
+TRANSFER_TIMEOUT = 3600 * 3
+BUFFER_FLUSH_SIZE = 100 * 1024 * 1024 # 10MB
+
+# Логирование
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger('FileTransfer')
+
+# Хранилище данных
+transfers = {}
+transfer_data = {}
+transfer_lock = threading.Lock()
+
+def cleanup_task():
+    """Фоновая очистка старых передач"""
+    while True:
+        time.sleep(CLEANUP_INTERVAL)
+        with transfer_lock:
+            now = datetime.now()
+            to_delete = [
+                tid for tid, info in transfers.items()
+                if info['completed'] and (now - info['completed_time']) > TRANSFER_LIFETIME
+            ]
+            for tid in to_delete:
+                del transfers[tid]
+                del transfer_data[tid]
+                logger.info(f"Cleaned transfer: {tid}")
+
+threading.Thread(target=cleanup_task, daemon=True).start()
+
+@app.route('/download/<transfer_id>', methods=['GET'])
+def download_file(transfer_id):
+    """Потоковая отдача файла с оптимизацией буферизации"""
+    def generate():
+        buffer = bytearray()
+        last_flush = time.time()
+        
+        try:
+            while True:
+                with transfer_lock:
+                    transfer = transfers.get(transfer_id)
+                    if not transfer or (transfer['completed'] and not transfer_data.get(transfer_id)):
+                        break
+                    
+                    chunks = transfer_data.get(transfer_id, [])
+                    while chunks:
+                        buffer.extend(chunks.pop(0))
+                        
+                        # Отправляем буфер при достижении размера или таймаута
+                        if (len(buffer) >= BUFFER_FLUSH_SIZE or 
+                            (time.time() - last_flush) > KEEP_ALIVE_INTERVAL):
+                            yield bytes(buffer)
+                            buffer.clear()
+                            last_flush = time.time()
+                
+                # Асинхронное ожидание новых данных
+                sleep(0.1)
+            
+            # Отправить оставшиеся данные
+            if buffer:
+                yield bytes(buffer)
+                
+        except Exception as e:
+            logger.error(f"Download error: {transfer_id} - {str(e)}")
+        finally:
+            logger.info(f"Download finished: {transfer_id}")
+
+    return Response(
+        generate(),
+        mimetype='application/octet-stream',
+        headers={
+            'Content-Disposition': f'attachment; filename="{transfers[transfer_id]["filename"]}"',
+            'Cache-Control': 'no-store'
+        }
+    )
+
+if __name__ == '__main__':
+    from gevent.pywsgi import WSGIServer
+    logger.info("Starting optimized server...")
+    http_server = WSGIServer(('0.0.0.0', 5000), app)
+    http_server.serve_forever()
 
 app = Flask(__name__)