mirror of
				https://github.com/prometheus/prometheus.git
				synced 2025-11-04 02:11:01 +01:00 
			
		
		
		
	This doesn't make the import order consistend everywhere, just where it was touched by the previous commit. Change-Id: I82fc75f8691da9901c7ceb808e6f6fe8e5d62c0e
		
			
				
	
	
		
			73 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			73 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2013 Prometheus Team
 | 
						|
// 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,
 | 
						|
// 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 api
 | 
						|
 | 
						|
import (
 | 
						|
	"encoding/json"
 | 
						|
	"net/http"
 | 
						|
 | 
						|
	clientmodel "github.com/prometheus/client_golang/model"
 | 
						|
 | 
						|
	"github.com/prometheus/prometheus/retrieval"
 | 
						|
	"github.com/prometheus/prometheus/web/httputils"
 | 
						|
)
 | 
						|
 | 
						|
// TargetGroup bundles endpoints and base labels with appropriate JSON
 | 
						|
// annotations.
 | 
						|
type TargetGroup struct {
 | 
						|
	Endpoints  []string          `json:"endpoints"`
 | 
						|
	BaseLabels map[string]string `json:"baseLabels"`
 | 
						|
}
 | 
						|
 | 
						|
// SetTargets handles the /api/targets endpoint.
 | 
						|
func (serv MetricsService) SetTargets(w http.ResponseWriter, r *http.Request) {
 | 
						|
	params := httputils.GetQueryParams(r)
 | 
						|
	jobName := params.Get("job")
 | 
						|
 | 
						|
	decoder := json.NewDecoder(r.Body)
 | 
						|
	var targetGroups []TargetGroup
 | 
						|
	err := decoder.Decode(&targetGroups)
 | 
						|
	if err != nil {
 | 
						|
		http.Error(w, err.Error(), http.StatusBadRequest)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	job := serv.Config.GetJobByName(jobName)
 | 
						|
	if job == nil {
 | 
						|
		http.Error(w, "job not found", http.StatusNotFound)
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	newTargets := []retrieval.Target{}
 | 
						|
 | 
						|
	for _, targetGroup := range targetGroups {
 | 
						|
		// Do mandatory map type conversion due to Go shortcomings.
 | 
						|
		baseLabels := clientmodel.LabelSet{
 | 
						|
			clientmodel.JobLabel: clientmodel.LabelValue(job.GetName()),
 | 
						|
		}
 | 
						|
		for label, value := range targetGroup.BaseLabels {
 | 
						|
			baseLabels[clientmodel.LabelName(label)] = clientmodel.LabelValue(value)
 | 
						|
		}
 | 
						|
 | 
						|
		for _, endpoint := range targetGroup.Endpoints {
 | 
						|
			newTarget := retrieval.NewTarget(endpoint, job.ScrapeTimeout(), baseLabels)
 | 
						|
			newTargets = append(newTargets, newTarget)
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	// BUG(julius): Validate that this ScrapeInterval is in fact the proper one
 | 
						|
	// for the job.
 | 
						|
	serv.TargetManager.ReplaceTargets(*job, newTargets)
 | 
						|
}
 |