54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/derailed/k9s/internal"
|
|
"github.com/derailed/k9s/internal/client"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
restclient "k8s.io/client-go/rest"
|
|
)
|
|
|
|
// Container represents a pod's container dao.
|
|
type Container struct {
|
|
Generic
|
|
}
|
|
|
|
var _ Accessor = (*Container)(nil)
|
|
var _ Loggable = (*Container)(nil)
|
|
|
|
// TailLogs tails a given container logs
|
|
func (c *Container) TailLogs(ctx context.Context, logChan chan<- string, opts LogOptions) error {
|
|
fac, ok := ctx.Value(internal.KeyFactory).(Factory)
|
|
if !ok {
|
|
return errors.New("Expecting an informer")
|
|
}
|
|
o, err := fac.Get("v1/pods", opts.Path, true, labels.Everything())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var po v1.Pod
|
|
if err := runtime.DefaultUnstructuredConverter.FromUnstructured(o.(*unstructured.Unstructured).Object, &po); err != nil {
|
|
return err
|
|
}
|
|
|
|
return tailLogs(ctx, c, logChan, opts)
|
|
}
|
|
|
|
// Logs fetch container logs for a given pod and container.
|
|
func (c *Container) Logs(path string, opts *v1.PodLogOptions) (*restclient.Request, error) {
|
|
ns, _ := client.Namespaced(path)
|
|
auth, err := c.Client().CanI(ns, "v1/pods:log", []string{"get"})
|
|
if !auth || err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ns, n := client.Namespaced(path)
|
|
return c.Client().DialOrDie().CoreV1().Pods(ns).GetLogs(n, opts), nil
|
|
}
|