2626import  com .google .bigtable .v2 .AuthorizedViewName ;
2727import  com .google .bigtable .v2 .CheckAndMutateRowRequest ;
2828import  com .google .bigtable .v2 .GenerateInitialChangeStreamPartitionsRequest ;
29+ import  com .google .bigtable .v2 .MaterializedViewName ;
2930import  com .google .bigtable .v2 .MutateRowRequest ;
3031import  com .google .bigtable .v2 .MutateRowsRequest ;
3132import  com .google .bigtable .v2 .ReadChangeStreamRequest ;
@@ -121,9 +122,11 @@ static TagValue extractStatusFromFuture(Future<?> future) {
121122  static  String  extractTableId (Object  request ) {
122123    String  tableName  = null ;
123124    String  authorizedViewName  = null ;
125+     String  materializedViewName  = null ;
124126    if  (request  instanceof  ReadRowsRequest ) {
125127      tableName  = ((ReadRowsRequest ) request ).getTableName ();
126128      authorizedViewName  = ((ReadRowsRequest ) request ).getAuthorizedViewName ();
129+       materializedViewName  = ((ReadRowsRequest ) request ).getMaterializedViewName ();
127130    } else  if  (request  instanceof  MutateRowsRequest ) {
128131      tableName  = ((MutateRowsRequest ) request ).getTableName ();
129132      authorizedViewName  = ((MutateRowsRequest ) request ).getAuthorizedViewName ();
@@ -133,6 +136,7 @@ static String extractTableId(Object request) {
133136    } else  if  (request  instanceof  SampleRowKeysRequest ) {
134137      tableName  = ((SampleRowKeysRequest ) request ).getTableName ();
135138      authorizedViewName  = ((SampleRowKeysRequest ) request ).getAuthorizedViewName ();
139+       materializedViewName  = ((SampleRowKeysRequest ) request ).getMaterializedViewName ();
136140    } else  if  (request  instanceof  CheckAndMutateRowRequest ) {
137141      tableName  = ((CheckAndMutateRowRequest ) request ).getTableName ();
138142      authorizedViewName  = ((CheckAndMutateRowRequest ) request ).getAuthorizedViewName ();
@@ -150,6 +154,9 @@ static String extractTableId(Object request) {
150154    if  (authorizedViewName  != null  && !authorizedViewName .isEmpty ()) {
151155      return  AuthorizedViewName .parse (authorizedViewName ).getTable ();
152156    }
157+     if  (materializedViewName  != null  && !materializedViewName .isEmpty ()) {
158+       return  MaterializedViewName .parse (materializedViewName ).getMaterializedView ();
159+     }
153160    return  "<unspecified>" ;
154161  }
155162
0 commit comments