mirror of
				https://github.com/optim-enterprises-bv/kubernetes.git
				synced 2025-11-04 04:08:16 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			124 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			124 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
/*
 | 
						|
Copyright 2014 The Kubernetes Authors All rights reserved.
 | 
						|
 | 
						|
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 unversioned
 | 
						|
 | 
						|
import (
 | 
						|
	"bytes"
 | 
						|
	"encoding/json"
 | 
						|
	"fmt"
 | 
						|
	"io"
 | 
						|
	"net"
 | 
						|
	"net/http"
 | 
						|
	"strconv"
 | 
						|
 | 
						|
	cadvisorapi "github.com/google/cadvisor/info/v1"
 | 
						|
)
 | 
						|
 | 
						|
type ContainerInfoGetter interface {
 | 
						|
	// GetContainerInfo returns information about a container.
 | 
						|
	GetContainerInfo(host, podID, containerID string, req *cadvisorapi.ContainerInfoRequest) (*cadvisorapi.ContainerInfo, error)
 | 
						|
	// GetRootInfo returns information about the root container on a machine.
 | 
						|
	GetRootInfo(host string, req *cadvisorapi.ContainerInfoRequest) (*cadvisorapi.ContainerInfo, error)
 | 
						|
	// GetMachineInfo returns the machine's information like number of cores, memory capacity.
 | 
						|
	GetMachineInfo(host string) (*cadvisorapi.MachineInfo, error)
 | 
						|
}
 | 
						|
 | 
						|
type HTTPContainerInfoGetter struct {
 | 
						|
	Client *http.Client
 | 
						|
	Port   int
 | 
						|
}
 | 
						|
 | 
						|
func (self *HTTPContainerInfoGetter) GetMachineInfo(host string) (*cadvisorapi.MachineInfo, error) {
 | 
						|
	request, err := http.NewRequest(
 | 
						|
		"GET",
 | 
						|
		fmt.Sprintf("http://%v/spec",
 | 
						|
			net.JoinHostPort(host, strconv.Itoa(self.Port)),
 | 
						|
		),
 | 
						|
		nil,
 | 
						|
	)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	response, err := self.Client.Do(request)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer response.Body.Close()
 | 
						|
	if response.StatusCode != http.StatusOK {
 | 
						|
		return nil, fmt.Errorf("trying to get machine spec from %v; received status %v",
 | 
						|
			host, response.Status)
 | 
						|
	}
 | 
						|
	var minfo cadvisorapi.MachineInfo
 | 
						|
	err = json.NewDecoder(response.Body).Decode(&minfo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &minfo, nil
 | 
						|
}
 | 
						|
 | 
						|
func (self *HTTPContainerInfoGetter) getContainerInfo(host, path string, req *cadvisorapi.ContainerInfoRequest) (*cadvisorapi.ContainerInfo, error) {
 | 
						|
	var body io.Reader
 | 
						|
	if req != nil {
 | 
						|
		content, err := json.Marshal(req)
 | 
						|
		if err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		body = bytes.NewBuffer(content)
 | 
						|
	}
 | 
						|
 | 
						|
	request, err := http.NewRequest(
 | 
						|
		"GET",
 | 
						|
		fmt.Sprintf("http://%v/stats/%v",
 | 
						|
			net.JoinHostPort(host, strconv.Itoa(self.Port)),
 | 
						|
			path,
 | 
						|
		),
 | 
						|
		body,
 | 
						|
	)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	response, err := self.Client.Do(request)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer response.Body.Close()
 | 
						|
	if response.StatusCode != http.StatusOK {
 | 
						|
		return nil, fmt.Errorf("trying to get info for %v from %v; received status %v",
 | 
						|
			path, host, response.Status)
 | 
						|
	}
 | 
						|
	var cinfo cadvisorapi.ContainerInfo
 | 
						|
	err = json.NewDecoder(response.Body).Decode(&cinfo)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &cinfo, nil
 | 
						|
}
 | 
						|
 | 
						|
func (self *HTTPContainerInfoGetter) GetContainerInfo(host, podID, containerID string, req *cadvisorapi.ContainerInfoRequest) (*cadvisorapi.ContainerInfo, error) {
 | 
						|
	return self.getContainerInfo(
 | 
						|
		host,
 | 
						|
		fmt.Sprintf("%v/%v", podID, containerID),
 | 
						|
		req,
 | 
						|
	)
 | 
						|
}
 | 
						|
 | 
						|
func (self *HTTPContainerInfoGetter) GetRootInfo(host string, req *cadvisorapi.ContainerInfoRequest) (*cadvisorapi.ContainerInfo, error) {
 | 
						|
	return self.getContainerInfo(host, "", req)
 | 
						|
}
 |