8e5b17cf13
Signed-off-by: Mrunal Patel <mrunalp@gmail.com>
258 lines
8.3 KiB
Go
258 lines
8.3 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors.
|
|
|
|
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 kuberuntime
|
|
|
|
import (
|
|
"time"
|
|
|
|
internalapi "k8s.io/kubernetes/pkg/kubelet/api"
|
|
runtimeapi "k8s.io/kubernetes/pkg/kubelet/api/v1alpha1/runtime"
|
|
"k8s.io/kubernetes/pkg/kubelet/metrics"
|
|
)
|
|
|
|
// instrumentedRuntimeService wraps the RuntimeService and records the operations
|
|
// and errors metrics.
|
|
type instrumentedRuntimeService struct {
|
|
service internalapi.RuntimeService
|
|
}
|
|
|
|
// Creates an instrumented RuntimeInterface from an existing RuntimeService.
|
|
func NewInstrumentedRuntimeService(service internalapi.RuntimeService) internalapi.RuntimeService {
|
|
return &instrumentedRuntimeService{service: service}
|
|
}
|
|
|
|
// instrumentedImageManagerService wraps the ImageManagerService and records the operations
|
|
// and errors metrics.
|
|
type instrumentedImageManagerService struct {
|
|
service internalapi.ImageManagerService
|
|
}
|
|
|
|
// Creates an instrumented ImageManagerService from an existing ImageManagerService.
|
|
func NewInstrumentedImageManagerService(service internalapi.ImageManagerService) internalapi.ImageManagerService {
|
|
return &instrumentedImageManagerService{service: service}
|
|
}
|
|
|
|
// recordOperation records the duration of the operation.
|
|
func recordOperation(operation string, start time.Time) {
|
|
metrics.RuntimeOperations.WithLabelValues(operation).Inc()
|
|
metrics.RuntimeOperationsLatency.WithLabelValues(operation).Observe(metrics.SinceInMicroseconds(start))
|
|
}
|
|
|
|
// recordError records error for metric if an error occurred.
|
|
func recordError(operation string, err error) {
|
|
if err != nil {
|
|
metrics.RuntimeOperationsErrors.WithLabelValues(operation).Inc()
|
|
}
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) Version(apiVersion string) (*runtimeapi.VersionResponse, error) {
|
|
const operation = "version"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.Version(apiVersion)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) Status() (*runtimeapi.RuntimeStatus, error) {
|
|
const operation = "status"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.Status()
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) CreateContainer(podSandboxID string, config *runtimeapi.ContainerConfig, sandboxConfig *runtimeapi.PodSandboxConfig) (string, error) {
|
|
const operation = "create_container"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.CreateContainer(podSandboxID, config, sandboxConfig)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) StartContainer(containerID string) error {
|
|
const operation = "start_container"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.StartContainer(containerID)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) StopContainer(containerID string, timeout int64) error {
|
|
const operation = "stop_container"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.StopContainer(containerID, timeout)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) RemoveContainer(containerID string) error {
|
|
const operation = "remove_container"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.RemoveContainer(containerID)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) ListContainers(filter *runtimeapi.ContainerFilter) ([]*runtimeapi.Container, error) {
|
|
const operation = "list_containers"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.ListContainers(filter)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) ContainerStatus(containerID string) (*runtimeapi.ContainerStatus, error) {
|
|
const operation = "container_status"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.ContainerStatus(containerID)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) ExecSync(containerID string, cmd []string, timeout time.Duration) ([]byte, []byte, error) {
|
|
const operation = "exec_sync"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
stdout, stderr, err := in.service.ExecSync(containerID, cmd, timeout)
|
|
recordError(operation, err)
|
|
return stdout, stderr, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) Exec(req *runtimeapi.ExecRequest) (*runtimeapi.ExecResponse, error) {
|
|
const operation = "exec"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
resp, err := in.service.Exec(req)
|
|
recordError(operation, err)
|
|
return resp, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) Attach(req *runtimeapi.AttachRequest) (*runtimeapi.AttachResponse, error) {
|
|
const operation = "attach"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
resp, err := in.service.Attach(req)
|
|
recordError(operation, err)
|
|
return resp, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) RunPodSandbox(config *runtimeapi.PodSandboxConfig) (string, error) {
|
|
const operation = "run_podsandbox"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.RunPodSandbox(config)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) StopPodSandbox(podSandboxID string) error {
|
|
const operation = "stop_podsandbox"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.StopPodSandbox(podSandboxID)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) RemovePodSandbox(podSandboxID string) error {
|
|
const operation = "remove_podsandbox"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.RemovePodSandbox(podSandboxID)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) PodSandboxStatus(podSandboxID string) (*runtimeapi.PodSandboxStatus, error) {
|
|
const operation = "podsandbox_status"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.PodSandboxStatus(podSandboxID)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) ListPodSandbox(filter *runtimeapi.PodSandboxFilter) ([]*runtimeapi.PodSandbox, error) {
|
|
const operation = "list_podsandbox"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.ListPodSandbox(filter)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) PortForward(req *runtimeapi.PortForwardRequest) (*runtimeapi.PortForwardResponse, error) {
|
|
const operation = "port_forward"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
resp, err := in.service.PortForward(req)
|
|
recordError(operation, err)
|
|
return resp, err
|
|
}
|
|
|
|
func (in instrumentedRuntimeService) UpdateRuntimeConfig(runtimeConfig *runtimeapi.RuntimeConfig) error {
|
|
const operation = "update_runtime_config"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.UpdateRuntimeConfig(runtimeConfig)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|
|
|
|
func (in instrumentedImageManagerService) ListImages(filter *runtimeapi.ImageFilter) ([]*runtimeapi.Image, error) {
|
|
const operation = "list_images"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.ListImages(filter)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedImageManagerService) ImageStatus(image *runtimeapi.ImageSpec) (*runtimeapi.Image, error) {
|
|
const operation = "image_status"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
out, err := in.service.ImageStatus(image)
|
|
recordError(operation, err)
|
|
return out, err
|
|
}
|
|
|
|
func (in instrumentedImageManagerService) PullImage(image *runtimeapi.ImageSpec, auth *runtimeapi.AuthConfig) (string, error) {
|
|
const operation = "pull_image"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
imageRef, err := in.service.PullImage(image, auth)
|
|
recordError(operation, err)
|
|
return imageRef, err
|
|
}
|
|
|
|
func (in instrumentedImageManagerService) RemoveImage(image *runtimeapi.ImageSpec) error {
|
|
const operation = "remove_image"
|
|
defer recordOperation(operation, time.Now())
|
|
|
|
err := in.service.RemoveImage(image)
|
|
recordError(operation, err)
|
|
return err
|
|
}
|