185 lines
8.1 KiB
Python
185 lines
8.1 KiB
Python
# Copyright 2011 OpenStack Foundation
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
from oslo import messaging
|
|
from trove.common.context import TroveContext
|
|
|
|
from oslo.utils import importutils
|
|
|
|
from trove.backup.models import Backup
|
|
import trove.common.cfg as cfg
|
|
import trove.common.rpc.version as rpc_version
|
|
from trove.common import exception
|
|
from trove.common.strategies.cluster import strategy
|
|
import trove.extensions.mgmt.instances.models as mgmtmodels
|
|
from trove.openstack.common import log as logging
|
|
from trove.openstack.common import periodic_task
|
|
from trove.taskmanager import models
|
|
from trove.taskmanager.models import FreshInstanceTasks
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class Manager(periodic_task.PeriodicTasks):
|
|
|
|
target = messaging.Target(version=rpc_version.RPC_API_VERSION)
|
|
|
|
def __init__(self):
|
|
super(Manager, self).__init__()
|
|
self.admin_context = TroveContext(
|
|
user=CONF.nova_proxy_admin_user,
|
|
auth_token=CONF.nova_proxy_admin_pass,
|
|
tenant=CONF.nova_proxy_admin_tenant_id)
|
|
if CONF.exists_notification_transformer:
|
|
self.exists_transformer = importutils.import_object(
|
|
CONF.exists_notification_transformer,
|
|
context=self.admin_context)
|
|
|
|
def resize_volume(self, context, instance_id, new_size):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.resize_volume(new_size)
|
|
|
|
def resize_flavor(self, context, instance_id, old_flavor, new_flavor):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.resize_flavor(old_flavor, new_flavor)
|
|
|
|
def reboot(self, context, instance_id):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.reboot()
|
|
|
|
def restart(self, context, instance_id):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.restart()
|
|
|
|
def detach_replica(self, context, instance_id):
|
|
slave = models.BuiltInstanceTasks.load(context, instance_id)
|
|
master_id = slave.slave_of_id
|
|
master = models.BuiltInstanceTasks.load(context, master_id)
|
|
slave.detach_replica(master)
|
|
|
|
def migrate(self, context, instance_id, host):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.migrate(host)
|
|
|
|
def delete_instance(self, context, instance_id):
|
|
try:
|
|
instance_tasks = models.BuiltInstanceTasks.load(context,
|
|
instance_id)
|
|
instance_tasks.delete_async()
|
|
except exception.UnprocessableEntity:
|
|
instance_tasks = models.FreshInstanceTasks.load(context,
|
|
instance_id)
|
|
instance_tasks.delete_async()
|
|
|
|
def delete_backup(self, context, backup_id):
|
|
models.BackupTasks.delete_backup(context, backup_id)
|
|
|
|
def create_backup(self, context, backup_info, instance_id):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.create_backup(backup_info)
|
|
|
|
def _create_replication_slave(self, context, instance_id, name, flavor,
|
|
image_id, databases, users,
|
|
datastore_manager, packages, volume_size,
|
|
availability_zone,
|
|
root_password, nics, overrides, slave_of_id):
|
|
|
|
instance_tasks = FreshInstanceTasks.load(context, instance_id)
|
|
|
|
snapshot = instance_tasks.get_replication_master_snapshot(context,
|
|
slave_of_id)
|
|
try:
|
|
instance_tasks.create_instance(flavor, image_id, databases, users,
|
|
datastore_manager, packages,
|
|
volume_size,
|
|
snapshot['dataset']['snapshot_id'],
|
|
availability_zone, root_password,
|
|
nics, overrides, None)
|
|
finally:
|
|
Backup.delete(context, snapshot['dataset']['snapshot_id'])
|
|
|
|
instance_tasks.attach_replication_slave(snapshot, flavor)
|
|
|
|
def create_instance(self, context, instance_id, name, flavor,
|
|
image_id, databases, users, datastore_manager,
|
|
packages, volume_size, backup_id, availability_zone,
|
|
root_password, nics, overrides, slave_of_id,
|
|
cluster_config):
|
|
if slave_of_id:
|
|
self._create_replication_slave(context, instance_id, name,
|
|
flavor, image_id, databases, users,
|
|
datastore_manager, packages,
|
|
volume_size,
|
|
availability_zone, root_password,
|
|
nics, overrides, slave_of_id)
|
|
else:
|
|
instance_tasks = FreshInstanceTasks.load(context, instance_id)
|
|
instance_tasks.create_instance(flavor, image_id, databases, users,
|
|
datastore_manager, packages,
|
|
volume_size, backup_id,
|
|
availability_zone, root_password,
|
|
nics, overrides, cluster_config)
|
|
|
|
def update_overrides(self, context, instance_id, overrides):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.update_overrides(overrides)
|
|
|
|
def unassign_configuration(self, context, instance_id, flavor,
|
|
configuration_id):
|
|
instance_tasks = models.BuiltInstanceTasks.load(context, instance_id)
|
|
instance_tasks.unassign_configuration(flavor, configuration_id)
|
|
|
|
def create_cluster(self, context, cluster_id):
|
|
cluster_tasks = models.load_cluster_tasks(context, cluster_id)
|
|
cluster_tasks.create_cluster(context, cluster_id)
|
|
|
|
def delete_cluster(self, context, cluster_id):
|
|
cluster_tasks = models.load_cluster_tasks(context, cluster_id)
|
|
cluster_tasks.delete_cluster(context, cluster_id)
|
|
|
|
if CONF.exists_notification_transformer:
|
|
@periodic_task.periodic_task(
|
|
ticks_between_runs=CONF.exists_notification_ticks)
|
|
def publish_exists_event(self, context):
|
|
"""
|
|
Push this in Instance Tasks to fetch a report/collection
|
|
:param context: currently None as specied in bin script
|
|
"""
|
|
mgmtmodels.publish_exist_events(self.exists_transformer,
|
|
self.admin_context)
|
|
|
|
def __getattr__(self, name):
|
|
"""
|
|
We should only get here if Python couldn't find a "real" method.
|
|
"""
|
|
|
|
def raise_error(msg):
|
|
raise AttributeError(msg)
|
|
|
|
manager, sep, method = name.partition('_')
|
|
if not manager:
|
|
raise_error('Cannot derive manager from attribute name "%s"' %
|
|
name)
|
|
|
|
task_strategy = strategy.load_taskmanager_strategy(manager)
|
|
if not task_strategy:
|
|
raise_error('No task manager strategy for manager "%s"' % manager)
|
|
|
|
if method not in task_strategy.task_manager_manager_actions:
|
|
raise_error('No method "%s" for task manager strategy for manager'
|
|
' "%s"' % (method, manager))
|
|
|
|
return task_strategy.task_manager_manager_actions.get(method)
|