gistfile1.py

__author__ = 'cristian'

from base import *


def pull_elasticsearch_image(pmk, ip):
    """

    @type param pmk: SSHClient
    :return:
    """

    exec_command(pmk, "docker pull dockerfile/elasticsearch", ip)


def run_container(pmk, server):

    exec_command(pmk, "docker run --restart=on-failure:10 -d -h {hostname} --name elasticsearch-{hostname} -d\
     -p {private_ip}:9200:9200 -p {private_ip}:9300:9300 -v /data/elasticsearch:/data \
     -e SERVICE_9200_CHECK_HTTP=/_cluster/health?pretty=true -e SERVICE_9200_CHECK_INTERVAL=60s \
     -e LOG=true dockerfile/elasticsearch /elasticsearch/bin/elasticsearch -Des.config=/data/elasticsearch.yml"
                 .format(**server), server.ip)


def create_data_volume(pmk, ip):
    exec_command(pmk, "mkdir -p /data/elasticsearch")
    exec_command(pmk, "echo 'path: {logs: /data/log, data: /data/data}' > /data/elasticsearch/elasticsearch.yml", ip)


def install_elasticsearch(server_name, server):
    """
        @type param server_name: str
        @type param server: dict
        :return:
    """

    if server_name != "master":
        return

    client = SSHClient()
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    client.connect(server["ip"], username=server["user"],  port=server["port"])

    pull_elasticsearch_image(client, server.ip)
    create_data_volume(client, server.ip)
    run_container(client, server)


execute = [install_elasticsearch]

if __name__ == "__main__":
    run_task(execute)