139 lines
5.2 KiB
Python
139 lines
5.2 KiB
Python
# Copyright 2014 eBay Software 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_log import log as logging
|
|
|
|
from trove.common import cfg
|
|
from trove.common.strategies.cluster import base
|
|
from trove.guestagent import api as guest_api
|
|
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
CONF = cfg.CONF
|
|
ADD_MEMBERS_TIMEOUT = CONF.mongodb.add_members_timeout
|
|
|
|
|
|
class MongoDbGuestAgentStrategy(base.BaseGuestAgentStrategy):
|
|
|
|
@property
|
|
def guest_client_class(self):
|
|
return MongoDbGuestAgentAPI
|
|
|
|
|
|
class MongoDbGuestAgentAPI(guest_api.API):
|
|
"""Cluster Specific Datastore Guest API
|
|
|
|
**** VERSION CONTROLLED API ****
|
|
|
|
The methods in this class are subject to version control as
|
|
coordinated by guestagent/api.py. Whenever a change is made to
|
|
any API method in this class, add a version number and comment
|
|
to the top of guestagent/api.py and use the version number as
|
|
appropriate in this file
|
|
"""
|
|
|
|
def add_shard(self, replica_set_name, replica_set_member):
|
|
LOG.debug("Adding shard with replSet %(replica_set_name)s and member "
|
|
"%(replica_set_member)s for instance "
|
|
"%(id)s", {'replica_set_name': replica_set_name,
|
|
'replica_set_member': replica_set_member,
|
|
'id': self.id})
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("add_shard", guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version,
|
|
replica_set_name=replica_set_name,
|
|
replica_set_member=replica_set_member)
|
|
|
|
def add_members(self, members):
|
|
LOG.debug("Adding members %(members)s on instance %(id)s", {
|
|
'members': members, 'id': self.id})
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("add_members", ADD_MEMBERS_TIMEOUT,
|
|
version=version, members=members)
|
|
|
|
def add_config_servers(self, config_servers):
|
|
LOG.debug("Adding config servers %(config_servers)s for instance "
|
|
"%(id)s", {'config_servers': config_servers,
|
|
'id': self.id})
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("add_config_servers", guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version,
|
|
config_servers=config_servers)
|
|
|
|
def cluster_complete(self):
|
|
LOG.debug("Notify regarding cluster install completion")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("cluster_complete", guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version)
|
|
|
|
def get_key(self):
|
|
LOG.debug("Requesting cluster key from guest")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("get_key", guest_api.AGENT_LOW_TIMEOUT,
|
|
version=version)
|
|
|
|
def prep_primary(self):
|
|
LOG.debug("Preparing member to be primary member.")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("prep_primary", guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version)
|
|
|
|
def create_admin_user(self, password):
|
|
LOG.debug("Creating admin user")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("create_admin_user", guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version, password=password)
|
|
|
|
def store_admin_password(self, password):
|
|
LOG.debug("Storing admin password")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("store_admin_password",
|
|
guest_api.AGENT_LOW_TIMEOUT,
|
|
version=version,
|
|
password=password)
|
|
|
|
def get_replica_set_name(self):
|
|
LOG.debug("Querying member for its replica set name")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("get_replica_set_name",
|
|
guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version)
|
|
|
|
def get_admin_password(self):
|
|
LOG.debug("Querying instance for its admin password")
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("get_admin_password",
|
|
guest_api.AGENT_LOW_TIMEOUT,
|
|
version=version)
|
|
|
|
def is_shard_active(self, replica_set_name):
|
|
LOG.debug("Checking if replica set %s is active", replica_set_name)
|
|
version = guest_api.API.API_BASE_VERSION
|
|
|
|
return self._call("is_shard_active",
|
|
guest_api.AGENT_HIGH_TIMEOUT,
|
|
version=version,
|
|
replica_set_name=replica_set_name)
|