@@ -127,6 +127,12 @@ public final class RouterMetrics {
127127 private MutableGaugeInt numGetRMNodeLabelsFailedRetrieved ;
128128 @ Metric ("# of checkUserAccessToQueue failed to be retrieved" )
129129 private MutableGaugeInt numCheckUserAccessToQueueFailedRetrieved ;
130+ @ Metric ("# of getDelegationToken failed to be retrieved" )
131+ private MutableGaugeInt numGetDelegationTokenFailedRetrieved ;
132+ @ Metric ("# of renewDelegationToken failed to be retrieved" )
133+ private MutableGaugeInt numRenewDelegationTokenFailedRetrieved ;
134+ @ Metric ("# of renewDelegationToken failed to be retrieved" )
135+ private MutableGaugeInt numCancelDelegationTokenFailedRetrieved ;
130136
131137 // Aggregate metrics are shared, and don't have to be looked up per call
132138 @ Metric ("Total number of successful Submitted apps and latency(ms)" )
@@ -215,6 +221,12 @@ public final class RouterMetrics {
215221 private MutableRate totalSucceededGetRMNodeLabelsRetrieved ;
216222 @ Metric ("Total number of successful Retrieved CheckUserAccessToQueue and latency(ms)" )
217223 private MutableRate totalSucceededCheckUserAccessToQueueRetrieved ;
224+ @ Metric ("Total number of successful Retrieved GetDelegationToken and latency(ms)" )
225+ private MutableRate totalSucceededGetDelegationTokenRetrieved ;
226+ @ Metric ("Total number of successful Retrieved RenewDelegationToken and latency(ms)" )
227+ private MutableRate totalSucceededRenewDelegationTokenRetrieved ;
228+ @ Metric ("Total number of successful Retrieved CancelDelegationToken and latency(ms)" )
229+ private MutableRate totalSucceededCancelDelegationTokenRetrieved ;
218230
219231 /**
220232 * Provide quantile counters for all latencies.
@@ -262,6 +274,9 @@ public final class RouterMetrics {
262274 private MutableQuantiles getRefreshQueuesLatency ;
263275 private MutableQuantiles getRMNodeLabelsLatency ;
264276 private MutableQuantiles checkUserAccessToQueueLatency ;
277+ private MutableQuantiles getDelegationTokenLatency ;
278+ private MutableQuantiles renewDelegationTokenLatency ;
279+ private MutableQuantiles cancelDelegationTokenLatency ;
265280
266281 private static volatile RouterMetrics instance = null ;
267282 private static MetricsRegistry registry ;
@@ -423,6 +438,15 @@ private RouterMetrics() {
423438
424439 checkUserAccessToQueueLatency = registry .newQuantiles ("checkUserAccessToQueueLatency" ,
425440 "latency of get apptimeouts timeouts" , "ops" , "latency" , 10 );
441+
442+ getDelegationTokenLatency = registry .newQuantiles ("getDelegationTokenLatency" ,
443+ "latency of get delegation token timeouts" , "ops" , "latency" , 10 );
444+
445+ renewDelegationTokenLatency = registry .newQuantiles ("renewDelegationTokenLatency" ,
446+ "latency of renew delegation token timeouts" , "ops" , "latency" , 10 );
447+
448+ cancelDelegationTokenLatency = registry .newQuantiles ("cancelDelegationTokenLatency" ,
449+ "latency of cancel delegation token timeouts" , "ops" , "latency" , 10 );
426450 }
427451
428452 public static RouterMetrics getMetrics () {
@@ -655,10 +679,25 @@ public long getNumSucceededGetRMNodeLabelsRetrieved() {
655679 }
656680
657681 @ VisibleForTesting
658- public long getNumSucceededCheckUserAccessToQueueRetrievedRetrieved () {
682+ public long getNumSucceededCheckUserAccessToQueueRetrieved () {
659683 return totalSucceededCheckUserAccessToQueueRetrieved .lastStat ().numSamples ();
660684 }
661685
686+ @ VisibleForTesting
687+ public long getNumSucceededGetDelegationTokenRetrieved () {
688+ return totalSucceededGetDelegationTokenRetrieved .lastStat ().numSamples ();
689+ }
690+
691+ @ VisibleForTesting
692+ public long getNumSucceededRenewDelegationTokenRetrieved () {
693+ return totalSucceededRenewDelegationTokenRetrieved .lastStat ().numSamples ();
694+ }
695+
696+ @ VisibleForTesting
697+ public long getNumSucceededCancelDelegationTokenRetrieved () {
698+ return totalSucceededCancelDelegationTokenRetrieved .lastStat ().numSamples ();
699+ }
700+
662701 @ VisibleForTesting
663702 public double getLatencySucceededAppsCreated () {
664703 return totalSucceededAppsCreated .lastStat ().mean ();
@@ -874,6 +913,21 @@ public double getLatencySucceededCheckUserAccessToQueueRetrieved() {
874913 return totalSucceededCheckUserAccessToQueueRetrieved .lastStat ().mean ();
875914 }
876915
916+ @ VisibleForTesting
917+ public double getLatencySucceededGetDelegationTokenRetrieved () {
918+ return totalSucceededGetDelegationTokenRetrieved .lastStat ().mean ();
919+ }
920+
921+ @ VisibleForTesting
922+ public double getLatencySucceededRenewDelegationTokenRetrieved () {
923+ return totalSucceededRenewDelegationTokenRetrieved .lastStat ().mean ();
924+ }
925+
926+ @ VisibleForTesting
927+ public double getLatencySucceededCancelDelegationTokenRetrieved () {
928+ return totalSucceededCancelDelegationTokenRetrieved .lastStat ().mean ();
929+ }
930+
877931 @ VisibleForTesting
878932 public int getAppsFailedCreated () {
879933 return numAppsFailedCreated .value ();
@@ -1068,6 +1122,18 @@ public int getCheckUserAccessToQueueFailedRetrieved() {
10681122 return numCheckUserAccessToQueueFailedRetrieved .value ();
10691123 }
10701124
1125+ public int getDelegationTokenFailedRetrieved () {
1126+ return numGetDelegationTokenFailedRetrieved .value ();
1127+ }
1128+
1129+ public int getRenewDelegationTokenFailedRetrieved () {
1130+ return numRenewDelegationTokenFailedRetrieved .value ();
1131+ }
1132+
1133+ public int getCancelDelegationTokenFailedRetrieved () {
1134+ return numCancelDelegationTokenFailedRetrieved .value ();
1135+ }
1136+
10711137 public void succeededAppsCreated (long duration ) {
10721138 totalSucceededAppsCreated .add (duration );
10731139 getNewApplicationLatency .add (duration );
@@ -1283,6 +1349,21 @@ public void succeededCheckUserAccessToQueueRetrieved(long duration) {
12831349 checkUserAccessToQueueLatency .add (duration );
12841350 }
12851351
1352+ public void succeededGetDelegationTokenRetrieved (long duration ) {
1353+ totalSucceededGetDelegationTokenRetrieved .add (duration );
1354+ getDelegationTokenLatency .add (duration );
1355+ }
1356+
1357+ public void succeededRenewDelegationTokenRetrieved (long duration ) {
1358+ totalSucceededRenewDelegationTokenRetrieved .add (duration );
1359+ renewDelegationTokenLatency .add (duration );
1360+ }
1361+
1362+ public void succeededCancelDelegationTokenRetrieved (long duration ) {
1363+ totalSucceededCancelDelegationTokenRetrieved .add (duration );
1364+ cancelDelegationTokenLatency .add (duration );
1365+ }
1366+
12861367 public void incrAppsFailedCreated () {
12871368 numAppsFailedCreated .incr ();
12881369 }
@@ -1454,4 +1535,16 @@ public void incrGetRMNodeLabelsFailedRetrieved() {
14541535 public void incrCheckUserAccessToQueueFailedRetrieved () {
14551536 numCheckUserAccessToQueueFailedRetrieved .incr ();
14561537 }
1457- }
1538+
1539+ public void incrGetDelegationTokenFailedRetrieved () {
1540+ numGetDelegationTokenFailedRetrieved .incr ();
1541+ }
1542+
1543+ public void incrRenewDelegationTokenFailedRetrieved () {
1544+ numRenewDelegationTokenFailedRetrieved .incr ();
1545+ }
1546+
1547+ public void incrCancelDelegationTokenFailedRetrieved () {
1548+ numCancelDelegationTokenFailedRetrieved .incr ();
1549+ }
1550+ }
0 commit comments