Create entrypoint.sh

entrypoint.sh

@@ -0,0 +1,38 @@
+#!/bin/bash
+set -e
+
+SNAPSHOT_DIR="/data/qdrant_snapshots"
+QDRANT_URL="http://localhost:6333"
+
+# Start Qdrant in background first
+/qdrant/qdrant &
+QDRANT_PID=$!
+
+# Wait for Qdrant to be ready
+echo "Waiting for Qdrant to start..."
+until curl -sf "$QDRANT_URL/readyz" > /dev/null; do
+    sleep 1
+done
+echo "Qdrant ready."
+
+# Recover any snapshot not yet loaded
+for SNAPSHOT in "$SNAPSHOT_DIR"/*.snapshot; do
+    [ -f "$SNAPSHOT" ] || continue
+    COLLECTION=$(basename "$SNAPSHOT" .snapshot)
+
+    # Check if collection already exists
+    EXISTS=$(curl -sf "$QDRANT_URL/collections/$COLLECTION" | python3 -c "import sys,json; print(json.load(sys.stdin)['result']['status'])" 2>/dev/null || echo "missing")
+
+    if [ "$EXISTS" != "green" ]; then
+        echo "Recovering collection: $COLLECTION"
+        curl -sf -X POST "$QDRANT_URL/collections/$COLLECTION/snapshots/recover" \
+            -H "Content-Type: application/json" \
+            -d "{\"location\": \"file://$SNAPSHOT\"}"
+        echo "Recovered: $COLLECTION"
+    else
+        echo "Collection $COLLECTION already exists, skipping."
+    fi
+done
+
+# Hand control back to Qdrant process
+wait $QDRANT_PID