def post(self, request, *args, **kwargs):
telephone = request.data.get('telephone')
serializer = UserModelSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
user = serializer.save()
user.set_password(serializer.data['password'])
master = Master(user=user, telephone=telephone)
master.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
def register(request):
user_data = request.data.pop('user')
user_serializer = UserModelSerializer(data=user_data)
salon_data = request.data.pop('salon')
salon_serializer = SalonModelSerializer(salon_data)
salon_id = salon_serializer.data.get('salon_id')
salon = Salon.objects.get(pk=int(salon_id))
if user_serializer.is_valid():
user = user_serializer.save()
user.set_password(user_data['password'])
user.save()
master = Master(user=user, salon=salon)
master.save()
token, _ = Token.objects.get_or_create(user=user)
if user:
# return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response({'key': token.key, 'master_id': user.Master.pk})
return Response(user_serializer.errors, status=status.HTTP_400_BAD_REQUEST)
class Master(object):
"""The master class watches the database for changes, queues and handles amqp messages,
and handles VM image conversions"""
AMQP_BROADCAST_XCHG = "broadcast"
AMQP_SLAVE_QUEUE = "slaves"
AMQP_SLAVE_STATUS_QUEUE = "slave_status"
# -------------------------
# class methods
# -------------------------
_instance = None
@classmethod
def instance(cls, intf=None):
"""Return the singleton instance of the Master class
:returns: TODO
"""
if intf is None:
intf = MASTER_INTF
sys.stdout.flush()
if cls._instance is None:
cls._instance = cls(intf)
return cls._instance
def __init__(self, intf):
"""docstring for Master constructor"""
super(Master, self).__init__()
self._log = logging.getLogger(self.__class__.__name__)
# this will be set when the docker container is linked to talus-db
self._db_conn_info = os.environ["TALUS_DB_PORT_27017_TCP"].replace("tcp://", "")
self._running = threading.Event()
self._watcher = None
self._amqp_man = AmqpManager.instance()
self._intf = intf
# TODO need a better way than just eth0
self._ip = netifaces.ifaddresses(intf)[2][0]['addr']
# delete all the previous master entries - there should
# only be _ONE_ master document in the DB
MasterModel.objects().delete()
self._master_obj_lock = threading.Lock()
self._master_obj = MasterModel()
self._master_obj.hostname = socket.gethostname()
self._master_obj.ip = self._ip
self._master_obj.vms = []
self._master_obj.queue = []
self._master_obj.save()
self._log.info("ready")
# -------------------------
# public functions
# -------------------------
def run(self):
"""Run the master daemon
:returns: TODO
"""
self._log.info("running")
self._running.set()
self._start_watcher()
self._log.info("started watcher")
self._amqp_man.do_start()
self._amqp_listen_for_slaves()
# stupid GIL
while self._watcher.is_alive():
self._watcher.join(2 ** 32)
self._shutdown_singletons()
self._log.info("done running")
def stop(self):
"""Stop the master service from running
:returns: TODO
"""
self._log.info("stopping")
self._running.clear()
self._watcher.stop()
def handle_signal(self, sig, frame):
"""TODO: Docstring for handle_signal.
:sig: TODO
:frame: TODO
:returns: TODO
"""
self.stop()
# -------------------------
# private functions
# -------------------------
def update_status(self, queues=None, vms=None):
"""Update the master document in mongodb
"""
with self._master_obj_lock:
if queues is not None:
self._master_obj.queues = queues
if vms is not None:
self._master_obj.vms = vms
self._master_obj.save()
def _amqp_listen_for_slaves(self):
"""Setup amqp queues to listen/respond to slaves
"""
self._amqp_man.declare_exchange(
self.AMQP_BROADCAST_XCHG,
"fanout"
)
self._amqp_man.declare_queue(self.AMQP_SLAVE_QUEUE,
durable=True,
auto_delete=False,
exclusive=False
)
self._amqp_man.declare_queue(self.AMQP_SLAVE_STATUS_QUEUE,
durable=True,
auto_delete=False,
exclusive=False
)
self._amqp_man.consume_queue(self.AMQP_SLAVE_STATUS_QUEUE, self._on_slave_status)
def _on_slave_status(self, channel, method, props, body):
"""Slaves will respond to commands/queries via this queue. Slaves
will also send an initial connection message via this queue
in order to get configuration details and report basic stats
"""
self._amqp_man.ack_method(method)
data = json.loads(body)
switch = dict(
new=self._handle_slave_new,
status=self._handle_slave_status,
heartbeat=self._handle_slave_heartbeat,
)
if "type" not in data or data["type"] not in switch:
self._log.warn("received slave data is in the wrong format")
else:
switch[data["type"]](data)
def _handle_slave_status(self, data):
"""Handle slave status messages"""
if "uuid" not in data:
self._log.warn("got a slave status message that does not include a uuid")
self._log.debug(data)
return
uuid = data["uuid"]
# self._log.info("got slave status update message")
slaves = Slave.objects(hostname=data.setdefault("hostname", ""), uuid=uuid)
if len(slaves) == 0:
self._log.warn("got a slave status for a slave that isn't defined yet, making a new one")
self._handle_slave_new(data)
return
else:
slave = slaves[0]
attrs = [
"running_vms",
"total_jobs_run",
"vms",
"used_cpus",
"used_ram",
"max_cpus",
"max_ram",
]
for attr in attrs:
if attr in data:
setattr(slave, attr, data[attr])
slave.timestamps["modified"] = time.time()
slave.save()
def _handle_slave_new(self, data):
"""Handle new slave messages"""
self._log.info("handling new slave message: {}".format(data))
# these must be unique
Slave.objects(
# ip=data["ip"],
hostname=data.setdefault("hostname", "")
).delete()
slave = Slave()
slave.max_cpus = data.setdefault("max_cpus", 1)
slave.max_ram = data.setdefault("max_ram", 1)
slave.max_vms = data.setdefault("max_vms", 1)
slave.ip = data.setdefault("ip", "")
slave.hostname = data.setdefault("hostname", "")
slave.uuid = data["uuid"]
slave.timestamps["created"] = datetime.datetime.utcnow()
slave.save()
self._amqp_man.queue_msg(
json.dumps(dict(
type="config",
db=self._ip,
image_url="http://{}/images/".format(self._ip),
code=dict(
loc="http://{}/code_cache".format(self._ip),
# TODO put these in a config file
username="******",
password="******"
),
)),
self.AMQP_SLAVE_QUEUE + "_" + slave.uuid
)
def _handle_slave_heartbeat(self, data):
"""Handle slave heartbeats"""
self._log.info("handling slave heartbeat: {}".format(data))
def _shutdown_singletons(self):
self._log.info("shutting down singletons")
AmqpManager.instance().stop()
def _start_watcher(self):
"""Create and start the DB watcher
:returns: TODO
"""
self._watcher = TalusDBWatcher(
parent_log=self._log,
connection=pymongo.MongoClient(self._db_conn_info.split(":")[0], 27017)
)
self._watcher.start()
# import all of the DB watchers defined in master/watchers/
for filename in glob.glob(os.path.join(os.path.dirname(__file__), "watchers", "*.py")):
if os.path.basename(filename) == "__init__.py":
continue
mod_name = os.path.basename(filename).replace(".py", "")
mod_base = __import__("master.watchers", globals(), locals(), fromlist=[mod_name])
mod = getattr(mod_base, mod_name)
for name in dir(mod):
item = getattr(mod, name)
if type(item) is not type:
continue
if item != master.watchers.WatcherBase and issubclass(item, master.watchers.WatcherBase):
watcher = getattr(mod, name)(self._log)
self._watcher.add_watcher(watcher.collection, watcher)
class Master(object):
"""The master class watches the database for changes, queues and handles amqp messages,
and handles VM image conversions"""
AMQP_BROADCAST_XCHG = "broadcast"
AMQP_SLAVE_QUEUE = "slaves"
AMQP_SLAVE_STATUS_QUEUE = "slave_status"
# -------------------------
# class methods
# -------------------------
_instance = None
@classmethod
def instance(cls, intf=None):
"""Return the singleton instance of the Master class
:returns: TODO
"""
if cls._instance is None:
cls._instance = cls(intf)
return cls._instance
def __init__(self, intf):
"""docstring for Master constructor"""
super(Master, self).__init__()
self._log = logging.getLogger(self.__class__.__name__)
# this will be set when the docker container is linked to talus_db
self._db_conn_info = os.environ["TALUS_DB_PORT_27017_TCP"].replace("tcp://", "")
self._running = threading.Event()
self._watcher = None
self._amqp_man = AmqpManager.instance()
self._intf = intf
# TODO need a better way than just eth0
self._ip = netifaces.ifaddresses(intf)[2][0]['addr']
# delete all the previous master entries - there should
# only be _ONE_ master document in the DB
MasterModel.objects().delete()
self._master_obj_lock = threading.Lock()
self._master_obj = MasterModel()
self._master_obj.hostname = socket.gethostname()
self._master_obj.ip = self._ip
self._master_obj.vms = []
self._master_obj.queue = []
self._master_obj.save()
self._log.info("ready")
# -------------------------
# public functions
# -------------------------
def run(self):
"""Run the master daemon
:returns: TODO
"""
self._log.info("running")
self._running.set()
self._start_watcher()
self._log.info("started watcher")
self._amqp_man.do_start()
self._amqp_listen_for_slaves()
# stupid GIL
while self._watcher.is_alive():
self._watcher.join(2**32)
self._shutdown_singletons()
self._log.info("done running")
def stop(self):
"""Stop the master service from running
:returns: TODO
"""
self._log.info("stopping")
self._running.clear()
self._watcher.stop()
def handle_signal(self, sig, frame):
"""TODO: Docstring for handle_signal.
:sig: TODO
:frame: TODO
:returns: TODO
"""
self.stop()
# -------------------------
# private functions
# -------------------------
def update_status(self, queues=None, vms=None):
"""Update the master document in mongodb
"""
with self._master_obj_lock:
if queues is not None:
self._master_obj.queues = queues
if vms is not None:
self._master_obj.vms = vms
self._master_obj.save()
def _amqp_listen_for_slaves(self):
"""Setup amqp queues to listen/respond to slaves
"""
self._amqp_man.declare_exchange(
self.AMQP_BROADCAST_XCHG,
"fanout"
)
self._amqp_man.declare_queue(self.AMQP_SLAVE_QUEUE,
durable = True,
auto_delete = False,
exclusive = False
)
self._amqp_man.declare_queue(self.AMQP_SLAVE_STATUS_QUEUE,
durable = True,
auto_delete = False,
exclusive = False
)
self._amqp_man.consume_queue(self.AMQP_SLAVE_STATUS_QUEUE, self._on_slave_status)
def _on_slave_status(self, channel, method, props, body):
"""Slaves will respond to commands/queries via this queue. Slaves
will also send an initial connection message via this queue
in order to get configuration details and report basic stats
"""
self._amqp_man.ack_method(method)
data = json.loads(body)
switch = dict(
new = self._handle_slave_new,
status = self._handle_slave_status,
heartbeat = self._handle_slave_heartbeat,
)
if "type" not in data or data["type"] not in switch:
self._log.warn("recieved slave data is in the wrong format")
else:
switch[data["type"]](data)
def _handle_slave_status(self, data):
"""Handle slave status messages"""
if "uuid" not in data:
self._log.warn("got a slave status message that does not include a uuid")
self._log.debug(data)
return
uuid = data["uuid"]
#self._log.info("got slave status update message")
slaves = Slave.objects(uuid=uuid)
if len(slaves) == 0:
self._log.warn("got a slave status for a slave that isn't defined yet, making a new one")
return
slave = slaves[0]
if "running_vms" in data:
slave.running_vms = data["running_vms"]
if "total_jobs_run" in data:
slave.total_jobs_run = data["total_jobs_run"]
if "vms" in data:
slave.vms = data["vms"]
slave.timestamps["modified"] = time.time()
slave.save()
def _handle_slave_new(self, data):
"""Handle new slave messages"""
self._log.info("handling new slave message: {}".format(data))
# these must be unique
Slave.objects(
#ip=data["ip"],
hostname=data["hostname"]
).delete()
slave = Slave()
slave.max_vms = data["max_vms"]
slave.ip = data["ip"]
slave.hostname = data["hostname"]
slave.uuid = data["uuid"]
slave.timestamps["created"] = datetime.datetime.utcnow()
slave.save()
self._amqp_man.queue_msg(
json.dumps(dict(
type = "config",
db = self._ip,
code = dict(
loc = "http://{}/code_cache".format(self._ip),
# TODO put these in a config file
username = "******",
password = "******"
),
image_url = "http://{}/images/".format(self._ip)
)),
self.AMQP_SLAVE_QUEUE + "_" + slave.uuid
)
def _handle_slave_heartbeat(self, data):
"""Handle slave heartbeats"""
self._log.info("handling slave heartbeat: {}".format(data))
def _shutdown_singletons(self):
self._log.info("shutting down singletons")
AmqpManager.instance().stop()
def _start_watcher(self):
"""Create and start the DB watcher
:returns: TODO
"""
self._watcher = TalusDBWatcher(
parent_log = self._log,
connection = pymongo.MongoClient(self._db_conn_info.split(":")[0], 27017)
)
self._watcher.start()
# import all of the DB watchers defined in master/watchers/
for filename in glob.glob(os.path.join(os.path.dirname(__file__), "watchers", "*.py")):
if os.path.basename(filename) == "__init__.py":
continue
mod_name = os.path.basename(filename).replace(".py", "")
mod_base = __import__("master.watchers", globals(), locals(), fromlist=[mod_name])
mod = getattr(mod_base, mod_name)
for name in dir(mod):
item = getattr(mod, name)
if type(item) is not type:
continue
if item != master.watchers.WatcherBase and issubclass(item, master.watchers.WatcherBase):
watcher = getattr(mod, name)(self._log)
self._watcher.add_watcher(watcher.collection, watcher)