new parameter $make_db_name_unique to append a decrementing numeric postfix to db_name and db_user, if db already exists

This commit is contained in:
Ralf Becker 2010-07-31 09:23:41 +00:00
parent 86a32e2f8b
commit 5d6e6cdc9c

View File

@ -20,6 +20,13 @@ class setup_cmd_database extends setup_cmd
*/ */
const SETUP_CLI_CALLABLE = true; const SETUP_CLI_CALLABLE = true;
/**
* Maximum length of database name (at least for MySQL this is the limit)
*
* @var int
*/
const MAX_DB_NAME_LEN = 16;
/** /**
* Instance of egw_db to connect or create the db * Instance of egw_db to connect or create the db
* *
@ -41,9 +48,11 @@ class setup_cmd_database extends setup_cmd
* @param string $db_root_pw=null * @param string $db_root_pw=null
* @param string $sub_command='create_db' 'create_db', 'test_db', 'test_db_root' * @param string $sub_command='create_db' 'create_db', 'test_db', 'test_db_root'
* @param string $db_grant_host='localhost' host/ip of webserver for grant * @param string $db_grant_host='localhost' host/ip of webserver for grant
* @param boolean $make_db_name_unique=false true: if create fails because db exists,
* try creating a unique name by shortening the name and adding a number to it
*/ */
function __construct($domain,$db_type=null,$db_host=null,$db_port=null,$db_name=null,$db_user=null,$db_pass=null, function __construct($domain,$db_type=null,$db_host=null,$db_port=null,$db_name=null,$db_user=null,$db_pass=null,
$db_root=null,$db_root_pw=null,$sub_command='create_db',$db_grant_host='localhost') $db_root=null,$db_root_pw=null,$sub_command='create_db',$db_grant_host='localhost',$make_db_name_unique=false)
{ {
if (!is_array($domain)) if (!is_array($domain))
{ {
@ -59,9 +68,10 @@ class setup_cmd_database extends setup_cmd
'db_root_pw' => $db_root_pw, 'db_root_pw' => $db_root_pw,
'sub_command' => $sub_command, 'sub_command' => $sub_command,
'db_grant_host' => $db_grant_host, 'db_grant_host' => $db_grant_host,
'make_db_name_unique' => $make_db_name_unique,
); );
} }
//echo __CLASS__.'::__construct()'; _debug_array($domain); //error_log(__METHOD__.'('.array2string($domain).") make_db_name_unique=".array2string($domain['make_db_name_unique']));
admin_cmd::__construct($domain); admin_cmd::__construct($domain);
} }
@ -154,11 +164,16 @@ class setup_cmd_database extends setup_cmd
* *
* The check will fail if the database exists, but already contains tables * The check will fail if the database exists, but already contains tables
* *
* if $this->make_db_name_unique is set, a decrementing nummeric prefix gets
* added to $this->db_name AND $this->db_user, if db already exists.
*
* @return string with success message * @return string with success message
* @throws egw_exception_wrong_userinput * @throws egw_exception_wrong_userinput
*/ */
private function create() private function create()
{ {
static $try_make_unique = 0; // to limit trials to create a unique name
try { try {
$msg = $this->connect(); $msg = $this->connect();
} }
@ -168,9 +183,36 @@ class setup_cmd_database extends setup_cmd
$this->test_db->create_database($this->db_root,$this->db_root_pw,$this->db_charset,$this->db_grant_host); $this->test_db->create_database($this->db_root,$this->db_root_pw,$this->db_charset,$this->db_grant_host);
$this->connect(); $this->connect();
} }
catch(egw_exception_wrong_userinput $e) { catch(egw_exception_db $e) { // catches failed to create database
// try connect as root to check if that's the problem // try connect as root to check if wrong root/root_pw is the problem
$this->connect($this->db_root,$this->db_root_pw,$this->db_meta); $this->connect($this->db_root,$this->db_root_pw,$this->db_meta);
// if we should create a db with a unique name (try it only N times, not endless!)
if ($this->make_db_name_unique && $try_make_unique++ < 20)
{
// check if we can connect as root to the db to create --> db exists already
try {
$this->connect($this->db_root,$this->db_root_pw);
// create new db_name by incrementing an existing numeric postfix
if (preg_match('/([0-9]+)$/',$this->db_name,$matches))
{
$num = (string)(++$matches[1]);
}
else // or adding one starting with 2
{
$num = '2';
}
$this->set_defaults['db_name'] = $this->db_name =
$this->set_defaults['db_user'] = $this->db_user = // change user too (otherwise existing user/db could not connect any more!)
substr($this->db_name,0,self::MAX_DB_NAME_LEN-strlen($num)).$num;
return $this->create();
}
catch (egw_exception_wrong_userinput $e2)
{
// we can NOT connect to db as root --> ignore exception to give general error
}
}
// if not give general error // if not give general error
throw new egw_exception_wrong_userinput(lang('Can not create %1 database %2 on %3 for user %4!', throw new egw_exception_wrong_userinput(lang('Can not create %1 database %2 on %3 for user %4!',
$this->db_type,$this->db_name,$this->db_host.($this->db_port?':'.$this->db_port:''),$this->db_user)); $this->db_type,$this->db_name,$this->db_host.($this->db_port?':'.$this->db_port:''),$this->db_user));
@ -263,7 +305,9 @@ class setup_cmd_database extends setup_cmd
if (strpos($this->$name,'$domain') !== false) if (strpos($this->$name,'$domain') !== false)
{ {
// limit names to 16 chars (16 char is user-name limit in MySQL) // limit names to 16 chars (16 char is user-name limit in MySQL)
$this->set_defaults[$name] = $this->$name = substr(str_replace(array('$domain','.','-'),array($this->domain,'_','_'),$this->$name),0,16); $this->set_defaults[$name] = $this->$name =
substr(str_replace(array('$domain','.','-'),array($this->domain,'_','_'),$this->$name),
0,self::MAX_DB_NAME_LEN);
} }
} }
} }