|
| 1 | +package io.javaoperatorsdk.operator; |
| 2 | + |
| 3 | +import io.fabric8.kubernetes.client.CustomResource; |
| 4 | +import io.javaoperatorsdk.operator.api.Context; |
| 5 | +import io.javaoperatorsdk.operator.api.DeleteControl; |
| 6 | +import io.javaoperatorsdk.operator.api.ResourceController; |
| 7 | +import io.javaoperatorsdk.operator.api.UpdateControl; |
| 8 | +import io.javaoperatorsdk.operator.api.config.ControllerConfiguration; |
| 9 | +import io.micrometer.core.instrument.*; |
| 10 | +import io.micrometer.core.instrument.distribution.DistributionStatisticConfig; |
| 11 | +import io.micrometer.core.instrument.distribution.pause.PauseDetector; |
| 12 | +import io.micrometer.core.instrument.noop.*; |
| 13 | +import java.util.concurrent.TimeUnit; |
| 14 | +import java.util.function.ToDoubleFunction; |
| 15 | +import java.util.function.ToLongFunction; |
| 16 | + |
| 17 | +public class Metrics { |
| 18 | + public static final Metrics NOOP = new Metrics(new NoopMeterRegistry(Clock.SYSTEM)); |
| 19 | + private final MeterRegistry registry; |
| 20 | + |
| 21 | + public Metrics(MeterRegistry registry) { |
| 22 | + this.registry = registry; |
| 23 | + } |
| 24 | + |
| 25 | + public <R extends CustomResource> UpdateControl<R> timeControllerCreateOrUpdate( |
| 26 | + ResourceController<R> controller, |
| 27 | + ControllerConfiguration<R> configuration, |
| 28 | + R resource, |
| 29 | + Context<R> context) { |
| 30 | + final var name = configuration.getName(); |
| 31 | + final var timer = |
| 32 | + Timer.builder("operator.sdk.controllers.execution.createorupdate") |
| 33 | + .tags("controller", name) |
| 34 | + .publishPercentiles(0.3, 0.5, 0.95) |
| 35 | + .publishPercentileHistogram() |
| 36 | + .register(registry); |
| 37 | + try { |
| 38 | + final var result = timer.record(() -> controller.createOrUpdateResource(resource, context)); |
| 39 | + String successType = "cr"; |
| 40 | + if (result.isUpdateStatusSubResource()) { |
| 41 | + successType = "status"; |
| 42 | + } |
| 43 | + if (result.isUpdateCustomResourceAndStatusSubResource()) { |
| 44 | + successType = "both"; |
| 45 | + } |
| 46 | + registry |
| 47 | + .counter( |
| 48 | + "operator.sdk.controllers.execution.success", "controller", name, "type", successType) |
| 49 | + .increment(); |
| 50 | + return result; |
| 51 | + } catch (Exception e) { |
| 52 | + registry |
| 53 | + .counter( |
| 54 | + "operator.sdk.controllers.execution.failure", |
| 55 | + "controller", |
| 56 | + name, |
| 57 | + "exception", |
| 58 | + e.getClass().getSimpleName()) |
| 59 | + .increment(); |
| 60 | + throw e; |
| 61 | + } |
| 62 | + } |
| 63 | + |
| 64 | + public DeleteControl timeControllerDelete( |
| 65 | + ResourceController controller, |
| 66 | + ControllerConfiguration configuration, |
| 67 | + CustomResource resource, |
| 68 | + Context context) { |
| 69 | + final var name = configuration.getName(); |
| 70 | + final var timer = |
| 71 | + Timer.builder("operator.sdk.controllers.execution.delete") |
| 72 | + .tags("controller", name) |
| 73 | + .publishPercentiles(0.3, 0.5, 0.95) |
| 74 | + .publishPercentileHistogram() |
| 75 | + .register(registry); |
| 76 | + try { |
| 77 | + final var result = timer.record(() -> controller.deleteResource(resource, context)); |
| 78 | + String successType = "notDelete"; |
| 79 | + if (result == DeleteControl.DEFAULT_DELETE) { |
| 80 | + successType = "delete"; |
| 81 | + } |
| 82 | + registry |
| 83 | + .counter( |
| 84 | + "operator.sdk.controllers.execution.success", "controller", name, "type", successType) |
| 85 | + .increment(); |
| 86 | + return result; |
| 87 | + } catch (Exception e) { |
| 88 | + registry |
| 89 | + .counter( |
| 90 | + "operator.sdk.controllers.execution.failure", |
| 91 | + "controller", |
| 92 | + name, |
| 93 | + "exception", |
| 94 | + e.getClass().getSimpleName()) |
| 95 | + .increment(); |
| 96 | + throw e; |
| 97 | + } |
| 98 | + } |
| 99 | + |
| 100 | + public void timeControllerRetry() { |
| 101 | + |
| 102 | + registry |
| 103 | + .counter( |
| 104 | + "operator.sdk.retry.on.exception", "retry", "retryCounter", "type", |
| 105 | + "retryException") |
| 106 | + .increment(); |
| 107 | + |
| 108 | + } |
| 109 | + |
| 110 | + public void timeControllerEvents() { |
| 111 | + |
| 112 | + registry |
| 113 | + .counter( |
| 114 | + "operator.sdk.total.events.received", "events", "totalEvents", "type", |
| 115 | + "eventsReceived") |
| 116 | + .increment(); |
| 117 | + |
| 118 | + } |
| 119 | + |
| 120 | + public static class NoopMeterRegistry extends MeterRegistry { |
| 121 | + public NoopMeterRegistry(Clock clock) { |
| 122 | + super(clock); |
| 123 | + } |
| 124 | + |
| 125 | + @Override |
| 126 | + protected <T> Gauge newGauge(Meter.Id id, T t, ToDoubleFunction<T> toDoubleFunction) { |
| 127 | + return new NoopGauge(id); |
| 128 | + } |
| 129 | + |
| 130 | + @Override |
| 131 | + protected Counter newCounter(Meter.Id id) { |
| 132 | + return new NoopCounter(id); |
| 133 | + } |
| 134 | + |
| 135 | + @Override |
| 136 | + protected Timer newTimer( |
| 137 | + Meter.Id id, |
| 138 | + DistributionStatisticConfig distributionStatisticConfig, |
| 139 | + PauseDetector pauseDetector) { |
| 140 | + return new NoopTimer(id); |
| 141 | + } |
| 142 | + |
| 143 | + @Override |
| 144 | + protected DistributionSummary newDistributionSummary( |
| 145 | + Meter.Id id, DistributionStatisticConfig distributionStatisticConfig, double v) { |
| 146 | + return new NoopDistributionSummary(id); |
| 147 | + } |
| 148 | + |
| 149 | + @Override |
| 150 | + protected Meter newMeter(Meter.Id id, Meter.Type type, Iterable<Measurement> iterable) { |
| 151 | + return new NoopMeter(id); |
| 152 | + } |
| 153 | + |
| 154 | + @Override |
| 155 | + protected <T> FunctionTimer newFunctionTimer( |
| 156 | + Meter.Id id, |
| 157 | + T t, |
| 158 | + ToLongFunction<T> toLongFunction, |
| 159 | + ToDoubleFunction<T> toDoubleFunction, |
| 160 | + TimeUnit timeUnit) { |
| 161 | + return new NoopFunctionTimer(id); |
| 162 | + } |
| 163 | + |
| 164 | + @Override |
| 165 | + protected <T> FunctionCounter newFunctionCounter( |
| 166 | + Meter.Id id, T t, ToDoubleFunction<T> toDoubleFunction) { |
| 167 | + return new NoopFunctionCounter(id); |
| 168 | + } |
| 169 | + |
| 170 | + @Override |
| 171 | + protected TimeUnit getBaseTimeUnit() { |
| 172 | + return TimeUnit.SECONDS; |
| 173 | + } |
| 174 | + |
| 175 | + @Override |
| 176 | + protected DistributionStatisticConfig defaultHistogramConfig() { |
| 177 | + return DistributionStatisticConfig.NONE; |
| 178 | + } |
| 179 | + } |
| 180 | +} |
0 commit comments