def __init__(self, unit=None, group_rules=None): state_path = os.path.join(os.environ.get("CHARM_DIR", ""), "rds.state") self._state = KVFile(state_path) self.unit = unit or Unit() self._group_rules = group_rules or ()
class Controller(object): def __init__(self, unit=None, group_rules=None): state_path = os.path.join(os.environ.get("CHARM_DIR", ""), "rds.state") self._state = KVFile(state_path) self.unit = unit or Unit() self._group_rules = group_rules or () def get_region(self): return self.unit.ec2metadata['availability-zone'][:-1] def get_rds(self, config): return rds.connect_to_region(self.get_region()) def get_ec2(self, config): return rds.connect_to_region(self.get_region()) def get_db(self, config, instance): if 'mysql' in config['engine'].lower(): db = MySQL(config) if not self._state.get('mysql.client'): db.install_driver() self._state.set('mysql.client', True) return db def get_svc_dbname(self): return self.get_db_identifier()[:16].replace('-', '_') def get_db_identifier(self): service = os.environ["JUJU_REMOTE_UNIT"].split("/")[0] rel_id = os.environ["JUJU_RELATION_ID"].split(":")[-1] return "-".join([ service, rel_id, os.environ["JUJU_ENV_UUID"]]) def wait_for_db_instance(self, rds, instance_id): # Don't wait more than 10m print "test: waiting for db instance", instance_id, t = time.time() seen = set() while True: time.sleep(3) sys.stdout.write(".") try: dbs = rds.get_all_dbinstances(instance_id) except boto.exception.EC2ResponseError, e: if e.error_code == "InvalidInstanceId.NotFound": continue db = dbs.pop() if db.status == "available": print "\ntest: db %s available in %s" % ( instance_id, time.time() - t) return db elif db.status not in seen: print "\n db status", db.status, (time.time() - t), seen.add(db.status) if time.time() - t > 15 * 60: raise RuntimeError( "Database not provisioned within threshold %s" % ( instance_id))