Page MenuHomePhabricator
Files F15302659

D21745.id51839.diff
No OneTemporary
Actions

D21745.id51839.diff
View Options

diff --git a/src/infrastructure/storage/management/PhabricatorStorageManagementAPI.php b/src/infrastructure/storage/management/PhabricatorStorageManagementAPI.php
--- a/src/infrastructure/storage/management/PhabricatorStorageManagementAPI.php
+++ b/src/infrastructure/storage/management/PhabricatorStorageManagementAPI.php
@@ -89,6 +89,17 @@
return $this->namespace.'_'.$fragment;
}
+ public function getInternalDatabaseName($name) {
+ $namespace = $this->getNamespace();
+
+ $prefix = $namespace.'_';
+ if (strncmp($name, $prefix, strlen($prefix))) {
+ return null;
+ }
+
+ return substr($name, strlen($prefix));
+ }
+
public function getDisplayName() {
return $this->getRef()->getDisplayName();
}
diff --git a/src/infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php b/src/infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php
--- a/src/infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php
+++ b/src/infrastructure/storage/management/workflow/PhabricatorStorageManagementDumpWorkflow.php
@@ -44,6 +44,16 @@
'With __--output__, overwrite the output file if it already '.
'exists.'),
),
+ array(
+ 'name' => 'database',
+ 'param' => 'database-name',
+ 'help' => pht(
+ 'Dump only tables in the named database (or databases, if '.
+ 'the flag is repeated). Specify database names without the '.
+ 'namespace prefix (that is: use "differential", not '.
+ '"phabricator_differential").'),
+ 'repeat' => true,
+ ),
));
}
@@ -58,6 +68,8 @@
$is_noindex = $args->getArg('no-indexes');
$is_replica = $args->getArg('for-replica');
+ $database_filter = $args->getArg('database');
+
if ($is_compress) {
if ($output_file === null) {
throw new PhutilArgumentUsageException(
@@ -128,11 +140,58 @@
$schemata = $actual_map[$ref_key];
$expect = $expect_map[$ref_key];
+ if ($database_filter) {
+ $expect_databases = $expect->getDatabases();
+ foreach ($expect_databases as $expect_database) {
+ $database_name = $expect_database->getName();
+
+ $internal_name = $api->getInternalDatabaseName($database_name);
+ if ($internal_name !== null) {
+ $internal_names[$internal_name] = $database_name;
+ }
+ }
+
+ ksort($internal_names);
+
+ $seen = array();
+ foreach ($database_filter as $filter) {
+ if (!isset($internal_names[$filter])) {
+ throw new PhutilArgumentUsageException(
+ pht(
+ 'Database "%s" is unknown. This script can only dump '.
+ 'databases known to the current version of Phabricator. '.
+ 'Valid databases are: %s.',
+ $filter,
+ implode(', ', array_keys($internal_names))));
+ }
+
+ if (isset($seen[$filter])) {
+ throw new PhutilArgumentUsageException(
+ pht(
+ 'Database "%s" is specified more than once. Specify each '.
+ 'database at most once.',
+ $filter));
+ }
+
+ $seen[$filter] = true;
+ }
+
+ $dump_databases = array_select_keys($internal_names, $database_filter);
+ $dump_databases = array_fuse($dump_databases);
+ } else {
+ $dump_databases = array_keys($schemata->getDatabases());
+ $dump_databases = array_fuse($dump_databases);
+ }
+
$with_caches = $is_replica;
$with_indexes = !$is_noindex;
$targets = array();
foreach ($schemata->getDatabases() as $database_name => $database) {
+ if (!isset($dump_databases[$database_name])) {
+ continue;
+ }
+
$expect_database = $expect->getDatabase($database_name);
foreach ($database->getTables() as $table_name => $table) {

File Metadata

Mime Type
text/plain
Expires
Thu, Mar 6, 7:28 PM (3 d, 13 h ago)
Storage Engine
blob
Storage Format
Encrypted (AES-256-CBC)
Storage Handle
7308975
Default Alt Text
D21745.id51839.diff (3 KB)

Event Timeline

Phabricator · Built by Phacility