2021-03-05 10:36:05 +00:00
package kubernetes
import (
"encoding/json"
"errors"
2021-03-11 14:41:09 +00:00
"flag"
2021-03-05 10:36:05 +00:00
"fmt"
"io"
"io/ioutil"
"net/http"
"net/url"
"reflect"
"strconv"
"strings"
"sync"
"time"
"github.com/VictoriaMetrics/VictoriaMetrics/lib/logger"
2021-03-11 14:41:09 +00:00
"github.com/VictoriaMetrics/VictoriaMetrics/lib/promauth"
2021-03-05 10:36:05 +00:00
"github.com/VictoriaMetrics/metrics"
)
2021-03-11 14:41:09 +00:00
var apiServerTimeout = flag . Duration ( "promscrape.kubernetes.apiServerTimeout" , 30 * time . Minute , "How frequently to reload the full state from Kuberntes API server" )
2021-03-05 10:36:05 +00:00
// WatchEvent is a watch event returned from API server endpoints if `watch=1` query arg is set.
//
// See https://kubernetes.io/docs/reference/using-api/api-concepts/#efficient-detection-of-changes
type WatchEvent struct {
Type string
Object json . RawMessage
}
// object is any Kubernetes object.
type object interface {
key ( ) string
2021-03-14 19:10:35 +00:00
getTargetLabels ( gw * groupWatcher ) [ ] map [ string ] string
2021-03-05 10:36:05 +00:00
}
// parseObjectFunc must parse object from the given data.
type parseObjectFunc func ( data [ ] byte ) ( object , error )
2021-03-11 14:41:09 +00:00
// parseObjectListFunc must parse objectList from the given r.
type parseObjectListFunc func ( r io . Reader ) ( map [ string ] object , ListMeta , error )
2021-03-05 10:36:05 +00:00
// apiWatcher is used for watching for Kuberntes object changes and caching their latest states.
type apiWatcher struct {
2021-03-14 19:10:35 +00:00
role string
2021-03-05 10:36:05 +00:00
2021-03-11 14:41:09 +00:00
// Constructor for creating ScrapeWork objects from labels
2021-03-05 10:36:05 +00:00
swcFunc ScrapeWorkConstructorFunc
2021-03-14 19:10:35 +00:00
gw * groupWatcher
2021-03-11 14:41:09 +00:00
// swos contains a map of ScrapeWork objects for the given apiWatcher
swosByKey map [ string ] [ ] interface { }
swosByKeyLock sync . Mutex
2021-03-05 10:36:05 +00:00
2021-03-14 19:10:35 +00:00
swosCount * metrics . Counter
2021-03-05 10:36:05 +00:00
}
2021-03-11 14:41:09 +00:00
func newAPIWatcher ( apiServer string , ac * promauth . Config , sdc * SDConfig , swcFunc ScrapeWorkConstructorFunc ) * apiWatcher {
2021-03-14 19:10:35 +00:00
namespaces := sdc . Namespaces . Names
selectors := sdc . Selectors
proxyURL := sdc . ProxyURL . URL ( )
gw := getGroupWatcher ( apiServer , ac , namespaces , selectors , proxyURL )
2021-03-05 10:36:05 +00:00
return & apiWatcher {
2021-03-14 19:10:35 +00:00
role : sdc . Role ,
2021-03-11 14:41:09 +00:00
swcFunc : swcFunc ,
2021-03-14 19:10:35 +00:00
gw : gw ,
swosByKey : make ( map [ string ] [ ] interface { } ) ,
swosCount : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_scrape_works { role=%q} ` , sdc . Role ) ) ,
2021-03-05 10:36:05 +00:00
}
}
2021-03-11 14:41:09 +00:00
func ( aw * apiWatcher ) mustStop ( ) {
2021-03-14 19:10:35 +00:00
aw . gw . unsubscribeAPIWatcher ( aw )
aw . reloadScrapeWorks ( make ( map [ string ] [ ] interface { } ) )
2021-03-11 14:41:09 +00:00
}
2021-03-14 19:10:35 +00:00
func ( aw * apiWatcher ) reloadScrapeWorks ( swosByKey map [ string ] [ ] interface { } ) {
2021-03-11 14:41:09 +00:00
aw . swosByKeyLock . Lock ( )
2021-03-14 19:10:35 +00:00
aw . swosCount . Add ( len ( swosByKey ) - len ( aw . swosByKey ) )
2021-03-11 14:41:09 +00:00
aw . swosByKey = swosByKey
aw . swosByKeyLock . Unlock ( )
}
func ( aw * apiWatcher ) setScrapeWorks ( key string , labels [ ] map [ string ] string ) {
swos := getScrapeWorkObjectsForLabels ( aw . swcFunc , labels )
aw . swosByKeyLock . Lock ( )
if len ( swos ) > 0 {
2021-03-14 19:10:35 +00:00
aw . swosCount . Add ( len ( swos ) - len ( aw . swosByKey [ key ] ) )
2021-03-11 14:41:09 +00:00
aw . swosByKey [ key ] = swos
} else {
2021-03-14 19:10:35 +00:00
aw . swosCount . Add ( - len ( aw . swosByKey [ key ] ) )
2021-03-11 14:41:09 +00:00
delete ( aw . swosByKey , key )
}
aw . swosByKeyLock . Unlock ( )
}
func ( aw * apiWatcher ) removeScrapeWorks ( key string ) {
aw . swosByKeyLock . Lock ( )
2021-03-14 19:10:35 +00:00
aw . swosCount . Add ( - len ( aw . swosByKey [ key ] ) )
2021-03-11 14:41:09 +00:00
delete ( aw . swosByKey , key )
aw . swosByKeyLock . Unlock ( )
}
func getScrapeWorkObjectsForLabels ( swcFunc ScrapeWorkConstructorFunc , labelss [ ] map [ string ] string ) [ ] interface { } {
swos := make ( [ ] interface { } , 0 , len ( labelss ) )
for _ , labels := range labelss {
swo := swcFunc ( labels )
// The reflect check is needed because of https://mangatmodi.medium.com/go-check-nil-interface-the-right-way-d142776edef1
if swo != nil && ! reflect . ValueOf ( swo ) . IsNil ( ) {
swos = append ( swos , swo )
2021-03-05 10:36:05 +00:00
}
}
2021-03-11 14:41:09 +00:00
return swos
}
// getScrapeWorkObjects returns all the ScrapeWork objects for the given aw.
func ( aw * apiWatcher ) getScrapeWorkObjects ( ) [ ] interface { } {
2021-03-14 19:10:35 +00:00
aw . gw . startWatchersForRole ( aw . role , aw )
2021-03-11 14:41:09 +00:00
aw . swosByKeyLock . Lock ( )
defer aw . swosByKeyLock . Unlock ( )
size := 0
for _ , swosLocal := range aw . swosByKey {
size += len ( swosLocal )
}
swos := make ( [ ] interface { } , 0 , size )
for _ , swosLocal := range aw . swosByKey {
swos = append ( swos , swosLocal ... )
}
2021-03-05 10:36:05 +00:00
return swos
}
2021-03-14 19:10:35 +00:00
// groupWatcher watches for Kubernetes objects on the given apiServer with the given namespaces,
// selectors and authorization using the given client.
type groupWatcher struct {
apiServer string
namespaces [ ] string
selectors [ ] Selector
authorization string
client * http . Client
mu sync . Mutex
m map [ string ] * urlWatcher
}
func newGroupWatcher ( apiServer string , ac * promauth . Config , namespaces [ ] string , selectors [ ] Selector , proxyURL * url . URL ) * groupWatcher {
var proxy func ( * http . Request ) ( * url . URL , error )
if proxyURL != nil {
proxy = http . ProxyURL ( proxyURL )
}
client := & http . Client {
Transport : & http . Transport {
TLSClientConfig : ac . NewTLSConfig ( ) ,
Proxy : proxy ,
TLSHandshakeTimeout : 10 * time . Second ,
IdleConnTimeout : * apiServerTimeout ,
} ,
Timeout : * apiServerTimeout ,
}
return & groupWatcher {
apiServer : apiServer ,
authorization : ac . Authorization ,
namespaces : namespaces ,
selectors : selectors ,
client : client ,
m : make ( map [ string ] * urlWatcher ) ,
}
}
func getGroupWatcher ( apiServer string , ac * promauth . Config , namespaces [ ] string , selectors [ ] Selector , proxyURL * url . URL ) * groupWatcher {
key := fmt . Sprintf ( "apiServer=%s, namespaces=%s, selectors=%s, proxyURL=%v, authConfig=%s" ,
apiServer , namespaces , selectorsKey ( selectors ) , proxyURL , ac . String ( ) )
groupWatchersLock . Lock ( )
gw := groupWatchers [ key ]
if gw == nil {
gw = newGroupWatcher ( apiServer , ac , namespaces , selectors , proxyURL )
groupWatchers [ key ] = gw
}
groupWatchersLock . Unlock ( )
return gw
}
func selectorsKey ( selectors [ ] Selector ) string {
var sb strings . Builder
for _ , s := range selectors {
fmt . Fprintf ( & sb , "{role=%q, label=%q, field=%q}" , s . Role , s . Label , s . Field )
}
return sb . String ( )
}
var (
groupWatchersLock sync . Mutex
groupWatchers = make ( map [ string ] * groupWatcher )
_ = metrics . NewGauge ( ` vm_promscrape_discovery_kubernetes_group_watchers ` , func ( ) float64 {
groupWatchersLock . Lock ( )
n := len ( groupWatchers )
groupWatchersLock . Unlock ( )
return float64 ( n )
} )
)
2021-03-05 10:36:05 +00:00
// getObjectByRole returns an object with the given (namespace, name) key and the given role.
2021-03-14 19:10:35 +00:00
func ( gw * groupWatcher ) getObjectByRole ( role , namespace , name string ) object {
if gw == nil {
2021-03-11 14:41:09 +00:00
// this is needed for testing
2021-03-05 10:36:05 +00:00
return nil
}
key := namespace + "/" + name
2021-03-14 19:10:35 +00:00
gw . startWatchersForRole ( role , nil )
gw . mu . Lock ( )
defer gw . mu . Unlock ( )
2021-03-11 14:41:09 +00:00
2021-03-14 19:10:35 +00:00
for _ , uw := range gw . m {
2021-03-05 10:36:05 +00:00
if uw . role != role {
continue
}
2021-03-11 14:41:09 +00:00
uw . mu . Lock ( )
o := uw . objectsByKey [ key ]
uw . mu . Unlock ( )
2021-03-05 10:36:05 +00:00
if o != nil {
2021-03-11 14:41:09 +00:00
return o
2021-03-05 10:36:05 +00:00
}
}
2021-03-11 14:41:09 +00:00
return nil
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
func ( gw * groupWatcher ) startWatchersForRole ( role string , aw * apiWatcher ) {
paths := getAPIPaths ( role , gw . namespaces , gw . selectors )
2021-03-05 10:36:05 +00:00
for _ , path := range paths {
2021-03-14 19:10:35 +00:00
apiURL := gw . apiServer + path
gw . mu . Lock ( )
uw := gw . m [ apiURL ]
if uw == nil {
uw = newURLWatcher ( role , apiURL , gw )
gw . m [ apiURL ] = uw
}
gw . mu . Unlock ( )
uw . subscribeAPIWatcher ( aw )
2021-03-05 10:36:05 +00:00
}
}
2021-03-14 19:10:35 +00:00
func ( gw * groupWatcher ) reloadScrapeWorksForAPIWatchers ( aws [ ] * apiWatcher , objectsByKey map [ string ] object ) {
if len ( aws ) == 0 {
2021-03-05 10:36:05 +00:00
return
}
2021-03-14 19:10:35 +00:00
swosByKey := make ( [ ] map [ string ] [ ] interface { } , len ( aws ) )
for i := range aws {
swosByKey [ i ] = make ( map [ string ] [ ] interface { } )
2021-03-13 13:18:47 +00:00
}
2021-03-14 19:10:35 +00:00
for key , o := range objectsByKey {
labels := o . getTargetLabels ( gw )
for i , aw := range aws {
swos := getScrapeWorkObjectsForLabels ( aw . swcFunc , labels )
if len ( swos ) > 0 {
swosByKey [ i ] [ key ] = swos
}
2021-03-13 13:18:47 +00:00
}
2021-03-14 19:10:35 +00:00
}
for i , aw := range aws {
aw . reloadScrapeWorks ( swosByKey [ i ] )
}
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
// doRequest performs http request to the given requestURL.
func ( gw * groupWatcher ) doRequest ( requestURL string ) ( * http . Response , error ) {
req , err := http . NewRequest ( "GET" , requestURL , nil )
if err != nil {
logger . Fatalf ( "cannot create a request for %q: %s" , requestURL , err )
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
if gw . authorization != "" {
req . Header . Set ( "Authorization" , gw . authorization )
}
return gw . client . Do ( req )
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
func ( gw * groupWatcher ) unsubscribeAPIWatcher ( aw * apiWatcher ) {
gw . mu . Lock ( )
for _ , uw := range gw . m {
uw . unsubscribeAPIWatcher ( aw )
}
gw . mu . Unlock ( )
}
2021-03-05 10:36:05 +00:00
// urlWatcher watches for an apiURL and updates object states in objectsByKey.
type urlWatcher struct {
2021-03-14 19:10:35 +00:00
role string
apiURL string
gw * groupWatcher
2021-03-05 10:36:05 +00:00
parseObject parseObjectFunc
parseObjectList parseObjectListFunc
2021-03-14 19:10:35 +00:00
// mu protects aws, awsPending, objectsByKey and resourceVersion
2021-03-11 14:41:09 +00:00
mu sync . Mutex
// aws contains registered apiWatcher objects
aws map [ * apiWatcher ] struct { }
2021-03-14 19:10:35 +00:00
// awsPending contains pending apiWatcher objects, which must be moved to aws in a batch
awsPending map [ * apiWatcher ] struct { }
2021-03-05 10:36:05 +00:00
// objectsByKey contains the latest state for objects obtained from apiURL
2021-03-11 14:41:09 +00:00
objectsByKey map [ string ] object
2021-03-05 15:29:54 +00:00
2021-03-07 17:50:01 +00:00
resourceVersion string
2021-03-05 10:36:05 +00:00
2021-03-26 10:28:10 +00:00
objectsCount * metrics . Counter
objectsAdded * metrics . Counter
objectsRemoved * metrics . Counter
objectsUpdated * metrics . Counter
staleResourceVersions * metrics . Counter
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
func newURLWatcher ( role , apiURL string , gw * groupWatcher ) * urlWatcher {
2021-03-11 14:41:09 +00:00
parseObject , parseObjectList := getObjectParsersForRole ( role )
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_url_watchers { role=%q} ` , role ) ) . Inc ( )
uw := & urlWatcher {
2021-03-14 19:10:35 +00:00
role : role ,
apiURL : apiURL ,
gw : gw ,
2021-03-05 10:36:05 +00:00
parseObject : parseObject ,
parseObjectList : parseObjectList ,
2021-03-11 14:41:09 +00:00
aws : make ( map [ * apiWatcher ] struct { } ) ,
2021-03-14 19:10:35 +00:00
awsPending : make ( map [ * apiWatcher ] struct { } ) ,
2021-03-11 14:41:09 +00:00
objectsByKey : make ( map [ string ] object ) ,
2021-03-05 10:36:05 +00:00
2021-03-26 10:28:10 +00:00
objectsCount : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_objects { role=%q} ` , role ) ) ,
objectsAdded : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_objects_added_total { role=%q} ` , role ) ) ,
objectsRemoved : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_objects_removed_total { role=%q} ` , role ) ) ,
objectsUpdated : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_objects_updated_total { role=%q} ` , role ) ) ,
staleResourceVersions : metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_stale_resource_versions_total { role=%q} ` , role ) ) ,
2021-03-11 14:41:09 +00:00
}
2021-03-14 19:10:35 +00:00
logger . Infof ( "started %s watcher for %q" , uw . role , uw . apiURL )
go uw . watchForUpdates ( )
go uw . processPendingSubscribers ( )
2021-03-11 14:41:09 +00:00
return uw
}
2021-03-05 10:36:05 +00:00
2021-03-14 19:10:35 +00:00
func ( uw * urlWatcher ) subscribeAPIWatcher ( aw * apiWatcher ) {
if aw == nil {
return
}
2021-03-11 14:41:09 +00:00
uw . mu . Lock ( )
2021-03-14 19:10:35 +00:00
if _ , ok := uw . aws [ aw ] ; ! ok {
if _ , ok := uw . awsPending [ aw ] ; ! ok {
uw . awsPending [ aw ] = struct { } { }
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_subscibers { role=%q,type="pending"} ` , uw . role ) ) . Inc ( )
}
2021-03-05 10:36:05 +00:00
}
2021-03-11 14:41:09 +00:00
uw . mu . Unlock ( )
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:10:35 +00:00
func ( uw * urlWatcher ) unsubscribeAPIWatcher ( aw * apiWatcher ) {
2021-03-11 14:41:09 +00:00
uw . mu . Lock ( )
2021-03-14 19:10:35 +00:00
if _ , ok := uw . aws [ aw ] ; ok {
delete ( uw . aws , aw )
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_subscibers { role=%q,type="permanent"} ` , uw . role ) ) . Dec ( )
} else if _ , ok := uw . awsPending [ aw ] ; ok {
delete ( uw . awsPending , aw )
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_subscibers { role=%q,type="pending"} ` , uw . role ) ) . Dec ( )
2021-03-11 14:41:09 +00:00
}
uw . mu . Unlock ( )
}
2021-03-14 19:10:35 +00:00
func ( uw * urlWatcher ) processPendingSubscribers ( ) {
t := time . NewTicker ( time . Second )
for range t . C {
var awsPending [ ] * apiWatcher
var objectsByKey map [ string ] object
uw . mu . Lock ( )
if len ( uw . awsPending ) > 0 {
awsPending = getAPIWatchers ( uw . awsPending )
for _ , aw := range awsPending {
if _ , ok := uw . aws [ aw ] ; ok {
logger . Panicf ( "BUG: aw=%p already exists in uw.aws" , aw )
}
uw . aws [ aw ] = struct { } { }
delete ( uw . awsPending , aw )
}
objectsByKey = make ( map [ string ] object , len ( uw . objectsByKey ) )
for key , o := range uw . objectsByKey {
objectsByKey [ key ] = o
}
}
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_subscibers { role=%q,type="pending"} ` , uw . role ) ) . Add ( - len ( awsPending ) )
metrics . GetOrCreateCounter ( fmt . Sprintf ( ` vm_promscrape_discovery_kubernetes_subscibers { role=%q,type="permanent"} ` , uw . role ) ) . Add ( len ( awsPending ) )
uw . mu . Unlock ( )
uw . gw . reloadScrapeWorksForAPIWatchers ( awsPending , objectsByKey )
2021-03-11 14:41:09 +00:00
}
2021-03-07 17:50:01 +00:00
}
2021-03-10 13:06:33 +00:00
func ( uw * urlWatcher ) setResourceVersion ( resourceVersion string ) {
2021-03-07 17:50:01 +00:00
uw . mu . Lock ( )
2021-03-10 13:06:33 +00:00
uw . resourceVersion = resourceVersion
2021-03-07 17:50:01 +00:00
uw . mu . Unlock ( )
}
2021-03-05 10:36:05 +00:00
// reloadObjects reloads objects to the latest state and returns resourceVersion for the latest state.
func ( uw * urlWatcher ) reloadObjects ( ) string {
2021-03-07 17:50:01 +00:00
uw . mu . Lock ( )
resourceVersion := uw . resourceVersion
uw . mu . Unlock ( )
if resourceVersion != "" {
2021-03-11 14:41:09 +00:00
// Fast path - there is no need in reloading the objects.
2021-03-07 17:50:01 +00:00
return resourceVersion
}
2021-03-05 10:36:05 +00:00
requestURL := uw . apiURL
2021-03-14 19:10:35 +00:00
resp , err := uw . gw . doRequest ( requestURL )
2021-03-05 10:36:05 +00:00
if err != nil {
2021-03-11 14:41:09 +00:00
logger . Errorf ( "cannot perform request to %q: %s" , requestURL , err )
2021-03-05 10:36:05 +00:00
return ""
}
if resp . StatusCode != http . StatusOK {
2021-03-11 14:41:09 +00:00
body , _ := ioutil . ReadAll ( resp . Body )
_ = resp . Body . Close ( )
2021-03-05 10:36:05 +00:00
logger . Errorf ( "unexpected status code for request to %q: %d; want %d; response: %q" , requestURL , resp . StatusCode , http . StatusOK , body )
return ""
}
2021-03-11 14:41:09 +00:00
objectsByKey , metadata , err := uw . parseObjectList ( resp . Body )
_ = resp . Body . Close ( )
2021-03-05 10:36:05 +00:00
if err != nil {
2021-03-11 14:41:09 +00:00
logger . Errorf ( "cannot parse objects from %q: %s" , requestURL , err )
2021-03-05 10:36:05 +00:00
return ""
}
2021-03-11 14:41:09 +00:00
uw . mu . Lock ( )
var updated , removed , added int
for key := range uw . objectsByKey {
if o , ok := objectsByKey [ key ] ; ok {
uw . objectsByKey [ key ] = o
updated ++
} else {
delete ( uw . objectsByKey , key )
removed ++
2021-03-05 10:36:05 +00:00
}
}
2021-03-11 14:41:09 +00:00
for key , o := range objectsByKey {
if _ , ok := uw . objectsByKey [ key ] ; ! ok {
uw . objectsByKey [ key ] = o
added ++
}
}
uw . objectsUpdated . Add ( updated )
uw . objectsRemoved . Add ( removed )
uw . objectsAdded . Add ( added )
uw . objectsCount . Add ( added - removed )
2021-03-07 17:50:01 +00:00
uw . resourceVersion = metadata . ResourceVersion
2021-03-14 19:10:35 +00:00
aws := getAPIWatchers ( uw . aws )
2021-03-05 10:36:05 +00:00
uw . mu . Unlock ( )
2021-03-07 17:50:01 +00:00
2021-03-14 19:10:35 +00:00
uw . gw . reloadScrapeWorksForAPIWatchers ( aws , objectsByKey )
logger . Infof ( "reloaded %d objects from %q" , len ( objectsByKey ) , requestURL )
2021-03-05 10:36:05 +00:00
return metadata . ResourceVersion
}
2021-03-14 19:10:35 +00:00
func getAPIWatchers ( awsMap map [ * apiWatcher ] struct { } ) [ ] * apiWatcher {
aws := make ( [ ] * apiWatcher , 0 , len ( awsMap ) )
for aw := range awsMap {
2021-03-11 14:41:09 +00:00
aws = append ( aws , aw )
2021-03-05 10:36:05 +00:00
}
2021-03-11 14:41:09 +00:00
return aws
2021-03-05 10:36:05 +00:00
}
2021-03-07 17:50:01 +00:00
// watchForUpdates watches for object updates starting from uw.resourceVersion and updates the corresponding objects to the latest state.
2021-03-05 10:36:05 +00:00
//
// See https://kubernetes.io/docs/reference/using-api/api-concepts/#efficient-detection-of-changes
2021-03-07 17:50:01 +00:00
func ( uw * urlWatcher ) watchForUpdates ( ) {
2021-03-05 10:36:05 +00:00
backoffDelay := time . Second
maxBackoffDelay := 30 * time . Second
backoffSleep := func ( ) {
time . Sleep ( backoffDelay )
backoffDelay *= 2
if backoffDelay > maxBackoffDelay {
backoffDelay = maxBackoffDelay
}
}
apiURL := uw . apiURL
delimiter := "?"
if strings . Contains ( apiURL , "?" ) {
delimiter = "&"
}
2021-03-14 19:10:35 +00:00
timeoutSeconds := time . Duration ( 0.9 * float64 ( uw . gw . client . Timeout ) ) . Seconds ( )
2021-03-10 13:06:33 +00:00
apiURL += delimiter + "watch=1&allowWatchBookmarks=true&timeoutSeconds=" + strconv . Itoa ( int ( timeoutSeconds ) )
2021-03-05 10:36:05 +00:00
for {
2021-03-07 17:50:01 +00:00
resourceVersion := uw . reloadObjects ( )
2021-03-14 19:55:00 +00:00
if resourceVersion == "" {
backoffSleep ( )
continue
2021-03-05 10:36:05 +00:00
}
2021-03-14 19:55:00 +00:00
requestURL := apiURL + "&resourceVersion=" + url . QueryEscape ( resourceVersion )
2021-03-14 19:10:35 +00:00
resp , err := uw . gw . doRequest ( requestURL )
2021-03-05 10:36:05 +00:00
if err != nil {
2021-03-13 13:18:47 +00:00
logger . Errorf ( "cannot perform request to %q: %s" , requestURL , err )
2021-03-05 10:36:05 +00:00
backoffSleep ( )
continue
}
if resp . StatusCode != http . StatusOK {
if resp . StatusCode == 410 {
// There is no need for sleep on 410 error. See https://kubernetes.io/docs/reference/using-api/api-concepts/#410-gone-responses
backoffDelay = time . Second
2021-03-26 10:28:10 +00:00
uw . staleResourceVersions . Inc ( )
2021-03-10 13:06:33 +00:00
uw . setResourceVersion ( "" )
2021-03-05 10:36:05 +00:00
} else {
2021-03-26 10:28:10 +00:00
body , _ := ioutil . ReadAll ( resp . Body )
_ = resp . Body . Close ( )
logger . Errorf ( "unexpected status code for request to %q: %d; want %d; response: %q" , requestURL , resp . StatusCode , http . StatusOK , body )
2021-03-05 10:36:05 +00:00
backoffSleep ( )
}
continue
}
backoffDelay = time . Second
err = uw . readObjectUpdateStream ( resp . Body )
_ = resp . Body . Close ( )
if err != nil {
if ! errors . Is ( err , io . EOF ) {
logger . Errorf ( "error when reading WatchEvent stream from %q: %s" , requestURL , err )
}
backoffSleep ( )
continue
}
}
}
// readObjectUpdateStream reads Kuberntes watch events from r and updates locally cached objects according to the received events.
func ( uw * urlWatcher ) readObjectUpdateStream ( r io . Reader ) error {
d := json . NewDecoder ( r )
var we WatchEvent
for {
if err := d . Decode ( & we ) ; err != nil {
return err
}
switch we . Type {
case "ADDED" , "MODIFIED" :
2021-03-11 11:06:40 +00:00
o , err := uw . parseObject ( we . Object )
if err != nil {
return err
}
key := o . key ( )
2021-03-05 10:36:05 +00:00
uw . mu . Lock ( )
2021-03-11 14:41:09 +00:00
if _ , ok := uw . objectsByKey [ key ] ; ! ok {
uw . objectsCount . Inc ( )
uw . objectsAdded . Inc ( )
2021-03-05 10:36:05 +00:00
} else {
2021-03-11 14:41:09 +00:00
uw . objectsUpdated . Inc ( )
2021-03-05 10:36:05 +00:00
}
2021-03-11 14:41:09 +00:00
uw . objectsByKey [ key ] = o
2021-03-14 19:10:35 +00:00
aws := getAPIWatchers ( uw . aws )
2021-03-05 10:36:05 +00:00
uw . mu . Unlock ( )
2021-03-14 19:10:35 +00:00
labels := o . getTargetLabels ( uw . gw )
for _ , aw := range aws {
2021-03-11 14:41:09 +00:00
aw . setScrapeWorks ( key , labels )
}
2021-03-05 10:36:05 +00:00
case "DELETED" :
2021-03-11 11:06:40 +00:00
o , err := uw . parseObject ( we . Object )
if err != nil {
return err
}
key := o . key ( )
2021-03-05 10:36:05 +00:00
uw . mu . Lock ( )
2021-03-11 14:41:09 +00:00
if _ , ok := uw . objectsByKey [ key ] ; ok {
uw . objectsCount . Dec ( )
uw . objectsRemoved . Inc ( )
delete ( uw . objectsByKey , key )
}
2021-03-14 19:10:35 +00:00
aws := getAPIWatchers ( uw . aws )
2021-03-05 10:36:05 +00:00
uw . mu . Unlock ( )
2021-03-14 19:10:35 +00:00
for _ , aw := range aws {
2021-03-11 14:41:09 +00:00
aw . removeScrapeWorks ( key )
}
2021-03-10 13:06:33 +00:00
case "BOOKMARK" :
// See https://kubernetes.io/docs/reference/using-api/api-concepts/#watch-bookmarks
2021-03-11 11:06:40 +00:00
bm , err := parseBookmark ( we . Object )
if err != nil {
return fmt . Errorf ( "cannot parse bookmark from %q: %w" , we . Object , err )
}
uw . setResourceVersion ( bm . Metadata . ResourceVersion )
2021-03-26 10:28:10 +00:00
case "ERROR" :
em , err := parseError ( we . Object )
if err != nil {
2021-03-26 10:45:59 +00:00
return fmt . Errorf ( "cannot parse error message from %q: %w" , we . Object , err )
2021-03-26 10:28:10 +00:00
}
if em . Code == 410 {
// See https://kubernetes.io/docs/reference/using-api/api-concepts/#410-gone-responses
uw . staleResourceVersions . Inc ( )
uw . setResourceVersion ( "" )
return nil
}
return fmt . Errorf ( "unexpected error message: %q" , we . Object )
2021-03-05 10:36:05 +00:00
default :
2021-03-26 10:28:10 +00:00
return fmt . Errorf ( "unexpected WatchEvent type %q: %q" , we . Type , we . Object )
2021-03-05 10:36:05 +00:00
}
}
}
2021-03-26 10:28:10 +00:00
// Bookmark is a bookmark message from Kubernetes Watch API.
2021-03-11 11:06:40 +00:00
// See https://kubernetes.io/docs/reference/using-api/api-concepts/#watch-bookmarks
type Bookmark struct {
Metadata struct {
ResourceVersion string
}
}
func parseBookmark ( data [ ] byte ) ( * Bookmark , error ) {
var bm Bookmark
if err := json . Unmarshal ( data , & bm ) ; err != nil {
return nil , err
}
return & bm , nil
}
2021-03-26 10:28:10 +00:00
// Error is an error message from Kubernetes Watch API.
type Error struct {
Code int
}
func parseError ( data [ ] byte ) ( * Error , error ) {
var em Error
if err := json . Unmarshal ( data , & em ) ; err != nil {
return nil , err
}
return & em , nil
}
2021-03-05 10:36:05 +00:00
func getAPIPaths ( role string , namespaces [ ] string , selectors [ ] Selector ) [ ] string {
objectName := getObjectNameByRole ( role )
if objectName == "nodes" || len ( namespaces ) == 0 {
query := joinSelectors ( role , selectors )
path := getAPIPath ( objectName , "" , query )
return [ ] string { path }
}
query := joinSelectors ( role , selectors )
paths := make ( [ ] string , len ( namespaces ) )
for i , namespace := range namespaces {
paths [ i ] = getAPIPath ( objectName , namespace , query )
}
return paths
}
func getAPIPath ( objectName , namespace , query string ) string {
suffix := objectName
if namespace != "" {
suffix = "namespaces/" + namespace + "/" + objectName
}
if len ( query ) > 0 {
suffix += "?" + query
}
if objectName == "ingresses" {
return "/apis/networking.k8s.io/v1beta1/" + suffix
}
if objectName == "endpointslices" {
return "/apis/discovery.k8s.io/v1beta1/" + suffix
}
return "/api/v1/" + suffix
}
func joinSelectors ( role string , selectors [ ] Selector ) string {
var labelSelectors , fieldSelectors [ ] string
for _ , s := range selectors {
if s . Role != role {
continue
}
if s . Label != "" {
labelSelectors = append ( labelSelectors , s . Label )
}
if s . Field != "" {
fieldSelectors = append ( fieldSelectors , s . Field )
}
}
var args [ ] string
if len ( labelSelectors ) > 0 {
args = append ( args , "labelSelector=" + url . QueryEscape ( strings . Join ( labelSelectors , "," ) ) )
}
if len ( fieldSelectors ) > 0 {
args = append ( args , "fieldSelector=" + url . QueryEscape ( strings . Join ( fieldSelectors , "," ) ) )
}
return strings . Join ( args , "&" )
}
func getObjectNameByRole ( role string ) string {
switch role {
case "node" :
return "nodes"
case "pod" :
return "pods"
case "service" :
return "services"
case "endpoints" :
return "endpoints"
case "endpointslices" :
return "endpointslices"
case "ingress" :
return "ingresses"
default :
logger . Panicf ( "BUG: unknonw role=%q" , role )
return ""
}
}
func getObjectParsersForRole ( role string ) ( parseObjectFunc , parseObjectListFunc ) {
switch role {
case "node" :
return parseNode , parseNodeList
case "pod" :
return parsePod , parsePodList
case "service" :
return parseService , parseServiceList
case "endpoints" :
return parseEndpoints , parseEndpointsList
case "endpointslices" :
return parseEndpointSlice , parseEndpointSliceList
case "ingress" :
return parseIngress , parseIngressList
default :
logger . Panicf ( "BUG: unsupported role=%q" , role )
return nil , nil
}
}