mirror of
https://github.com/siderolabs/talos.git
synced 2025-09-02 12:31:12 +02:00
070cbc9d60
Instead of pulling a full list of containers, implement inspector query for a single container following the spec to build display name. Also adds many more tests for the container inspector. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
package containers
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/containerd/containerd"
|
|
tasks "github.com/containerd/containerd/api/services/tasks/v1"
|
|
"github.com/containerd/containerd/api/types"
|
|
|
|
"github.com/talos-systems/talos/internal/pkg/chunker"
|
|
"github.com/talos-systems/talos/internal/pkg/chunker/file"
|
|
"github.com/talos-systems/talos/internal/pkg/chunker/stream"
|
|
)
|
|
|
|
// Container presents information about a container
|
|
type Container struct {
|
|
inspector *Inspector
|
|
|
|
Display string // Friendly Name
|
|
Name string // container name
|
|
ID string // container sha/id
|
|
Digest string // Container Digest
|
|
Image string
|
|
PodName string
|
|
Sandbox string
|
|
Status containerd.Status // Running state of container
|
|
Pid uint32
|
|
RestartCount string
|
|
Metrics *types.Metric
|
|
}
|
|
|
|
// GetProcessStderr returns process stderr
|
|
func (c *Container) GetProcessStderr() (string, error) {
|
|
task, err := c.inspector.client.TaskService().Get(c.inspector.nsctx, &tasks.GetRequest{ContainerID: c.ID})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return task.Process.Stderr, nil
|
|
}
|
|
|
|
// GetLogFile returns path to log file, k8s-style
|
|
func (c *Container) GetLogFile() string {
|
|
if c.Sandbox == "" || !strings.Contains(c.Display, ":") {
|
|
return ""
|
|
}
|
|
|
|
return filepath.Join(c.Sandbox, c.Name, c.RestartCount+".log")
|
|
}
|
|
|
|
// Kill sends signal to container task
|
|
func (c *Container) Kill(signal syscall.Signal) error {
|
|
_, err := c.inspector.client.TaskService().Kill(c.inspector.nsctx, &tasks.KillRequest{ContainerID: c.ID, Signal: uint32(signal)})
|
|
return err
|
|
}
|
|
|
|
// GetLogChunker returns chunker for container log file
|
|
func (c *Container) GetLogChunker() (chunker.Chunker, io.Closer, error) {
|
|
logFile := c.GetLogFile()
|
|
log.Printf("logFile = %q", logFile)
|
|
if logFile != "" {
|
|
f, err := os.OpenFile(logFile, os.O_RDONLY, 0)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return file.NewChunker(f), f, nil
|
|
}
|
|
|
|
filename, err := c.GetProcessStderr()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
if filename == "" {
|
|
return nil, nil, fmt.Errorf("no log available")
|
|
}
|
|
|
|
f, err := os.OpenFile(filename, os.O_RDONLY, 0)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return stream.NewChunker(f), f, nil
|
|
}
|