94 lines
3.3 KiB
Bash
Executable File
94 lines
3.3 KiB
Bash
Executable File
#!/bin/sh
|
||
set -e
|
||
|
||
if [[ "$ES_ADDR" == *"localhost"* || "$ES_ADDR" == *"127.0.0.1"* ]]; then
|
||
echo "ES_ADDR is localhost, using docker address: http://elasticsearch:9200"
|
||
ES_ADDR="http://elasticsearch:9200"
|
||
fi
|
||
# ES_ADDR=http://localhost:31160
|
||
INDEX_DIR=/es_index_schema
|
||
|
||
echo "ES_ADDR: $ES_ADDR"
|
||
|
||
AUTH_PARAM=""
|
||
if [ -n "$ES_USERNAME" ]; then
|
||
AUTH_PARAM="-k -u $ES_USERNAME:$ES_PASSWORD"
|
||
fi
|
||
|
||
for i in $(seq 1 60); do
|
||
echo "Checking Elasticsearch availability... (attempt $i)"
|
||
if curl -s -f $AUTH_PARAM "${ES_ADDR}/_cat/health"; then
|
||
echo "Elasticsearch is up and running!"
|
||
break
|
||
fi
|
||
echo "Elasticsearch not available, retrying in 1 seconds..."
|
||
sleep 1
|
||
done
|
||
|
||
echo -e "🔍 Checking smartcn plugin status..."
|
||
if ! curl -s $AUTH_PARAM "${ES_ADDR}/_cat/plugins" | grep -q "analysis-smartcn"; then
|
||
echo -e "❌ smartcn plugin not loaded correctly, please ensure the plugin is installed and Elasticsearch is restarted"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "🔍 Initializing Elasticsearch index templates..."
|
||
ES_TEMPLATES=$(find "$INDEX_DIR" -type f -name "*.index-template.json" | sort)
|
||
if [ -z "$ES_TEMPLATES" ]; then
|
||
echo -e "ℹ️ No Elasticsearch index templates found in $INDEX_DIR"
|
||
exit 1
|
||
else
|
||
# Add index creation logic
|
||
echo -e "🔄 Creating Elasticsearch indexes..."
|
||
for template_file in $ES_TEMPLATES; do
|
||
|
||
template_name=$(basename "$template_file" | sed 's/\.index-template\.json$//')
|
||
echo -e "➡️ Registering template: $template_name"
|
||
|
||
# Attempt to register index template
|
||
response=$(curl -s $AUTH_PARAM -X PUT "${ES_ADDR}/_index_template/$template_name" \
|
||
-H "Content-Type: application/json" \
|
||
-d @"$template_file" 2>&1)
|
||
|
||
# Check if successful
|
||
if echo "$response" | grep -q '"acknowledged":true'; then
|
||
echo -e "✅ Template $template_name registered successfully"
|
||
else
|
||
echo -e "❌ Failed to register template $template_name. Response: $response"
|
||
exit 1
|
||
fi
|
||
|
||
index_name=$(basename "$template_file" | sed 's/\.index-template\.json$//')
|
||
echo -e "➡️ Creating index: $index_name"
|
||
|
||
# Check if index exists
|
||
if ! curl -s -f $AUTH_PARAM "${ES_ADDR}/_cat/indices/$index_name" >/dev/null; then
|
||
# Create index (matching template's index_patterns)
|
||
curl $AUTH_PARAM -X PUT "${ES_ADDR}/$index_name" -H "Content-Type: application/json"
|
||
echo ""
|
||
|
||
# Set refresh interval if index was just created
|
||
echo -e "🔄 Setting refresh_interval for index: $index_name..."
|
||
CURL_OUTPUT=$(curl -s $AUTH_PARAM -w "\nHTTP_STATUS_CODE:%{http_code}" -X PUT "${ES_ADDR}/${index_name}/_settings" -H 'Content-Type: application/json' -d'
|
||
{
|
||
"index": {
|
||
"refresh_interval": "10ms"
|
||
}
|
||
}')
|
||
echo -e "📄 Curl command output for $index_name:\n$CURL_OUTPUT"
|
||
# Extract the JSON body from the output, excluding the HTTP_STATUS_CODE line
|
||
JSON_BODY=$(echo "$CURL_OUTPUT" | sed '$d')
|
||
if ! echo "$JSON_BODY" | grep -q '"acknowledged":true'; then
|
||
echo -e "⚠️ Warning: Failed to set refresh interval for $index_name index. Response Body: $JSON_BODY. Please check and set manually."
|
||
exit 1
|
||
else
|
||
echo -e "✅ Successfully set refresh_interval for $index_name."
|
||
fi
|
||
else
|
||
echo -e "ℹ️ Index $index_name already exists"
|
||
fi
|
||
done
|
||
fi
|
||
|
||
echo "Elasticsearch setup completed."
|
||
exit 0
|