26
26
import com .google .bigtable .v2 .AuthorizedViewName ;
27
27
import com .google .bigtable .v2 .CheckAndMutateRowRequest ;
28
28
import com .google .bigtable .v2 .GenerateInitialChangeStreamPartitionsRequest ;
29
+ import com .google .bigtable .v2 .MaterializedViewName ;
29
30
import com .google .bigtable .v2 .MutateRowRequest ;
30
31
import com .google .bigtable .v2 .MutateRowsRequest ;
31
32
import com .google .bigtable .v2 .ReadChangeStreamRequest ;
@@ -121,9 +122,11 @@ static TagValue extractStatusFromFuture(Future<?> future) {
121
122
static String extractTableId (Object request ) {
122
123
String tableName = null ;
123
124
String authorizedViewName = null ;
125
+ String materializedViewName = null ;
124
126
if (request instanceof ReadRowsRequest ) {
125
127
tableName = ((ReadRowsRequest ) request ).getTableName ();
126
128
authorizedViewName = ((ReadRowsRequest ) request ).getAuthorizedViewName ();
129
+ materializedViewName = ((ReadRowsRequest ) request ).getMaterializedViewName ();
127
130
} else if (request instanceof MutateRowsRequest ) {
128
131
tableName = ((MutateRowsRequest ) request ).getTableName ();
129
132
authorizedViewName = ((MutateRowsRequest ) request ).getAuthorizedViewName ();
@@ -133,6 +136,7 @@ static String extractTableId(Object request) {
133
136
} else if (request instanceof SampleRowKeysRequest ) {
134
137
tableName = ((SampleRowKeysRequest ) request ).getTableName ();
135
138
authorizedViewName = ((SampleRowKeysRequest ) request ).getAuthorizedViewName ();
139
+ materializedViewName = ((SampleRowKeysRequest ) request ).getMaterializedViewName ();
136
140
} else if (request instanceof CheckAndMutateRowRequest ) {
137
141
tableName = ((CheckAndMutateRowRequest ) request ).getTableName ();
138
142
authorizedViewName = ((CheckAndMutateRowRequest ) request ).getAuthorizedViewName ();
@@ -150,6 +154,9 @@ static String extractTableId(Object request) {
150
154
if (authorizedViewName != null && !authorizedViewName .isEmpty ()) {
151
155
return AuthorizedViewName .parse (authorizedViewName ).getTable ();
152
156
}
157
+ if (materializedViewName != null && !materializedViewName .isEmpty ()) {
158
+ return MaterializedViewName .parse (materializedViewName ).getMaterializedView ();
159
+ }
153
160
return "<unspecified>" ;
154
161
}
155
162
0 commit comments