144 lines
4.2 KiB
PHP
144 lines
4.2 KiB
PHP
<?php
|
|
|
|
use Appwrite\Extend\PDO;
|
|
use Redis;
|
|
use Utopia\App;
|
|
use Utopia\Cache\Cache;
|
|
use Utopia\Cache\Adapter\Redis as RedisCache;
|
|
use Utopia\CLI\Console;
|
|
use Utopia\Database\Database;
|
|
use Utopia\Database\Adapter\MariaDB;
|
|
|
|
/** @var Utopia\Registry\Registry $register */
|
|
|
|
require_once __DIR__.'/init.php';
|
|
|
|
$register->set('db', function () {
|
|
$dbHost = App::getEnv('_APP_DB_HOST', '');
|
|
$dbUser = App::getEnv('_APP_DB_USER', '');
|
|
$dbPass = App::getEnv('_APP_DB_PASS', '');
|
|
$dbScheme = App::getEnv('_APP_DB_SCHEMA', '');
|
|
|
|
$pdo = new PDO("mysql:host={$dbHost};dbname={$dbScheme};charset=utf8mb4", $dbUser, $dbPass, array(
|
|
PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES utf8mb4',
|
|
PDO::ATTR_TIMEOUT => 3, // Seconds
|
|
PDO::ATTR_PERSISTENT => true,
|
|
PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
|
|
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
|
));
|
|
|
|
return $pdo;
|
|
});
|
|
|
|
$register->set('cache', function () { // Register cache connection
|
|
$redis = new Redis();
|
|
$redis->pconnect(App::getEnv('_APP_REDIS_HOST', ''), App::getEnv('_APP_REDIS_PORT', ''));
|
|
$redis->setOption(Redis::OPT_READ_TIMEOUT, -1);
|
|
|
|
return $redis;
|
|
});
|
|
|
|
/**
|
|
* Get internal project database
|
|
* @param string $projectId
|
|
* @return Database
|
|
*/
|
|
function getInternalDB(string $projectId): Database
|
|
{
|
|
global $register;
|
|
|
|
$attempts = 0;
|
|
$max = 10;
|
|
$sleep = 2;
|
|
|
|
do {
|
|
try {
|
|
$attempts++;
|
|
$cache = new Cache(new RedisCache($register->get('cache')));
|
|
$dbForInternal = new Database(new MariaDB($register->get('db')), $cache);
|
|
$dbForInternal->setNamespace("project_{$projectId}_internal"); // Main DB
|
|
if (!$dbForInternal->exists()) {
|
|
throw new Exception("Table does not exist: {$dbForInternal->getNamespace()}");
|
|
}
|
|
break; // leave loop if successful
|
|
} catch(\Exception $e) {
|
|
Console::warning("Database not ready. Retrying connection ({$attempts})...");
|
|
if ($attempts >= $max) {
|
|
throw new \Exception('Failed to connect to database: '. $e->getMessage());
|
|
}
|
|
sleep($sleep);
|
|
}
|
|
} while ($attempts < $max);
|
|
|
|
return $dbForInternal;
|
|
}
|
|
|
|
/**
|
|
* Get external project database
|
|
* @param string $projectId
|
|
* @return Database
|
|
*/
|
|
function getExternalDB(string $projectId): Database
|
|
{
|
|
global $register;
|
|
|
|
$attempts = 0;
|
|
$max = 10;
|
|
$sleep = 2;
|
|
|
|
do {
|
|
try {
|
|
$attempts++;
|
|
$cache = new Cache(new RedisCache($register->get('cache')));
|
|
$dbForExternal = new Database(new MariaDB($register->get('db')), $cache);
|
|
$dbForExternal->setNamespace("project_{$projectId}_external"); // Main DB
|
|
if (!$dbForExternal->exists()) {
|
|
throw new Exception("Table does not exist: {$dbForExternal->getNamespace()}");
|
|
}
|
|
break; // leave loop if successful
|
|
} catch(\Exception $e) {
|
|
Console::warning("Database not ready. Retrying connection ({$attempts})...");
|
|
if ($attempts >= $max) {
|
|
throw new \Exception('Failed to connect to database: '. $e->getMessage());
|
|
}
|
|
sleep($sleep);
|
|
}
|
|
} while ($attempts < $max);
|
|
|
|
return $dbForExternal;
|
|
}
|
|
|
|
/**
|
|
* Get console database
|
|
* @return Database
|
|
*/
|
|
function getConsoleDB(): Database
|
|
{
|
|
global $register;
|
|
|
|
$attempts = 0;
|
|
$max = 5;
|
|
$sleep = 5;
|
|
|
|
do {
|
|
try {
|
|
$attempts++;
|
|
$cache = new Cache(new RedisCache($register->get('cache')));
|
|
$dbForConsole = new Database(new MariaDB($register->get('db')), $cache);
|
|
$dbForConsole->setNamespace('project_console_internal'); // Main DB
|
|
if (!$dbForConsole->exists()) {
|
|
throw new Exception("Table does not exist: {$dbForConsole->getNamespace()}");
|
|
}
|
|
break; // leave loop if successful
|
|
} catch(\Exception $e) {
|
|
Console::warning("Database not ready. Retrying connection ({$attempts})...");
|
|
if ($attempts >= $max) {
|
|
throw new \Exception('Failed to connect to database: '. $e->getMessage());
|
|
}
|
|
sleep($sleep);
|
|
}
|
|
} while ($attempts < $max);
|
|
|
|
return $dbForConsole;
|
|
}
|