samples: adding copy backup operation support

Author: asthamohta

samples/snippets/src/main/java/com/example/spanner/SpannerSample.java

@@ -1661,21 +1661,17 @@ static void cancelCreateBackup(
   // [END spanner_cancel_backup_create]
 
   // [START spanner_list_backup_operations]
-  static void listBackupOperations(InstanceAdminClient instanceAdminClient, DatabaseId databaseId) {
+  static void listBackupOperations(InstanceAdminClient instanceAdminClient, DatabaseId databaseId, BackupId backupId) {
     Instance instance = instanceAdminClient.getInstance(databaseId.getInstanceId().getInstance());
     // Get create backup operations for the sample database.
-    Timestamp last24Hours = Timestamp.ofTimeSecondsAndNanos(TimeUnit.SECONDS.convert(
-        TimeUnit.HOURS.convert(Timestamp.now().getSeconds(), TimeUnit.SECONDS) - 24,
-        TimeUnit.HOURS), 0);
     String filter =
         String.format(
-            "(metadata.database:%s) AND "
-                + "(metadata.@type:type.googleapis.com/"
-                + "google.spanner.admin.database.v1.CreateBackupMetadata) AND "
-                + "(metadata.progress.start_time > \"%s\")",
-            databaseId.getName(), last24Hours);
-    Page<Operation> operations = instance.listBackupOperations(Options.filter(filter));
-    for (Operation op : operations.iterateAll()) {
+            "(metadata.@type:type.googleapis.com/google.spanner.admin.database.v1.CreateBackupMetadata) "
+                + "AND (metadata.database:%s)",
+            databaseId.getName());
+    Page<Operation> createBackupOperations = instance.listBackupOperations(Options.filter(filter));
+    System.out.println("Create Backup Operations:");
+    for (Operation op : createBackupOperations.iterateAll()) {
       try {
         CreateBackupMetadata metadata = op.getMetadata().unpack(CreateBackupMetadata.class);
         System.out.println(
@@ -1689,6 +1685,30 @@ static void listBackupOperations(InstanceAdminClient instanceAdminClient, Databa
         System.err.println(e.getMessage());
       }
     }
+
+    // Get copy backup operations for the sample database.
+    filter =
+            String.format(
+                    "(metadata.@type:type.googleapis.com/"
+                            + "google.spanner.admin.database.v1.CopyBackupMetadata) "
+                            + "AND (metadata.source_backup:%s)",
+                    backupId.getName());
+    Page<Operation>  copyBackupOperations = instance.listBackupOperations(Options.filter(filter));
+    System.out.println("Copy Backup Operations:");
+    for (Operation op : copyBackupOperations.iterateAll()) {
+      try {
+        CopyBackupMetadata copyBackupMetadata = op.getMetadata().unpack(CopyBackupMetadata.class);
+        System.out.println(
+                String.format(
+                        "Copy Backup %s on backup %s pending: %d%% complete",
+                        copyBackupMetadata.getName(),
+                        copyBackupMetadata.getSourceBackup(),
+                        copyBackupMetadata.getProgress().getProgressPercent()));
+      } catch (InvalidProtocolBufferException e) {
+        // The returned operation does not contain CopyBackupMetadata.
+        System.err.println(e.getMessage());
+      }
+    }
   }
   // [END spanner_list_backup_operations]
 
@@ -2055,7 +2075,7 @@ static void run(
             BackupId.of(backup.getInstanceId(), backup.getBackup() + "_cancel"));
         break;
       case "listbackupoperations":
-        listBackupOperations(instanceAdminClient, database);
+        listBackupOperations(instanceAdminClient, database, backup);
         break;
       case "listdatabaseoperations":
         listDatabaseOperations(instanceAdminClient, dbAdminClient, database.getInstanceId());
@@ -2151,14 +2171,14 @@ static void printUsageAndExit() {
     System.err.println("    SpannerExample querywithqueryoptions my-instance example-db");
     System.err.println("    SpannerExample createbackup my-instance example-db");
     System.err.println("    SpannerExample listbackups my-instance example-db");
-    System.err.println("    SpannerExample listbackupoperations my-instance example-db");
+    System.err.println("    SpannerExample listbackupoperations my-instance example-db backup-id");
     System.err.println("    SpannerExample listdatabaseoperations my-instance example-db");
     System.err.println("    SpannerExample restorebackup my-instance example-db");
     System.exit(1);
   }
 
   public static void main(String[] args) throws Exception {
-    if (args.length != 3) {
+    if (args.length != 3 && args.length != 4) {
       printUsageAndExit();
     }
     // [START init_client]
@@ -2182,6 +2202,9 @@ public static void main(String[] args) throws Exception {
           String.format(
               "%s_%02d",
               db.getDatabase(), LocalDate.now().get(ChronoField.ALIGNED_WEEK_OF_YEAR));
+      if( args.length == 4) {
+        backupName = args[3];
+      }
       BackupId backup = BackupId.of(db.getInstanceId(), backupName);
 
       // [START init_client]