mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-21 21:33:18 +01:00
100 lines
3.8 KiB
Go
100 lines
3.8 KiB
Go
package docker_support
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/amir20/dozzle/internal/container"
|
|
)
|
|
|
|
type ClientService interface {
|
|
FindContainer(ctx context.Context, id string, filter container.ContainerFilter) (container.Container, error)
|
|
ListContainers(ctx context.Context, filter container.ContainerFilter) ([]container.Container, error)
|
|
Host(ctx context.Context) (container.Host, error)
|
|
ContainerAction(ctx context.Context, container container.Container, action container.ContainerAction) error
|
|
LogsBetweenDates(ctx context.Context, container container.Container, from time.Time, to time.Time, stdTypes container.StdType) (<-chan *container.LogEvent, error)
|
|
RawLogs(ctx context.Context, container container.Container, from time.Time, to time.Time, stdTypes container.StdType) (io.ReadCloser, error)
|
|
|
|
// Subscriptions
|
|
SubscribeStats(ctx context.Context, stats chan<- container.ContainerStat)
|
|
SubscribeEvents(ctx context.Context, events chan<- container.ContainerEvent)
|
|
SubscribeContainersStarted(ctx context.Context, containers chan<- container.Container)
|
|
|
|
// Blocking streaming functions that should be used in a goroutine
|
|
StreamLogs(ctx context.Context, container container.Container, from time.Time, stdTypes container.StdType, events chan<- *container.LogEvent) error
|
|
}
|
|
|
|
type dockerClientService struct {
|
|
client container.Client
|
|
store *container.ContainerStore
|
|
}
|
|
|
|
func NewDockerClientService(client container.Client, filter container.ContainerFilter) ClientService {
|
|
return &dockerClientService{
|
|
client: client,
|
|
store: container.NewContainerStore(context.Background(), client, filter),
|
|
}
|
|
}
|
|
|
|
func (d *dockerClientService) RawLogs(ctx context.Context, container container.Container, from time.Time, to time.Time, stdTypes container.StdType) (io.ReadCloser, error) {
|
|
return d.client.ContainerLogsBetweenDates(ctx, container.ID, from, to, stdTypes)
|
|
}
|
|
|
|
func (d *dockerClientService) LogsBetweenDates(ctx context.Context, c container.Container, from time.Time, to time.Time, stdTypes container.StdType) (<-chan *container.LogEvent, error) {
|
|
reader, err := d.client.ContainerLogsBetweenDates(ctx, c.ID, from, to, stdTypes)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
g := container.NewEventGenerator(ctx, reader, c)
|
|
return g.Events, nil
|
|
}
|
|
|
|
func (d *dockerClientService) StreamLogs(ctx context.Context, c container.Container, from time.Time, stdTypes container.StdType, events chan<- *container.LogEvent) error {
|
|
reader, err := d.client.ContainerLogs(ctx, c.ID, from, stdTypes)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
g := container.NewEventGenerator(ctx, reader, c)
|
|
for event := range g.Events {
|
|
events <- event
|
|
}
|
|
|
|
select {
|
|
case e := <-g.Errors:
|
|
return e
|
|
default:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (d *dockerClientService) FindContainer(ctx context.Context, id string, filter container.ContainerFilter) (container.Container, error) {
|
|
return d.store.FindContainer(id, filter)
|
|
}
|
|
|
|
func (d *dockerClientService) ContainerAction(ctx context.Context, container container.Container, action container.ContainerAction) error {
|
|
return d.client.ContainerActions(ctx, action, container.ID)
|
|
}
|
|
|
|
func (d *dockerClientService) ListContainers(ctx context.Context, filter container.ContainerFilter) ([]container.Container, error) {
|
|
return d.store.ListContainers(filter)
|
|
}
|
|
|
|
func (d *dockerClientService) Host(ctx context.Context) (container.Host, error) {
|
|
return d.client.Host(), nil
|
|
}
|
|
|
|
func (d *dockerClientService) SubscribeStats(ctx context.Context, stats chan<- container.ContainerStat) {
|
|
d.store.SubscribeStats(ctx, stats)
|
|
}
|
|
|
|
func (d *dockerClientService) SubscribeEvents(ctx context.Context, events chan<- container.ContainerEvent) {
|
|
d.store.SubscribeEvents(ctx, events)
|
|
}
|
|
|
|
func (d *dockerClientService) SubscribeContainersStarted(ctx context.Context, containers chan<- container.Container) {
|
|
d.store.SubscribeNewContainers(ctx, containers)
|
|
}
|