forked from uncloud/uncloud
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import json
|
|
from os.path import join
|
|
from uuid import uuid4
|
|
|
|
from .etcd_wrapper import PsuedoEtcdEntry
|
|
from .classes import SpecificEtcdEntryBase
|
|
|
|
|
|
class RequestType:
|
|
CreateVM = "CreateVM"
|
|
ScheduleVM = "ScheduleVM"
|
|
StartVM = "StartVM"
|
|
StopVM = "StopVM"
|
|
InitVMMigration = "InitVMMigration"
|
|
TransferVM = "TransferVM"
|
|
DeleteVM = "DeleteVM"
|
|
|
|
|
|
class RequestEntry(SpecificEtcdEntryBase):
|
|
|
|
def __init__(self, e):
|
|
self.destination_sock_path = None
|
|
self.destination_host_key = None
|
|
self.type = None # type: str
|
|
self.migration = None # type: bool
|
|
self.destination = None # type: str
|
|
self.uuid = None # type: str
|
|
self.hostname = None # type: str
|
|
super().__init__(e)
|
|
|
|
@classmethod
|
|
def from_scratch(cls, request_prefix, **kwargs):
|
|
e = PsuedoEtcdEntry(join(request_prefix, uuid4().hex),
|
|
value=json.dumps(kwargs).encode("utf-8"), value_in_json=True)
|
|
return cls(e)
|
|
|
|
|
|
class RequestPool:
|
|
def __init__(self, etcd_client, request_prefix):
|
|
self.client = etcd_client
|
|
self.prefix = request_prefix
|
|
|
|
def put(self, obj: RequestEntry):
|
|
if not obj.key.startswith(self.prefix):
|
|
obj.key = join(self.prefix, obj.key)
|
|
|
|
self.client.put(obj.key, obj.value, value_in_json=True)
|