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