|
| 1 | +/* |
| 2 | + * Copyright OpenSearch Contributors |
| 3 | + * SPDX-License-Identifier: Apache-2.0 |
| 4 | + */ |
| 5 | + |
| 6 | +package org.opensearch.indexmanagement.spi.indexstatemanagement.metrics |
| 7 | + |
| 8 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.AliasActionMetrics |
| 9 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.AllocationActionMetrics |
| 10 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.CloseActionMetrics |
| 11 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.DeleteActionMetrics |
| 12 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.ForceMergeActionMetrics |
| 13 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.MoveShardActionMetrics |
| 14 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.NotificationActionMetrics |
| 15 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.OpenActionMetrics |
| 16 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.ReplicaCountActionMetrics |
| 17 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.RolloverActionMetrics |
| 18 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.SetIndexPriorityActionMetrics |
| 19 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.SetReadOnlyActionMetrics |
| 20 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.ShrinkActionMetrics |
| 21 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.SnapshotActionMetrics |
| 22 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.metrics.actionmetrics.TransitionActionMetrics |
| 23 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.model.StepContext |
| 24 | +import org.opensearch.indexmanagement.spi.indexstatemanagement.model.StepMetaData |
| 25 | +import org.opensearch.telemetry.metrics.MetricsRegistry |
| 26 | +import org.opensearch.telemetry.metrics.tags.Tags |
| 27 | + |
| 28 | +abstract class ActionMetrics { |
| 29 | + abstract val actionName: String |
| 30 | + |
| 31 | + fun createTags(context: StepContext): Tags { |
| 32 | + val tags = Tags.create() |
| 33 | + .addTag("index_name", context.metadata.index) |
| 34 | + .addTag("policy_id", context.metadata.policyID) |
| 35 | + .addTag("node_id", context.clusterService.nodeName ?: "") |
| 36 | + .addTag("index_uuid", context.metadata.indexUuid) |
| 37 | + return tags |
| 38 | + } |
| 39 | + |
| 40 | + abstract fun emitMetrics( |
| 41 | + context: StepContext, |
| 42 | + indexManagementActionsMetrics: IndexManagementActionsMetrics, |
| 43 | + stepMetaData: StepMetaData?, |
| 44 | + ) |
| 45 | +} |
| 46 | + |
| 47 | +class IndexManagementActionsMetrics private constructor() { |
| 48 | + private lateinit var metricsRegistry: MetricsRegistry |
| 49 | + private lateinit var actionMetricsMap: Map<String, ActionMetrics> |
| 50 | + |
| 51 | + companion object { |
| 52 | + val instance: IndexManagementActionsMetrics by lazy { HOLDER.instance } |
| 53 | + |
| 54 | + const val ROLLOVER = "rollover" |
| 55 | + const val NOTIFICATION = "notification" |
| 56 | + const val FORCE_MERGE = "force_merge" |
| 57 | + const val DELETE = "delete" |
| 58 | + const val REPLICA_COUNT = "replica_count" |
| 59 | + const val TRANSITION = "transition" |
| 60 | + const val CLOSE = "close" |
| 61 | + const val SET_INDEX_PRIORITY = "set_index_priority" |
| 62 | + const val OPEN = "open" |
| 63 | + const val MOVE_SHARD = "move_shard" |
| 64 | + const val SET_READ_ONLY = "set_read_only" |
| 65 | + const val SHRINK = "shrink" |
| 66 | + const val SNAPSHOT = "snapshot" |
| 67 | + const val ALIAS_ACTION = "alias_action" |
| 68 | + const val ALLOCATION = "allocation" |
| 69 | + |
| 70 | + private object HOLDER { |
| 71 | + val instance = IndexManagementActionsMetrics() |
| 72 | + } |
| 73 | + } |
| 74 | + |
| 75 | + fun initialize(metricsRegistry: MetricsRegistry) { |
| 76 | + this.metricsRegistry = metricsRegistry |
| 77 | + |
| 78 | + RolloverActionMetrics.instance.initializeCounters(metricsRegistry) |
| 79 | + NotificationActionMetrics.instance.initializeCounters(metricsRegistry) |
| 80 | + ForceMergeActionMetrics.instance.initializeCounters(metricsRegistry) |
| 81 | + DeleteActionMetrics.instance.initializeCounters(metricsRegistry) |
| 82 | + ReplicaCountActionMetrics.instance.initializeCounters(metricsRegistry) |
| 83 | + TransitionActionMetrics.instance.initializeCounters(metricsRegistry) |
| 84 | + CloseActionMetrics.instance.initializeCounters(metricsRegistry) |
| 85 | + SetIndexPriorityActionMetrics.instance.initializeCounters(metricsRegistry) |
| 86 | + OpenActionMetrics.instance.initializeCounters(metricsRegistry) |
| 87 | + MoveShardActionMetrics.instance.initializeCounters(metricsRegistry) |
| 88 | + SetReadOnlyActionMetrics.instance.initializeCounters(metricsRegistry) |
| 89 | + ShrinkActionMetrics.instance.initializeCounters(metricsRegistry) |
| 90 | + SnapshotActionMetrics.instance.initializeCounters(metricsRegistry) |
| 91 | + AliasActionMetrics.instance.initializeCounters(metricsRegistry) |
| 92 | + AllocationActionMetrics.instance.initializeCounters(metricsRegistry) |
| 93 | + |
| 94 | + actionMetricsMap = mapOf( |
| 95 | + ROLLOVER to RolloverActionMetrics.instance, |
| 96 | + NOTIFICATION to NotificationActionMetrics.instance, |
| 97 | + FORCE_MERGE to ForceMergeActionMetrics.instance, |
| 98 | + DELETE to DeleteActionMetrics.instance, |
| 99 | + REPLICA_COUNT to ReplicaCountActionMetrics.instance, |
| 100 | + TRANSITION to TransitionActionMetrics.instance, |
| 101 | + CLOSE to CloseActionMetrics.instance, |
| 102 | + SET_INDEX_PRIORITY to SetIndexPriorityActionMetrics.instance, |
| 103 | + OPEN to OpenActionMetrics.instance, |
| 104 | + MOVE_SHARD to MoveShardActionMetrics.instance, |
| 105 | + SET_READ_ONLY to SetReadOnlyActionMetrics.instance, |
| 106 | + SHRINK to ShrinkActionMetrics.instance, |
| 107 | + SNAPSHOT to SnapshotActionMetrics.instance, |
| 108 | + ALIAS_ACTION to AliasActionMetrics.instance, |
| 109 | + ALLOCATION to AllocationActionMetrics.instance, |
| 110 | + ) |
| 111 | + } |
| 112 | + |
| 113 | + fun getActionMetrics(actionName: String): ActionMetrics? { |
| 114 | + return actionMetricsMap[actionName] |
| 115 | + } |
| 116 | +} |
0 commit comments