Fixed cluster configuration

elasticsearch/config/config_cluster.sh

@@ -3,39 +3,36 @@
 
 elastic_config_file="/usr/share/elasticsearch/config/elasticsearch.yml"
 
+remove_single_node_conf(){
+  if grep -Fq "discovery.type" $1; then
+    sed -i '/discovery.type\: /d' $1 
+  fi
+}
 
-# If Elasticsearch cluster is enable
-if [[ $ELASTIC_CLUSTER == "true" ]]
-then
+# If Elasticsearch cluster is enable, then set up the elasticsearch.yml
+if [[ $ELASTIC_CLUSTER == "true" && $CLUSTER_NODE_MASTER != "" && $CLUSTER_NODE_DATA != "" && $CLUSTER_NODE_INGEST != "" ]];then
 
-  # Set the cluster.name and discovery.zen.minimun_master_nodes variables
-  sed -i 's:cluster.name\: "docker-cluster":cluster.name\: "'$CLUSTER_NAME'":g' $elastic_config_file
-  #sed -i 's:discovery.zen.minimum_master_nodes\: 1:discovery.zen.minimum_master_nodes\: '$CLUSTER_NUMBER_OF_MASTERS':g' $elastic_config_file
+  remove_single_node_conf $elastic_config_file
 
-  # Add the cluster configuration
-  echo "
+  # Remove the old configuration
+  sed -i '/# cluster node/,/# end cluster config/d' $elastic_config_file
+
+  # Add the current cluster configuration
+cat > $elastic_config_file << EOF
 # cluster node
-node:
-  master: ${CLUSTER_NODE_MASTER}
-  data: ${CLUSTER_NODE_DATA}
-  ingest: ${CLUSTER_NODE_INGEST}
-  name: ${CLUSTER_NODE_NAME}
-  max_local_storage_nodes: ${CLUSTER_MAX_NODES}
-
-bootstrap:
-  memory_lock: ${CLUSTER_MEMORY_LOCK} 
+network.host: 0.0.0.0
+node.name: $CLUSTER_NODE_NAME
+node.master: $CLUSTER_NODE_MASTER
 
 cluster.initial_master_nodes: 
-  - '${CLUSTER_INITIAL_MASTER_NODES}'
-
-" >> $elastic_config_file
-else
-
-cat >> $elastic_config_file <<'EOF'
-cluster.initial_master_nodes:
-  - 'elasticsearch'
+  - $CLUSTER_INITIAL_MASTER_NODES
+# end cluster config" 
 EOF
 
-# echo 'discovery.type: single-node'
-
+# If the cluster is disabled, then set a single-node configuration
+else
+  sed -i '/# cluster node/,/# end cluster config/d' $elastic_config_file
+  # If it's not already configured
+  remove_single_node_conf $elastic_config_file
+  echo "discovery.type: single-node" >> $elastic_config_file
 fi