server_lifecycle.py

#!/usr/bin/env python3

from pymongo import MongoClient
import json
from kafka import KafkaProducer
import time
from flask import Flask, request, jsonify


client = MongoClient(
    "mongodb+srv://mongo2mongo:test123@cluster0.7ik1k.mongodb.net/myFirstDatabase?retryWrites=true&w=majority")

db = client["initialiser_db"]
collection = db["services"]

services = []
for x in collection.find():
    services.append(x["service"])

print(services)
# print("i am here")

# get dynamic ip
# ip1 = "52.140.61.100"
ip1 = "20.207.107.115"
ip2 = "20.204.220.249"


def send_message(topic_name, message):
    producer = KafkaProducer(bootstrap_servers=[
                             '20.219.102.74:9092'], value_serializer=lambda v: json.dumps(v).encode('utf-8'))
    producer.send(topic_name, message)


no_of_services = len(services)
for i in range(0, len(services)-1, 2):
    data1 = {
        "command": "START",
        "service": services[i]
    }
    data2 = {
        "command": "START",
        "service": services[i+1]
    }
    # data3 = {
    # "command":"START",
    # "service":services[i+2]
    # }
    send_message("service_{}".format(ip1), data1)
    send_message("service_{}".format(ip2), data2)
    # send_message("service_{}".format(ip3), data3)
    print("service_{}".format(ip1))
    print("service_{}".format(ip2))
    # print("service_{}".format(ip3))
    time.sleep(2)

# data3 = {
# "command":"START",
# "service":services[len(services)-1]
# }
# send_message("service_{}".format(ip1), data3)
# time.sleep(2)


app = Flask(__name__)

collection1 = db["running_services"]


@app.route('/initialiser/getService/<serviceName>', methods=['GET'])
def getServiceAddress(serviceName):
    # print("hello")
    doc = collection1.find_one({"name": serviceName})
    ip = doc["ip"]
    port = doc["port"]
    return jsonify({"ip": ip, "port": port})


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5010)