proxy/pkg/metrics/metrics.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

//Package metrics is a system-independent module
//it consider metrics key as first-class citizen
//each function is for recording one kind of metrics key and value
//it expose standard API to record runtime metrics for a service
//use prom as default metrics system
package metrics

import (
	mesherConf "github.com/apache/servicecomb-mesher/proxy/config"
	"github.com/go-chassis/go-chassis/pkg/runtime"
	"sync"
	"time"
)

//Constants with attributes for metrics data
//Label start with word "L"
const (
	LTotalRequest          = "requests_total"
	LTotalSuccess          = "successes_total"
	LTotalFailures         = "failures_total"
	LRequestLatencySeconds = "request_latency_seconds"
	LError4XX              = "status_4xx"
	LError5XX              = "status_5xx"
	LServiceName           = "service_name"
	LApp                   = "app"
	LVersion               = "version"
	LStartTime             = "start_time_seconds"
)

var (
	//LabelNames is a fixed list with service name, appID, version
	LabelNames = []string{LServiceName, LApp, LVersion}
	mutex      = sync.Mutex{}
)

//Recorder export the metrics into various of system
type Recorder interface {
	RecordStatus(labelValues map[string]string, status int, opts *RecordOptions)
	RecordLatency(labelValues map[string]string, latency float64, opts *RecordOptions)
	RecordStartTime(labelValues map[string]string, start time.Time, opts *RecordOptions)
}

//Options define recorder options
type Options struct {
	LabelNames             []string //default label names, if RecordOptions LabelNames is nil
	EnableGoRuntimeMetrics bool
}

//RecordOptions is options for record method
type RecordOptions struct {
	LabelNames []string // able top custom label names
}

var defaultRecorder Recorder = &promRecorder{}

//RecordStatus record an operation status
func RecordStatus(labelValues map[string]string, statusCode int, opts *RecordOptions) {
	defaultRecorder.RecordStatus(labelValues, statusCode, opts)
}

//RecordLatency record an operation latency
func RecordLatency(labelValues map[string]string, latency float64, opts *RecordOptions) {
	defaultRecorder.RecordLatency(labelValues, latency, opts)
}

//RecordStartTime record app start time
func RecordStartTime(labelValues map[string]string, start time.Time, opts *RecordOptions) {
	defaultRecorder.RecordStartTime(labelValues, start, opts)
}

//Init initiate the recorder
func Init() error {
	var err error
	LabelValues := map[string]string{LServiceName: runtime.ServiceName, LApp: runtime.App, LVersion: runtime.Version}
	defaultRecorder, err = NewPromRecorder(&Options{
		LabelNames:             LabelNames,
		EnableGoRuntimeMetrics: mesherConf.GetConfig().Admin.GoRuntimeMetrics,
	})
	if err != nil {
		return err
	}
	RecordStartTime(LabelValues, time.Now(), nil)
	return nil
}