blob: 56423822001529101fa894e604bb7e9c4b9f7f4b [file] [log] [blame]
/*
* 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 rest
import (
"fmt"
"github.com/apache/incubator-servicecomb-service-center/pkg/rest"
"github.com/apache/incubator-servicecomb-service-center/server/core"
"github.com/prometheus/client_golang/prometheus"
"net/http"
"strconv"
"strings"
"time"
)
var (
incomingRequests = prometheus.NewCounterVec(
prometheus.CounterOpts{
Namespace: "service_center",
Subsystem: "http",
Name: "request_total",
Help: "Counter of requests received into ROA handler",
}, []string{"method", "code", "instance", "api"})
successfulRequests = prometheus.NewCounterVec(
prometheus.CounterOpts{
Namespace: "service_center",
Subsystem: "http",
Name: "success_total",
Help: "Counter of successful requests processed by ROA handler",
}, []string{"method", "code", "instance", "api"})
reqDurations = prometheus.NewSummaryVec(
prometheus.SummaryOpts{
Namespace: "service_center",
Subsystem: "http",
Name: "request_durations_microseconds",
Help: "HTTP request latency summary of ROA handler",
Objectives: prometheus.DefObjectives,
}, []string{"method", "instance", "api"})
)
func init() {
prometheus.MustRegister(incomingRequests, successfulRequests, reqDurations)
http.Handle("/metrics", prometheus.Handler())
}
func ReportRequestCompleted(w http.ResponseWriter, r *http.Request, start time.Time) {
instance := fmt.Sprint(core.Instance.Endpoints)
elapsed := float64(time.Since(start).Nanoseconds()) / float64(time.Microsecond)
route, _ := r.Context().Value(rest.CTX_MATCH_PATTERN).(string)
if strings.Index(r.Method, "WATCH") != 0 {
reqDurations.WithLabelValues(r.Method, instance, route).Observe(elapsed)
}
success, code := codeOf(w.Header())
incomingRequests.WithLabelValues(r.Method, code, instance, route).Inc()
if success {
successfulRequests.WithLabelValues(r.Method, code, instance, route).Inc()
}
}
func codeOf(h http.Header) (bool, string) {
statusCode := h.Get("X-Response-Status")
if statusCode == "" {
return true, "200"
}
if code, _ := strconv.Atoi(statusCode); code >= http.StatusOK && code <= http.StatusAccepted {
return true, statusCode
}
return false, statusCode
}