2020-02-23 12:35:47 +01:00
package main
import (
2022-05-31 01:55:28 +02:00
"embed"
2020-02-23 12:35:47 +01:00
"flag"
"fmt"
2021-09-20 13:49:28 +02:00
"io"
2020-02-23 12:35:47 +01:00
"net/http"
2020-05-16 10:59:30 +02:00
"os"
2020-02-23 12:35:47 +01:00
"strings"
2020-11-04 19:29:18 +01:00
"sync/atomic"
2020-02-23 12:35:47 +01:00
"time"
2020-03-10 18:35:58 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/csvimport"
2021-09-28 21:47:45 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/datadog"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/graphite"
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/influx"
2020-09-26 03:29:45 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/native"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/opentsdb"
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/opentsdbhttp"
2020-07-10 11:00:35 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/prometheusimport"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/promremotewrite"
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/remotewrite"
"github.com/VictoriaMetrics/VictoriaMetrics/app/vmagent/vmimport"
2021-08-05 08:44:29 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/auth"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/buildinfo"
2022-05-25 08:57:11 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/bytesutil"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/envflag"
2020-12-03 20:40:30 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/flagutil"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/httpserver"
2021-03-15 20:37:13 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/influxutils"
2020-02-23 12:35:47 +01:00
graphiteserver "github.com/VictoriaMetrics/VictoriaMetrics/lib/ingestserver/graphite"
2020-02-25 18:09:46 +01:00
influxserver "github.com/VictoriaMetrics/VictoriaMetrics/lib/ingestserver/influx"
2020-02-23 12:35:47 +01:00
opentsdbserver "github.com/VictoriaMetrics/VictoriaMetrics/lib/ingestserver/opentsdb"
opentsdbhttpserver "github.com/VictoriaMetrics/VictoriaMetrics/lib/ingestserver/opentsdbhttp"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/logger"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/procutil"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promscrape"
2020-09-28 03:11:55 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/protoparser/common"
2022-07-21 18:58:22 +02:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/pushmetrics"
2020-02-23 12:35:47 +01:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/writeconcurrencylimiter"
"github.com/VictoriaMetrics/metrics"
)
var (
2020-02-26 19:57:51 +01:00
httpListenAddr = flag . String ( "httpListenAddr" , ":8429" , "TCP address to listen for http connections. " +
"Set this flag to empty value in order to disable listening on any port. This mode may be useful for running multiple vmagent instances on the same server. " +
"Note that /targets and /metrics pages aren't available if -httpListenAddr=''" )
2022-05-13 15:49:22 +02:00
influxListenAddr = flag . String ( "influxListenAddr" , "" , "TCP and UDP address to listen for InfluxDB line protocol data. Usually :8089 must be set. Doesn't work if empty. " +
2021-03-15 20:59:25 +01:00
"This flag isn't needed when ingesting data over HTTP - just send it to http://<vmagent>:8429/write" )
2020-02-23 12:35:47 +01:00
graphiteListenAddr = flag . String ( "graphiteListenAddr" , "" , "TCP and UDP address to listen for Graphite plaintext data. Usually :2003 must be set. Doesn't work if empty" )
opentsdbListenAddr = flag . String ( "opentsdbListenAddr" , "" , "TCP and UDP address to listen for OpentTSDB metrics. " +
"Telnet put messages and HTTP /api/put messages are simultaneously served on TCP port. " +
"Usually :4242 must be set. Doesn't work if empty" )
opentsdbHTTPListenAddr = flag . String ( "opentsdbHTTPListenAddr" , "" , "TCP address to listen for OpentTSDB HTTP put requests. Usually :4242 must be set. Doesn't work if empty" )
2021-11-01 15:44:51 +01:00
configAuthKey = flag . String ( "configAuthKey" , "" , "Authorization key for accessing /config page. It must be passed via authKey query arg" )
2020-05-21 14:22:01 +02:00
dryRun = flag . Bool ( "dryRun" , false , "Whether to check only config files without running vmagent. The following files are checked: " +
2020-11-25 21:26:23 +01:00
"-promscrape.config, -remoteWrite.relabelConfig, -remoteWrite.urlRelabelConfig . " +
2022-02-08 14:37:38 +01:00
"Unknown config entries aren't allowed in -promscrape.config by default. This can be changed by passing -promscrape.config.strictParse=false command-line flag" )
2020-02-23 12:35:47 +01:00
)
var (
2020-02-25 18:09:46 +01:00
influxServer * influxserver . Server
2020-02-23 12:35:47 +01:00
graphiteServer * graphiteserver . Server
opentsdbServer * opentsdbserver . Server
opentsdbhttpServer * opentsdbhttpserver . Server
2022-05-31 11:27:54 +02:00
)
var (
2022-05-31 01:55:28 +02:00
//go:embed static
staticFiles embed . FS
staticServer = http . FileServer ( http . FS ( staticFiles ) )
2020-02-23 12:35:47 +01:00
)
func main ( ) {
2020-05-16 10:59:30 +02:00
// Write flags and help message to stdout, since it is easier to grep or pipe.
flag . CommandLine . SetOutput ( os . Stdout )
2020-06-05 09:37:36 +02:00
flag . Usage = usage
2020-02-23 12:35:47 +01:00
envflag . Parse ( )
2020-09-29 18:48:53 +02:00
remotewrite . InitSecretFlags ( )
2020-02-23 12:35:47 +01:00
buildinfo . Init ( )
logger . Init ( )
2022-07-22 12:35:58 +02:00
pushmetrics . Init ( )
2020-05-21 14:22:01 +02:00
2020-11-25 21:59:13 +01:00
if promscrape . IsDryRun ( ) {
if err := promscrape . CheckConfig ( ) ; err != nil {
logger . Fatalf ( "error when checking -promscrape.config: %s" , err )
}
logger . Infof ( "-promscrape.config is ok; exitting with 0 status code" )
return
}
2020-05-21 14:22:01 +02:00
if * dryRun {
if err := remotewrite . CheckRelabelConfigs ( ) ; err != nil {
logger . Fatalf ( "error when checking relabel configs: %s" , err )
}
if err := promscrape . CheckConfig ( ) ; err != nil {
2020-11-25 21:59:13 +01:00
logger . Fatalf ( "error when checking -promscrape.config: %s" , err )
2020-05-21 14:22:01 +02:00
}
logger . Infof ( "all the configs are ok; exitting with 0 status code" )
return
}
2020-02-23 12:35:47 +01:00
logger . Infof ( "starting vmagent at %q..." , * httpListenAddr )
startTime := time . Now ( )
2021-08-05 08:46:19 +02:00
remotewrite . Init ( )
2020-09-28 03:11:55 +02:00
common . StartUnmarshalWorkers ( )
2020-02-23 12:35:47 +01:00
writeconcurrencylimiter . Init ( )
2020-02-25 18:09:46 +01:00
if len ( * influxListenAddr ) > 0 {
2021-09-20 13:49:28 +02:00
influxServer = influxserver . MustStart ( * influxListenAddr , func ( r io . Reader ) error {
return influx . InsertHandlerForReader ( r , false )
} )
2020-02-25 18:09:46 +01:00
}
2020-02-23 12:35:47 +01:00
if len ( * graphiteListenAddr ) > 0 {
graphiteServer = graphiteserver . MustStart ( * graphiteListenAddr , graphite . InsertHandler )
}
if len ( * opentsdbListenAddr ) > 0 {
2022-08-24 15:18:50 +02:00
httpInsertHandler := getOpenTSDBHTTPInsertHandler ( )
opentsdbServer = opentsdbserver . MustStart ( * opentsdbListenAddr , opentsdb . InsertHandler , httpInsertHandler )
2020-02-23 12:35:47 +01:00
}
if len ( * opentsdbHTTPListenAddr ) > 0 {
2022-08-24 15:18:50 +02:00
httpInsertHandler := getOpenTSDBHTTPInsertHandler ( )
opentsdbhttpServer = opentsdbhttpserver . MustStart ( * opentsdbHTTPListenAddr , httpInsertHandler )
2020-02-23 12:35:47 +01:00
}
promscrape . Init ( remotewrite . Push )
2020-02-26 19:57:51 +01:00
if len ( * httpListenAddr ) > 0 {
go httpserver . Serve ( * httpListenAddr , requestHandler )
}
2020-02-23 12:35:47 +01:00
logger . Infof ( "started vmagent in %.3f seconds" , time . Since ( startTime ) . Seconds ( ) )
sig := procutil . WaitForSigterm ( )
logger . Infof ( "received signal %s" , sig )
startTime = time . Now ( )
2020-02-26 19:57:51 +01:00
if len ( * httpListenAddr ) > 0 {
logger . Infof ( "gracefully shutting down webservice at %q" , * httpListenAddr )
if err := httpserver . Stop ( * httpListenAddr ) ; err != nil {
logger . Fatalf ( "cannot stop the webservice: %s" , err )
}
logger . Infof ( "successfully shut down the webservice in %.3f seconds" , time . Since ( startTime ) . Seconds ( ) )
2020-02-23 12:35:47 +01:00
}
promscrape . Stop ( )
2020-02-25 18:09:46 +01:00
if len ( * influxListenAddr ) > 0 {
influxServer . MustStop ( )
}
2020-02-23 12:35:47 +01:00
if len ( * graphiteListenAddr ) > 0 {
graphiteServer . MustStop ( )
}
if len ( * opentsdbListenAddr ) > 0 {
opentsdbServer . MustStop ( )
}
if len ( * opentsdbHTTPListenAddr ) > 0 {
opentsdbhttpServer . MustStop ( )
}
2020-09-28 03:11:55 +02:00
common . StopUnmarshalWorkers ( )
2020-02-23 12:35:47 +01:00
remotewrite . Stop ( )
logger . Infof ( "successfully stopped vmagent in %.3f seconds" , time . Since ( startTime ) . Seconds ( ) )
}
2022-08-24 15:18:50 +02:00
func getOpenTSDBHTTPInsertHandler ( ) func ( req * http . Request ) error {
if ! remotewrite . MultitenancyEnabled ( ) {
return func ( req * http . Request ) error {
path := strings . Replace ( req . URL . Path , "//" , "/" , - 1 )
if path != "/api/put" {
return fmt . Errorf ( "unsupported path requested: %q; expecting '/api/put'" , path )
}
return opentsdbhttp . InsertHandler ( nil , req )
}
}
return func ( req * http . Request ) error {
path := strings . Replace ( req . URL . Path , "//" , "/" , - 1 )
at , err := getAuthTokenFromPath ( path )
if err != nil {
return fmt . Errorf ( "cannot obtain auth token from path %q: %w" , path , err )
}
return opentsdbhttp . InsertHandler ( at , req )
}
}
func getAuthTokenFromPath ( path string ) ( * auth . Token , error ) {
p , err := httpserver . ParsePath ( path )
if err != nil {
return nil , fmt . Errorf ( "cannot parse multitenant path: %w" , err )
}
if p . Prefix != "insert" {
return nil , fmt . Errorf ( ` unsupported multitenant prefix: %q; expected "insert" ` , p . Prefix )
}
if p . Suffix != "opentsdb/api/put" {
return nil , fmt . Errorf ( "unsupported path requested: %q; expecting 'opentsdb/api/put'" , p . Suffix )
}
return auth . NewToken ( p . AuthToken )
}
2020-02-23 12:35:47 +01:00
func requestHandler ( w http . ResponseWriter , r * http . Request ) bool {
2020-12-14 13:02:57 +01:00
if r . URL . Path == "/" {
2021-04-02 21:54:06 +02:00
if r . Method != "GET" {
return false
}
2022-03-21 14:34:49 +01:00
w . Header ( ) . Add ( "Content-Type" , "text/html; charset=utf-8" )
2021-04-30 08:19:08 +02:00
fmt . Fprintf ( w , "<h2>vmagent</h2>" )
fmt . Fprintf ( w , "See docs at <a href='https://docs.victoriametrics.com/vmagent.html'>https://docs.victoriametrics.com/vmagent.html</a></br>" )
fmt . Fprintf ( w , "Useful endpoints:</br>" )
httpserver . WriteAPIHelp ( w , [ ] [ 2 ] string {
2022-06-30 17:23:56 +02:00
{ "targets" , "status for discovered active targets" } ,
{ "service-discovery" , "labels before and after relabeling for discovered targets" } ,
{ "api/v1/targets" , "advanced information about discovered targets in JSON format" } ,
{ "config" , "-promscrape.config contents" } ,
{ "metrics" , "available service metrics" } ,
{ "flags" , "command-line flags" } ,
{ "-/reload" , "reload configuration" } ,
2021-04-30 08:19:08 +02:00
} )
2020-10-06 14:09:07 +02:00
return true
}
2021-08-05 08:44:29 +02:00
2022-06-30 17:23:56 +02:00
path := strings . Replace ( r . URL . Path , "//" , "/" , - 1 )
2022-11-26 01:40:23 +01:00
if strings . HasPrefix ( path , "/prometheus/api/v1/import/prometheus" ) || strings . HasPrefix ( path , "/api/v1/import/prometheus" ) {
prometheusimportRequests . Inc ( )
if err := prometheusimport . InsertHandler ( nil , r ) ; err != nil {
prometheusimportErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
}
2022-06-30 17:23:56 +02:00
if strings . HasPrefix ( path , "datadog/" ) {
2022-06-07 14:16:48 +02:00
// Trim suffix from paths starting from /datadog/ in order to support legacy DataDog agent.
// See https://github.com/VictoriaMetrics/VictoriaMetrics/pull/2670
2022-06-30 17:23:56 +02:00
path = strings . TrimSuffix ( path , "/" )
2022-06-07 14:16:48 +02:00
}
2022-06-30 17:23:56 +02:00
switch path {
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/write" , "/api/v1/write" :
2020-02-23 12:35:47 +01:00
prometheusWriteRequests . Inc ( )
2021-08-05 08:44:29 +02:00
if err := promremotewrite . InsertHandler ( nil , r ) ; err != nil {
2020-02-23 12:35:47 +01:00
prometheusWriteErrors . Inc ( )
2021-07-07 11:59:03 +02:00
httpserver . Errorf ( w , r , "%s" , err )
2020-02-23 12:35:47 +01:00
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/import" , "/api/v1/import" :
2020-02-23 12:35:47 +01:00
vmimportRequests . Inc ( )
2021-08-05 08:44:29 +02:00
if err := vmimport . InsertHandler ( nil , r ) ; err != nil {
2020-02-23 12:35:47 +01:00
vmimportErrors . Inc ( )
2021-07-07 11:59:03 +02:00
httpserver . Errorf ( w , r , "%s" , err )
2020-02-23 12:35:47 +01:00
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/import/csv" , "/api/v1/import/csv" :
2020-03-10 18:35:58 +01:00
csvimportRequests . Inc ( )
2021-08-05 08:44:29 +02:00
if err := csvimport . InsertHandler ( nil , r ) ; err != nil {
2020-03-10 18:35:58 +01:00
csvimportErrors . Inc ( )
2021-07-07 11:59:03 +02:00
httpserver . Errorf ( w , r , "%s" , err )
2020-03-10 18:35:58 +01:00
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/import/native" , "/api/v1/import/native" :
2020-09-26 03:29:45 +02:00
nativeimportRequests . Inc ( )
2021-08-05 08:44:29 +02:00
if err := native . InsertHandler ( nil , r ) ; err != nil {
2020-09-26 03:29:45 +02:00
nativeimportErrors . Inc ( )
2021-07-07 11:59:03 +02:00
httpserver . Errorf ( w , r , "%s" , err )
2020-09-26 03:29:45 +02:00
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
2022-10-01 12:47:52 +02:00
case "/influx/write" , "/influx/api/v2/write" , "/write" , "/api/v2/write" :
2020-02-23 12:35:47 +01:00
influxWriteRequests . Inc ( )
2021-08-05 08:44:29 +02:00
if err := influx . InsertHandlerForHTTP ( nil , r ) ; err != nil {
2020-02-23 12:35:47 +01:00
influxWriteErrors . Inc ( )
2021-07-07 11:59:03 +02:00
httpserver . Errorf ( w , r , "%s" , err )
2020-02-23 12:35:47 +01:00
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
2022-10-01 12:47:52 +02:00
case "/influx/query" , "/query" :
2020-02-23 12:35:47 +01:00
influxQueryRequests . Inc ( )
2021-03-15 20:37:13 +01:00
influxutils . WriteDatabaseNames ( w )
2020-02-23 12:35:47 +01:00
return true
2021-09-28 21:47:45 +02:00
case "/datadog/api/v1/series" :
datadogWriteRequests . Inc ( )
if err := datadog . InsertHandlerForHTTP ( nil , r ) ; err != nil {
datadogWriteErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
// See https://docs.datadoghq.com/api/latest/metrics/#submit-metrics
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
w . WriteHeader ( 202 )
fmt . Fprintf ( w , ` { "status":"ok"} ` )
return true
case "/datadog/api/v1/validate" :
datadogValidateRequests . Inc ( )
// See https://docs.datadoghq.com/api/latest/authentication/#validate-api-key
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
fmt . Fprintf ( w , ` { "valid":true} ` )
return true
case "/datadog/api/v1/check_run" :
datadogCheckRunRequests . Inc ( )
// See https://docs.datadoghq.com/api/latest/service-checks/#submit-a-service-check
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
w . WriteHeader ( 202 )
fmt . Fprintf ( w , ` { "status":"ok"} ` )
return true
2022-06-07 14:16:48 +02:00
case "/datadog/intake" :
2021-09-28 21:47:45 +02:00
datadogIntakeRequests . Inc ( )
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
fmt . Fprintf ( w , ` { } ` )
return true
2022-06-13 08:52:13 +02:00
case "/datadog/api/v1/metadata" :
datadogMetadataRequests . Inc ( )
w . Header ( ) . Set ( "Content-Type" , "application/json" )
fmt . Fprintf ( w , ` { } ` )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/targets" , "/targets" :
2020-02-25 17:13:11 +01:00
promscrapeTargetsRequests . Inc ( )
2020-12-14 13:02:57 +01:00
promscrape . WriteHumanReadableTargetsStatus ( w , r )
2020-02-25 17:13:11 +01:00
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/service-discovery" , "/service-discovery" :
2022-06-06 23:57:05 +02:00
promscrapeServiceDiscoveryRequests . Inc ( )
promscrape . WriteServiceDiscovery ( w , r )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/targets" , "/api/v1/targets" :
promscrapeAPIV1TargetsRequests . Inc ( )
w . Header ( ) . Set ( "Content-Type" , "application/json" )
state := r . FormValue ( "state" )
promscrape . WriteAPIV1Targets ( w , state )
return true
case "/prometheus/target_response" , "/target_response" :
2022-02-03 17:57:36 +01:00
promscrapeTargetResponseRequests . Inc ( )
if err := promscrape . WriteTargetResponse ( w , r ) ; err != nil {
promscrapeTargetResponseErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/config" , "/config" :
2021-11-01 15:44:51 +01:00
if * configAuthKey != "" && r . FormValue ( "authKey" ) != * configAuthKey {
err := & httpserver . ErrorWithStatusCode {
Err : fmt . Errorf ( "The provided authKey doesn't match -configAuthKey" ) ,
StatusCode : http . StatusUnauthorized ,
}
httpserver . Errorf ( w , r , "%s" , err )
return true
}
2021-10-12 15:23:42 +02:00
promscrapeConfigRequests . Inc ( )
w . Header ( ) . Set ( "Content-Type" , "text/plain; charset=utf-8" )
promscrape . WriteConfigData ( w )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/api/v1/status/config" , "/api/v1/status/config" :
2022-05-25 08:57:11 +02:00
// See https://prometheus.io/docs/prometheus/latest/querying/api/#config
if * configAuthKey != "" && r . FormValue ( "authKey" ) != * configAuthKey {
err := & httpserver . ErrorWithStatusCode {
Err : fmt . Errorf ( "The provided authKey doesn't match -configAuthKey" ) ,
StatusCode : http . StatusUnauthorized ,
}
httpserver . Errorf ( w , r , "%s" , err )
return true
}
promscrapeStatusConfigRequests . Inc ( )
w . Header ( ) . Set ( "Content-Type" , "application/json" )
var bb bytesutil . ByteBuffer
promscrape . WriteConfigData ( & bb )
fmt . Fprintf ( w , ` { "status":"success","data": { "yaml":%q}} ` , bb . B )
return true
2022-10-01 12:47:52 +02:00
case "/prometheus/-/reload" , "/-/reload" :
2020-04-30 01:00:32 +02:00
promscrapeConfigReloadRequests . Inc ( )
2020-04-30 01:15:39 +02:00
procutil . SelfSIGHUP ( )
2020-05-07 18:29:27 +02:00
w . WriteHeader ( http . StatusOK )
2020-04-30 01:00:32 +02:00
return true
2020-11-04 19:29:18 +01:00
case "/ready" :
if rdy := atomic . LoadInt32 ( & promscrape . PendingScrapeConfigs ) ; rdy > 0 {
errMsg := fmt . Sprintf ( "waiting for scrapes to init, left: %d" , rdy )
http . Error ( w , errMsg , http . StatusTooEarly )
} else {
2020-11-13 09:25:39 +01:00
w . Header ( ) . Set ( "Content-Type" , "text/plain; charset=utf-8" )
2020-11-04 19:29:18 +01:00
w . WriteHeader ( http . StatusOK )
w . Write ( [ ] byte ( "OK" ) )
}
return true
2022-05-31 01:55:28 +02:00
default :
if strings . HasPrefix ( r . URL . Path , "/static" ) {
staticServer . ServeHTTP ( w , r )
return true
}
if remotewrite . MultitenancyEnabled ( ) {
2022-06-30 17:23:56 +02:00
return processMultitenantRequest ( w , r , path )
2022-05-31 01:55:28 +02:00
}
return false
2020-02-23 12:35:47 +01:00
}
}
2021-08-05 08:46:19 +02:00
func processMultitenantRequest ( w http . ResponseWriter , r * http . Request , path string ) bool {
p , err := httpserver . ParsePath ( path )
if err != nil {
// Cannot parse multitenant path. Skip it - probably it will be parsed later.
return false
}
if p . Prefix != "insert" {
httpserver . Errorf ( w , r , ` unsupported multitenant prefix: %q; expected "insert" ` , p . Prefix )
return true
}
at , err := auth . NewToken ( p . AuthToken )
if err != nil {
httpserver . Errorf ( w , r , "cannot obtain auth token: %s" , err )
return true
}
2022-11-26 01:40:23 +01:00
if strings . HasPrefix ( p . Suffix , "prometheus/api/v1/import/prometheus" ) {
prometheusimportRequests . Inc ( )
if err := prometheusimport . InsertHandler ( at , r ) ; err != nil {
prometheusimportErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
}
2022-06-07 14:16:48 +02:00
if strings . HasPrefix ( p . Suffix , "datadog/" ) {
// Trim suffix from paths starting from /datadog/ in order to support legacy DataDog agent.
// See https://github.com/VictoriaMetrics/VictoriaMetrics/pull/2670
p . Suffix = strings . TrimSuffix ( p . Suffix , "/" )
}
2021-08-05 08:46:19 +02:00
switch p . Suffix {
case "prometheus/" , "prometheus" , "prometheus/api/v1/write" :
prometheusWriteRequests . Inc ( )
if err := promremotewrite . InsertHandler ( at , r ) ; err != nil {
prometheusWriteErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
case "prometheus/api/v1/import" :
vmimportRequests . Inc ( )
if err := vmimport . InsertHandler ( at , r ) ; err != nil {
vmimportErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
case "prometheus/api/v1/import/csv" :
csvimportRequests . Inc ( )
if err := csvimport . InsertHandler ( at , r ) ; err != nil {
csvimportErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
case "prometheus/api/v1/import/native" :
nativeimportRequests . Inc ( )
if err := native . InsertHandler ( at , r ) ; err != nil {
nativeimportErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
case "influx/write" , "influx/api/v2/write" :
influxWriteRequests . Inc ( )
if err := influx . InsertHandlerForHTTP ( at , r ) ; err != nil {
influxWriteErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
w . WriteHeader ( http . StatusNoContent )
return true
case "influx/query" :
influxQueryRequests . Inc ( )
influxutils . WriteDatabaseNames ( w )
return true
2021-09-28 21:47:45 +02:00
case "datadog/api/v1/series" :
datadogWriteRequests . Inc ( )
if err := datadog . InsertHandlerForHTTP ( at , r ) ; err != nil {
datadogWriteErrors . Inc ( )
httpserver . Errorf ( w , r , "%s" , err )
return true
}
// See https://docs.datadoghq.com/api/latest/metrics/#submit-metrics
w . WriteHeader ( 202 )
fmt . Fprintf ( w , ` { "status":"ok"} ` )
return true
case "datadog/api/v1/validate" :
datadogValidateRequests . Inc ( )
// See https://docs.datadoghq.com/api/latest/authentication/#validate-api-key
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
fmt . Fprintf ( w , ` { "valid":true} ` )
return true
case "datadog/api/v1/check_run" :
datadogCheckRunRequests . Inc ( )
// See https://docs.datadoghq.com/api/latest/service-checks/#submit-a-service-check
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
w . WriteHeader ( 202 )
fmt . Fprintf ( w , ` { "status":"ok"} ` )
return true
2022-06-07 14:16:48 +02:00
case "datadog/intake" :
2021-09-28 21:47:45 +02:00
datadogIntakeRequests . Inc ( )
2021-11-09 17:03:50 +01:00
w . Header ( ) . Set ( "Content-Type" , "application/json" )
2021-09-28 21:47:45 +02:00
fmt . Fprintf ( w , ` { } ` )
return true
2022-06-13 08:54:11 +02:00
case "datadog/api/v1/metadata" :
datadogMetadataRequests . Inc ( )
w . Header ( ) . Set ( "Content-Type" , "application/json" )
fmt . Fprintf ( w , ` { } ` )
return true
2021-08-05 08:46:19 +02:00
default :
httpserver . Errorf ( w , r , "unsupported multitenant path suffix: %q" , p . Suffix )
return true
}
}
2020-02-23 12:35:47 +01:00
var (
2020-03-10 18:35:58 +01:00
prometheusWriteRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/write", protocol="promremotewrite"} ` )
prometheusWriteErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/api/v1/write", protocol="promremotewrite"} ` )
vmimportRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/import", protocol="vmimport"} ` )
vmimportErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/api/v1/import", protocol="vmimport"} ` )
2020-02-23 12:35:47 +01:00
2020-03-10 18:35:58 +01:00
csvimportRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/import/csv", protocol="csvimport"} ` )
csvimportErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/api/v1/import/csv", protocol="csvimport"} ` )
2020-02-23 12:35:47 +01:00
2020-07-10 11:00:35 +02:00
prometheusimportRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/import/prometheus", protocol="prometheusimport"} ` )
prometheusimportErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/api/v1/import/prometheus", protocol="prometheusimport"} ` )
2020-09-26 03:29:45 +02:00
nativeimportRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/import/native", protocol="nativeimport"} ` )
nativeimportErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/api/v1/import/native", protocol="nativeimport"} ` )
2021-09-28 21:47:45 +02:00
influxWriteRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/influx/write", protocol="influx"} ` )
influxWriteErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/influx/write", protocol="influx"} ` )
influxQueryRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/influx/query", protocol="influx"} ` )
datadogWriteRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/datadog/api/v1/series", protocol="datadog"} ` )
datadogWriteErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/datadog/api/v1/series", protocol="datadog"} ` )
2020-02-23 12:35:47 +01:00
2021-09-28 21:47:45 +02:00
datadogValidateRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/datadog/api/v1/validate", protocol="datadog"} ` )
datadogCheckRunRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/datadog/api/v1/check_run", protocol="datadog"} ` )
2022-06-07 14:16:48 +02:00
datadogIntakeRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/datadog/intake", protocol="datadog"} ` )
2022-06-13 08:52:13 +02:00
datadogMetadataRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/datadog/api/v1/metadata", protocol="datadog"} ` )
2020-02-25 17:13:11 +01:00
2022-06-06 23:57:05 +02:00
promscrapeTargetsRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/targets"} ` )
promscrapeServiceDiscoveryRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/service-discovery"} ` )
promscrapeAPIV1TargetsRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/targets"} ` )
2020-04-30 01:00:32 +02:00
2022-02-03 17:57:36 +01:00
promscrapeTargetResponseRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/target_response"} ` )
promscrapeTargetResponseErrors = metrics . NewCounter ( ` vmagent_http_request_errors_total { path="/target_response"} ` )
2022-05-25 08:57:11 +02:00
promscrapeConfigRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/config"} ` )
promscrapeStatusConfigRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/api/v1/status/config"} ` )
2021-10-12 15:23:42 +02:00
2020-04-30 01:00:32 +02:00
promscrapeConfigReloadRequests = metrics . NewCounter ( ` vmagent_http_requests_total { path="/-/reload"} ` )
2020-02-23 12:35:47 +01:00
)
2020-06-05 09:37:36 +02:00
func usage ( ) {
const s = `
vmagent collects metrics data via popular data ingestion protocols and routes it to VictoriaMetrics .
2021-04-20 19:16:17 +02:00
See the docs at https : //docs.victoriametrics.com/vmagent.html .
2020-06-05 09:37:36 +02:00
`
2020-12-03 20:40:30 +01:00
flagutil . Usage ( s )
2020-06-05 09:37:36 +02:00
}