Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(Bigquery): Undelete table sample. #1774

Merged
merged 10 commits into from
Feb 20, 2023
77 changes: 77 additions & 0 deletions bigquery/api/src/undelete_table.php
Original file line number Diff line number Diff line change
@@ -0,0 +1,77 @@
<?php
/**
* Copyright 2023 Google LLC.
*
* 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.
*/

/**
* For instructions on how to run the full sample:
*
* @see https://github.com/GoogleCloudPlatform/php-docs-samples/tree/main/bigquery/api/README.md
*/

namespace Google\Cloud\Samples\BigQuery;

# [START bigquery_undelete_table]
use Google\Cloud\BigQuery\BigQueryClient;

/**
* Restore a deleted table from its snapshot.
*
* @param string $projectId The project Id of your Google Cloud Project.
* @param string $datasetId The BigQuery dataset ID.
* @param string $tableId Table ID of the table to delete.
* @param string $restoredTableId Table Id for the restored table.
*/
function undelete_table(
string $projectId,
string $datasetId,
string $tableId,
string $restoredTableId
): void {
$bigQuery = new BigQueryClient(['projectId' => $projectId]);
$dataset = $bigQuery->dataset($datasetId);

// Choose an appropriate snapshot point as epoch milliseconds.
// For this example, we choose the current time as we're about to delete the
// table immediately afterwards
$snapshotEpoch = date_create()->format('Uv');

// Delete the table.
$dataset->table($tableId)->delete();

// Construct the restore-from table ID using a snapshot decorator.
$snapshotId = "{$tableId}@{$snapshotEpoch}";

// Restore the deleted table
$restoredTable = $dataset->table($restoredTableId);
$copyConfig = $dataset->table($snapshotId)->copy($restoredTable);
$job = $bigQuery->runJob($copyConfig);

// check if the job is complete
$job->reload();
if (!$job->isComplete()) {
throw new \Exception('Job has not yet completed', 500);
}
// check if the job has errors
if (isset($job->info()['status']['errorResult'])) {
$error = $job->info()['status']['errorResult']['message'];
printf('Error running job: %s' . PHP_EOL, $error);
} else {
print('Snapshot restored successfully' . PHP_EOL);
}
}
# [END bigquery_undelete_table]
require_once __DIR__ . '/../../../testing/sample_helpers.php';
\Google\Cloud\Samples\execute_sample(__FILE__, __NAMESPACE__, $argv);
34 changes: 26 additions & 8 deletions bigquery/api/test/bigqueryTest.php
Original file line number Diff line number Diff line change
Expand Up @@ -99,8 +99,8 @@ public function testCreateAndDeleteTable()
{
$tempTableId = sprintf('test_table_%s', time());
$fields = json_encode([
['name' => 'name', 'type' => 'string', 'mode' => 'nullable'],
['name' => 'title', 'type' => 'string', 'mode' => 'nullable']
['name' => 'name', 'type' => 'string', 'mode' => 'nullable'],
['name' => 'title', 'type' => 'string', 'mode' => 'nullable']
]);
$output = $this->runFunctionSnippet('create_table', [
self::$datasetId,
Expand Down Expand Up @@ -352,8 +352,8 @@ public function testAddColumnLoadAppend()
{
$tableId = $this->createTempTable();
$output = $this->runFunctionSnippet('add_column_load_append', [
self::$datasetId,
$tableId
self::$datasetId,
$tableId
]);

$this->assertStringContainsString('name', $output);
Expand All @@ -365,14 +365,32 @@ public function testAddColumnQueryAppend()
{
$tableId = $this->createTempTable();
$output = $this->runFunctionSnippet('add_column_query_append', [
self::$datasetId,
$tableId
self::$datasetId,
$tableId
]);
$this->assertStringContainsString('name', $output);
$this->assertStringContainsString('title', $output);
$this->assertStringContainsString('description', $output);
}

public function testUndeleteTable()
{
// Create a base table
$sourceTableId = $this->createTempTable();

// run the sample
$restoredTableId = uniqid('restored_');
$output = $this->runFunctionSnippet('undelete_table', [
self::$datasetId,
$sourceTableId,
$restoredTableId,
]);

$restoredTable = self::$dataset->table($restoredTableId);
$this->assertStringContainsString('Snapshot restored successfully', $output);
$this->verifyTable($restoredTable, 'Brent Shaffer', 3);
}

private function runFunctionSnippet($sampleName, $params = [])
{
array_unshift($params, self::$projectId);
Expand All @@ -386,8 +404,8 @@ private function createTempEmptyTable()
{
$tempTableId = sprintf('test_table_%s_%s', time(), rand());
$fields = json_encode([
['name' => 'name', 'type' => 'string', 'mode' => 'nullable'],
['name' => 'title', 'type' => 'string', 'mode' => 'nullable']
['name' => 'name', 'type' => 'string', 'mode' => 'nullable'],
['name' => 'title', 'type' => 'string', 'mode' => 'nullable']
]);
$this->runFunctionSnippet('create_table', [
self::$datasetId,
Expand Down