// Copyright 2016 TiKV Project Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// See the License for the specific language governing permissions and
// limitations under the License.

package pd

import "github.com/prometheus/client_golang/prometheus"

var (
	cmdDuration = prometheus.NewHistogramVec(
		prometheus.HistogramOpts{
			Namespace: "pd_client",
			Subsystem: "cmd",
			Name:      "handle_cmds_duration_seconds",
			Help:      "Bucketed histogram of processing time (s) of handled success cmds.",
			Buckets:   prometheus.ExponentialBuckets(0.0005, 2, 13),
		}, []string{"type"})

	cmdFailedDuration = prometheus.NewHistogramVec(
		prometheus.HistogramOpts{
			Namespace: "pd_client",
			Subsystem: "cmd",
			Name:      "handle_failed_cmds_duration_seconds",
			Help:      "Bucketed histogram of processing time (s) of failed handled cmds.",
			Buckets:   prometheus.ExponentialBuckets(0.0005, 2, 13),
		}, []string{"type"})

	requestDuration = prometheus.NewHistogramVec(
		prometheus.HistogramOpts{
			Namespace: "pd_client",
			Subsystem: "request",
			Name:      "handle_requests_duration_seconds",
			Help:      "Bucketed histogram of processing time (s) of handled requests.",
			Buckets:   prometheus.ExponentialBuckets(0.0005, 2, 13),
		}, []string{"type"})

	tsoBatchSize = prometheus.NewHistogram(
		prometheus.HistogramOpts{
			Namespace: "pd_client",
			Subsystem: "request",
			Name:      "handle_tso_batch_size",
			Help:      "Bucketed histogram of the batch size of handled requests.",
			Buckets:   prometheus.ExponentialBuckets(1, 2, 13),
		})
)

var (
	// WithLabelValues is a heavy operation, define variable to avoid call it every time.
	cmdDurationWait                     = cmdDuration.WithLabelValues("wait")
	cmdDurationTSO                      = cmdDuration.WithLabelValues("tso")
	cmdDurationTSOAsyncWait             = cmdDuration.WithLabelValues("tso_async_wait")
	cmdDurationGetRegion                = cmdDuration.WithLabelValues("get_region")
	cmdDurationGetAllMembers            = cmdDuration.WithLabelValues("get_member_info")
	cmdDurationGetPrevRegion            = cmdDuration.WithLabelValues("get_prev_region")
	cmdDurationGetRegionByID            = cmdDuration.WithLabelValues("get_region_byid")
	cmdDurationScanRegions              = cmdDuration.WithLabelValues("scan_regions")
	cmdDurationGetStore                 = cmdDuration.WithLabelValues("get_store")
	cmdDurationGetAllStores             = cmdDuration.WithLabelValues("get_all_stores")
	cmdDurationUpdateGCSafePoint        = cmdDuration.WithLabelValues("update_gc_safe_point")
	cmdDurationUpdateServiceGCSafePoint = cmdDuration.WithLabelValues("update_service_gc_safe_point")
	cmdDurationScatterRegion            = cmdDuration.WithLabelValues("scatter_region")
	cmdDurationGetOperator              = cmdDuration.WithLabelValues("get_operator")

	cmdFailDurationGetRegion                  = cmdFailedDuration.WithLabelValues("get_region")
	cmdFailDurationTSO                        = cmdFailedDuration.WithLabelValues("tso")
	cmdFailDurationGetAllMembers              = cmdFailedDuration.WithLabelValues("get_member_info")
	cmdFailDurationGetPrevRegion              = cmdFailedDuration.WithLabelValues("get_prev_region")
	cmdFailedDurationGetRegionByID            = cmdFailedDuration.WithLabelValues("get_region_byid")
	cmdFailedDurationScanRegions              = cmdFailedDuration.WithLabelValues("scan_regions")
	cmdFailedDurationGetStore                 = cmdFailedDuration.WithLabelValues("get_store")
	cmdFailedDurationGetAllStores             = cmdFailedDuration.WithLabelValues("get_all_stores")
	cmdFailedDurationUpdateGCSafePoint        = cmdFailedDuration.WithLabelValues("update_gc_safe_point")
	cmdFailedDurationUpdateServiceGCSafePoint = cmdFailedDuration.WithLabelValues("update_service_gc_safe_point")
	requestDurationTSO                        = requestDuration.WithLabelValues("tso")
)

func init() {
	prometheus.MustRegister(cmdDuration)
	prometheus.MustRegister(cmdFailedDuration)
	prometheus.MustRegister(requestDuration)
	prometheus.MustRegister(tsoBatchSize)
}