vm: add remote operation classes

circle/vm/operations.py

@@ -42,13 +42,28 @@ from .models import (
     Instance, InstanceActivity, InstanceTemplate, Interface, Node,
     Instance, InstanceActivity, InstanceTemplate, Interface, Node,
     NodeActivity, pwgen
     NodeActivity, pwgen
 )
 )
-from .tasks import agent_tasks, local_agent_tasks
+from .tasks import agent_tasks, local_agent_tasks, vm_tasks
 
 
 from dashboard.store_api import Store, NoStoreException
 from dashboard.store_api import Store, NoStoreException
 
 
 logger = getLogger(__name__)
 logger = getLogger(__name__)
 
 
 
 
+class RemoteOperationMixin(object):
+
+    remote_timeout = 30
+
+    def _operation(self, **kwargs):
+        args = self._get_remote_args(**kwargs),
+        return self.task.apply_async(
+            args=args, queue=self._get_remote_queue()
+        ).get(timeout=self.remote_timeout)
+
+    def check_precond(self):
+        super(RemoteOperationMixin, self).check_precond()
+        self._get_remote_queue()
+
+
 class InstanceOperation(Operation):
 class InstanceOperation(Operation):
     acl_level = 'owner'
     acl_level = 'owner'
     async_operation = abortable_async_instance_operation
     async_operation = abortable_async_instance_operation
@@ -116,6 +131,18 @@ class InstanceOperation(Operation):
         return False
         return False
 
 
 
 
+class RemoteInstanceOperation(RemoteOperationMixin, InstanceOperation):
+
+    remote_queue = ('vm', 'fast')
+    # activity_code_suffix = property(lambda self: self.id or self.task.name)
+
+    def _get_remote_queue(self):
+        return self.instance.get_remote_queue_name(*self.remote_queue)
+
+    def _get_remote_args(self, **kwargs):
+        return [self.instance.vm_name]
+
+
 @register_operation
 @register_operation
 class AddInterfaceOperation(InstanceOperation):
 class AddInterfaceOperation(InstanceOperation):
     activity_code_suffix = 'add_interface'
     activity_code_suffix = 'add_interface'